def test_multi_line(self):
line = """a = 2
return a
"""
dsl = evaluate_line(line)
def test_parans(self):
line = 'a = (1 + 1) + (1 + 1)'
dsl = evaluate_line(line)
assert dsl.variables['a'] == 4
def test_return_const(self):
line = 'return 2'
assert evaluate_line(line).result == 2
def test_multiple_minus(self):
line = 'a = 3 - 1 - 1'
dsl = evaluate_line(line)
assert dsl.variables['a'] == 1
def test_mix_plus_minus(self):
line = 'a = 3 + 1 - 1 + 2 - 1'
dsl = evaluate_line(line)
assert dsl.variables['a'] == 4
def test_div(self):
line = 'a = 6 / 2'
dsl = evaluate_line(line)
assert dsl.variables['a'] == 3
def test_prod(self):
line = 'a = 3 * 2'
dsl = evaluate_line(line)
assert dsl.variables['a'] == 6
def test_mulitple_plus(self):
line = 'a = 1 + 1 + 1 + 1'
dsl = evaluate_line(line)
assert dsl.variables['a'] == 4
def test_var_ref(self):
line = 'a = 2 * b'
d = DSL()
d.variables['b'] = 3
dsl = evaluate_line(line, d)
assert dsl.variables['a'] == 6
def test_assign(self):
line = 'a = 2'
dsl = evaluate_line(line)
assert dsl.variables['a'] == 2