def test_hard_call() -> None: func_tokens = "function f one argument file name return one".split() engine = RuleEngine(func_tokens) code = Code() code = engine.parse(code) tokens = "call f file path".split() engine.add_tokens(tokens) assert engine.parse_call(code) == "f(file_path)" assert engine.tokens == "".split()
def test_call_with_multi_words4() -> None: func_tokens = "function read file one argument file name return one".split() engine = RuleEngine(func_tokens) code = Code() code = engine.parse(code) tokens = "call read file file path".split() engine.add_tokens(tokens) assert engine.parse_call(code) == "read_file(file_path)" assert engine.tokens == "".split()
def test_basic_call_with_digits() -> None: func_tokens = "function f two arguments a and b return a plus b".split() engine = RuleEngine(func_tokens) code = Code() code = engine.parse(code) code.symbols.add_variable_symbol("a") code.symbols.add_variable_symbol("b") tokens = "call f two four".split() engine.add_tokens(tokens) assert engine.parse_call(code) == "f(2, 4)" assert engine.tokens == "".split()
def test_call_with_multi_words2() -> None: func_tokens = ( "function read file two arguments file name and length return length".split() ) engine = RuleEngine(func_tokens) code = Code() code = engine.parse(code) code.symbols.add_variable_symbol("file_path") tokens = "call read file foo dot txt twelve".split() engine.add_tokens(tokens) assert engine.parse_call(code) == "read_file('foo.txt', 12)" assert engine.tokens == "".split()
def test_delete_line() -> None: tokens = "function y two argument x and y".split() rule_engine = RuleEngine(tokens) code = Code([]) for i in range(10): tokens = f"set x to {i}".split() rule_engine.add_tokens(tokens) tokens = "delete line three".split() rule_engine.add_tokens(tokens) code = rule_engine.parse(code) assert len(code.lines) == 10 assert "1" not in code.print_lines()
def test_goto_three() -> None: tokens = "function y two argument x and y".split() rule_engine = RuleEngine(tokens) code = Code([]) for i in range(20, 30): tokens = f"set x to {i}".split() rule_engine.add_tokens(tokens) tokens = "goto line three".split() rule_engine.add_tokens(tokens) code = rule_engine.parse(code) print(code.print_lines()) assert len(code.lines) == 11 assert code.cursor_position == 3
def test_goto_beginning() -> None: tokens = "function y two argument x and y".split() rule_engine = RuleEngine(tokens) code = Code([]) for i in range(10, 20): tokens = f"set x to {i}".split() rule_engine.add_tokens(tokens) tokens = "goto line beginning".split() rule_engine.add_tokens(tokens) code = rule_engine.parse(code) assert len(code.lines) == 11 assert code.cursor_position == 1
def test_goto_end() -> None: tokens = "function y two argument x and y".split() rule_engine = RuleEngine(tokens) code = Code([]) for i in range(10): tokens = f"set x to {i}".split() rule_engine.add_tokens(tokens) tokens = "goto line end".split() rule_engine.add_tokens(tokens) print(code.print_lines()) new_code = rule_engine.parse(code) print(new_code.print_lines()) print(rule_engine.tokens) assert len(new_code.lines) == 11 assert new_code.cursor_position == 11