def test_base() -> None:
tokens = "function main zero arguments".split()
engine = RuleEngine(tokens)
code = Code()
assert engine.parse_function(code) == "def main():"
assert engine.tokens == "".split()
def test_multi_word_func_one_multi_word_arg() -> None:
tokens = "function read file one argument file location".split()
engine = RuleEngine(tokens)
code = Code()
assert engine.parse_function(code) == "def read_file(file_location):"
assert engine.tokens == "".split()
def test_func_zero_args() -> None:
tokens = "function f zero arguments".split()
engine = RuleEngine(tokens)
code = Code()
assert engine.parse_function(code) == "def f():"
assert engine.tokens == "".split()
def test_main_func() -> None:
tokens = ["function", "main", "zero", "arguments"]
code = Code()
engine = RuleEngine()
engine.add_tokens(tokens)
assert engine.parse_function(code) == "def main():"
assert engine.tokens == "".split()
def test_basic_func() -> None:
tokens = "function f two arguments a and b".split()
engine = RuleEngine(tokens)
code = Code()
assert engine.parse_function(code) == "def f(a, b):"
assert engine.tokens == "".split()