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