added tests for lambdify with optlevel

Author: ichumuh

symengine/tests/test_lambdify.py

@@ -75,6 +75,10 @@ def test_Lambdify():
     assert allclose(L(range(n, n+len(args))),
                     [3*n+3, n**2, -1/(n+2), n*(n+1)*(n+2)])
 
+@unittest.skipUnless(have_numpy, "Numpy not installed")
+def test_Lambdify_with_opt_level():
+    args = x, y, z = se.symbols('x y z')
+    raises(TypeError, lambda: se.Lambdify(args, [x+y+z, x**2, (x-y)/z, x*y*z], backend='lambda', opt_level=0))
 
 def _test_Lambdify_Piecewise(Lambdify):
     x = se.symbols('x')
@@ -91,7 +95,6 @@ def test_Lambdify_Piecewise():
     if se.have_llvm:
         _test_Lambdify_Piecewise(lambda *args: se.Lambdify(*args, backend='llvm'))
 
-
 @unittest.skipUnless(have_numpy, "Numpy not installed")
 def test_Lambdify_LLVM():
     n = 7
@@ -105,6 +108,19 @@ def test_Lambdify_LLVM():
     assert allclose(L(range(n, n+len(args))),
                     [3*n+3, n**2, -1/(n+2), n*(n+1)*(n+2)])
 
+@unittest.skipUnless(have_numpy, "Numpy not installed")
+def test_Lambdify_LLVM_with_opt_level():
+    for opt_level in range(4):
+        n = 7
+        args = x, y, z = se.symbols('x y z')
+        if not se.have_llvm:
+            raises(ValueError, lambda: se.Lambdify(args, [x+y+z, x**2,
+                                                          (x-y)/z, x*y*z],
+                                                   backend='llvm', opt_level=opt_level))
+            raise SkipTest("No LLVM support")
+        L = se.Lambdify(args, [x+y+z, x**2, (x-y)/z, x*y*z], backend='llvm', opt_level=opt_level)
+        assert allclose(L(range(n, n+len(args))),
+                        [3*n+3, n**2, -1/(n+2), n*(n+1)*(n+2)])
 
 def _get_2_to_2by2():
     args = x, y = se.symbols('x y')