Handle numpy types in sympify_petab (#294)

Author: dweindl

petab/v1/math/sympify.py

@@ -1,5 +1,6 @@
 """PEtab math to sympy conversion."""
 
+import numpy as np
 import sympy as sp
 from antlr4 import CommonTokenStream, InputStream
 from antlr4.error.ErrorListener import ErrorListener
@@ -25,9 +26,9 @@ def sympify_petab(expr: str | int | float) -> sp.Expr | sp.Basic:
         The sympy expression corresponding to `expr`.
         Boolean values are converted to numeric values.
     """
-    if isinstance(expr, int):
+    if isinstance(expr, int) or isinstance(expr, np.integer):
         return sp.Integer(expr)
-    if isinstance(expr, float):
+    if isinstance(expr, float) or isinstance(expr, np.floating):
         return sp.Float(expr)
 
     # Set error listeners

tests/v1/math/test_math.py

@@ -11,6 +11,10 @@
 from petab.math import sympify_petab
 
 
+def test_sympify_numpy():
+    assert sympify_petab(np.float64(1.0)) == sp.Float(1.0)
+
+
 def test_parse_simple():
     """Test simple numeric expressions."""
     assert float(sympify_petab("1 + 2")) == 3