diff --git a/test/unit/test_uop_symbolic.py b/test/unit/test_uop_symbolic.py index d5fb8a5655..e4552679d6 100644 --- a/test/unit/test_uop_symbolic.py +++ b/test/unit/test_uop_symbolic.py @@ -617,6 +617,7 @@ class TestSymInfer(unittest.TestCase): c = Variable("c", 0, 10) var_vals = {a: 2, b: 3, c: 4} assert sym_infer(5, var_vals) == 5 + assert sym_infer(4.2, var_vals) == 4.2 assert sym_infer(a, var_vals) == 2 assert sym_infer(b, var_vals) == 3 assert sym_infer(a+b, var_vals) == 5