Fix an integration test (#576)

Somehow modulo operator doesn't work with pyarrow, which is executed inside python wrapper in rust code. In this change, we remove usages of the modulo operator.
fennel-ai · Sep 26, 2024 · 7ed5ff6 · 7ed5ff6
1 parent 503fd1e
commit 7ed5ff6
Showing 1 changed file with 8 additions and 7 deletions.
diff --git a/fennel/client_tests/test_preproc_filter.py b/fennel/client_tests/test_preproc_filter.py
@@ -29,28 +29,29 @@ def test_preproc_filter(client):
             cdc="append",
             disorder="14d",
             env="prod",
-            where=lambda x: ((x["age"] > 0) & (x["age"] % 100 == 1)),
+            where=lambda x: ((x["age"] > 0) & (x["age"] == 1)),
         )
         @source(
             webhook.endpoint("A1"),
             cdc="append",
             disorder="14d",
             env="staging",
+            # repeat the conditions to test long lamda
             where=lambda x: (
                 (x["age"] > 0)
-                & (x["age"] % 100 == 1)
+                & (x["age"] == 1)
                 & (x["age"] > 0)
-                & (x["age"] % 100 == 1)
+                & (x["age"] == 1)
                 & True
                 & (x["age"] > 0)
-                & (x["age"] % 100 == 1)
+                & (x["age"] == 1)
                 & (x["age"] > 0)
-                & (x["age"] % 100 == 1)
+                & (x["age"] == 1)
                 & True
                 & (x["age"] > 0)
-                & (x["age"] % 100 == 1)
+                & (x["age"] == 1)
                 & (x["age"] > 0)
-                & (x["age"] % 100 == 1)
+                & (x["age"] == 1)
                 & True
             ),
         )