explicit cast before test

dlakaplan · Jun 17, 2024 · ee48713 · ee48713
1 parent 2e3ecb2
commit ee48713
Showing 1 changed file with 13 additions and 5 deletions.
diff --git a/tests/test_model_derivatives.py b/tests/test_model_derivatives.py
@@ -163,19 +163,27 @@ def f(value):
         h3 = model.H3.value
         stepgen = numdifftools.MaxStepGenerator(abs(h3) / 2)
     elif param == "FB0":
-        stepgen = numdifftools.MaxStepGenerator(np.abs(model.FB0.value) * 1e-2)
+        stepgen = numdifftools.MaxStepGenerator(
+            np.abs(model.FB0.value.astype(np.float64)) * 1e-2
+        )
     elif param == "FB1":
-        stepgen = numdifftools.MaxStepGenerator(np.abs(model.FB1.value) * 1e3)
+        stepgen = numdifftools.MaxStepGenerator(
+            np.abs(model.FB1.value.astype(np.float64)) * 1e3
+        )
     elif param == "FB2":
-        stepgen = numdifftools.MaxStepGenerator(np.abs(model.FB2.value) * 1e5)
+        stepgen = numdifftools.MaxStepGenerator(
+            np.abs(model.FB2.value.astype(np.float64)) * 1e5
+        )
     elif param == "FB3":
-        stepgen = numdifftools.MaxStepGenerator(np.abs(model.FB3.value) * 1e7)
+        stepgen = numdifftools.MaxStepGenerator(
+            np.abs(model.FB3.value.astype(np.float64)) * 1e7
+        )
     else:
         stepgen = None
     df = numdifftools.Derivative(f, step=stepgen)
 
     a = model.d_phase_d_param(toas, delay=None, param=param).to_value(1 / units)
-    b = df(getattr(model, param).value)
+    b = df(getattr(model, param).value.astype(np.float64))
     if param.startswith("FB"):
         assert np.amax(np.abs(a - b)) / np.amax(np.abs(a) + np.abs(b)) < 1.5e-6
     else: