'''test_read.py
Tests the L-system class (Lsystem) read method with the following L-systems:
- systemKoch.txt (one rule)
- systemC.txt (two rules)
Hannen Wolfe & Oliver W. Layton
CS151: Computational Thinking: Visual Media
Fall 2023
'''
import lsystem


def testKoch():
    print('Testing systemKoch.txt...')

    filename = 'systemKoch.txt'

    try:
        f = open(filename, 'r')
        f.close()
    except FileNotFoundError:
        print('Could not find', filename, '. Is it in your lab folder?')
        exit()

    lsys = lsystem.Lsystem(filename=filename)

    print('Your base string is\n', lsys.getBase(), 'and it should be\n F++F++F')
    print('There are', lsys.getKeys(), 'keys and there should be [\'F\']')
    rule = lsys.getRule("F")
    print('Your rule is\n', rule, ' and should be\n F-F++F-F')

    print('\nTesting out your set methods...')
    print('Trying to set the base string with F++F...')
    lsys.setBase('F++F')
    print('Your base string is\n', lsys.getBase(), 'and it should be\n F++F')
    print('Trying to add the rule [+, -]...')
    lsys.addRule('+', '-')
    print('There are', lsys.getKeys(), 'keys and there should be [\'F\',\'+\']')
    rule1 = lsys.getRule('F')
    rule2 = lsys.getRule('+')
    print('Your 1st rule is\n', rule1, ' and should be\n  F-F++F-F')
    print('Your 2nd rule is\n', rule2, ' and should be\n -')


def testC():
    print('Testing systemC.txt...')

    filename = 'systemC.txt'

    try:
        f = open(filename, 'r')
        f.close()
    except FileNotFoundError:
        print('Could not find', filename, '. Is it in your lab folder?')
        exit()

    lsys = lsystem.Lsystem()
    lsys.read(filename)

    print('Your base string is\n', lsys.getBase(), 'and it should be\n X')
    print('There are', lsys.getKeys(), 'keys and there should be [\'X\',\'F\']')
    rule1 = lsys.getRule('X')
    rule2 = lsys.getRule('F')
    print('Your 1st rule is\n', rule1, ' and should be\n  F[+XL][-XL]FXL')
    print('Your 2nd rule is\n', rule2, ' and should be\n FF')


if __name__ == '__main__':
    print('----------------------------------------------------------')
    testKoch()
    print('----------------------------------------------------------')
    testC()