def test_compile_array():
code = vec.compile_expr('(log2 x) + (log2 y)')
env = {'x': np.array([2, 4]), 'y': np.array([4, 8])}
res = code(env)
expected = np.array([3,5])
print "Expected:", expected, "Received:", res
assert np.all(res == expected)
def test_compiler_env():
env = {'x':3, 'y':4}
code = vec.compile_expr('x+y')
res1 = code(env)
print "Expected 7, got", res1
assert res1 == 7
env['x'] = 4
res2 = code(env)
print "Expected 8, got", res2
assert res2 == 8
def test_compiler_nested():
code = vec.compile_expr('(10+4)%2')
res = code({})
print "Received: ", res
assert res == 7
def test_compiler_simple():
code = vec.compile_expr('3+4')
assert code({}) == 7
