added reference data tot est

basf · Feb 15, 2024 · de3f6e1 · de3f6e1
1 parent 0966743
commit de3f6e1
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/rormula/test/test_arithmetic.py b/rormula/test/test_arithmetic.py
@@ -61,13 +61,15 @@ def eval_asdf():
     res = rormula.eval_asdf(df)
     assert np.allclose(res.to_numpy().item(), (5.0 - 2.5) / 4.0)
 
-    data = np.ones((100, 6))
+    data = np.random.random((100, 6)) * 1000
     df = pd.DataFrame(
         data=data, columns=["alpha", "beta", "gamma", "delta", "epsilon", "phi"]
     )
     s = "5/3 * alpha / beta * (0.2 / 200.0 / (29.22+gamma+epsilon+phi) / 1000)"
     rormula = Arithmetic(s, "testslash")
     res = rormula.eval_asdf(df)
+    ref = df.eval(s)
+    np.allclose(res.values, ref.values)
 
 
 if __name__ == "__main__":