Merge pull request #352 from mtreinish/add-float32

Add support for 32 and 16 bit numpy floats

Author: rikardn

symengine/lib/symengine_wrapper.pyx

@@ -558,6 +558,8 @@ def _sympify(a, raise_error=True):
         return Integer(a)
     elif isinstance(a, float):
         return RealDouble(a)
+    elif have_numpy and isinstance(a, (np.float16, np.float32)):
+        return RealDouble(a)
     elif isinstance(a, complex):
         return ComplexDouble(a)
     elif hasattr(a, '_symengine_'):

symengine/tests/test_subs.py

@@ -1,5 +1,7 @@
+import unittest
+
 from symengine.utilities import raises
-from symengine import Symbol, sin, cos, sqrt, Add, function_symbol
+from symengine import Symbol, sin, cos, sqrt, Add, function_symbol, have_numpy
 
 
 def test_basic():
@@ -56,3 +58,19 @@ def test_xreplace():
     y = Symbol("y")
     f = sin(cos(x))
     assert f.xreplace({x: y}) == sin(cos(y))
+
+
+@unittest.skipUnless(have_numpy, "Numpy not installed")
+def test_float32():
+    import numpy as np
+    x = Symbol("x")
+    expr = x * 2
+    assert expr.subs({x: np.float32(2)}) == 4.0
+
+
+@unittest.skipUnless(have_numpy, "Numpy not installed")
+def test_float16():
+    import numpy as np
+    x = Symbol("x")
+    expr = x * 2
+    assert expr.subs({x: np.float16(2)}) == 4.0