Skip to content

Commit de3f6e1

Browse files
committed
added reference data tot est
1 parent 0966743 commit de3f6e1

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

rormula/test/test_arithmetic.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -61,13 +61,15 @@ def eval_asdf():
6161
res = rormula.eval_asdf(df)
6262
assert np.allclose(res.to_numpy().item(), (5.0 - 2.5) / 4.0)
6363

64-
data = np.ones((100, 6))
64+
data = np.random.random((100, 6)) * 1000
6565
df = pd.DataFrame(
6666
data=data, columns=["alpha", "beta", "gamma", "delta", "epsilon", "phi"]
6767
)
6868
s = "5/3 * alpha / beta * (0.2 / 200.0 / (29.22+gamma+epsilon+phi) / 1000)"
6969
rormula = Arithmetic(s, "testslash")
7070
res = rormula.eval_asdf(df)
71+
ref = df.eval(s)
72+
np.allclose(res.values, ref.values)
7173

7274

7375
if __name__ == "__main__":

0 commit comments

Comments
 (0)