Fix sympy test

qutech · Jul 13, 2018 · ff6de46 · ff6de46
1 parent 498f7e8
commit ff6de46
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/utils/sympy_tests.py b/tests/utils/sympy_tests.py
@@ -240,7 +240,7 @@ def test_get_variables_indexed(self):
 class EvaluationTests(TestCase):
     def evaluate(self, expression: Union[sympy.Expr, np.ndarray], parameters):
         if isinstance(expression, np.ndarray):
-            variables = set.union(map(set, map(get_variables, expression.flat)))
+            variables = set.union(*map(set, map(get_variables, expression.flat)))
         else:
             variables = get_variables(expression)
         return evaluate_lambdified(expression, variables=list(variables), parameters=parameters, lambdified=None)[0]