test for until

2022-04-13 13:12:19 +02:00 · 2022-04-13 13:12:19 +02:00 · b4f279c549
commit b4f279c549
parent f5881f2229
1 changed files with 6 additions and 4 deletions
--- a/test/unit_tests/utils_test.py
+++ b/test/unit_tests/utils_test.py
@ -1,6 +1,8 @@
-from funcy import rcompose
+from image_prediction.utils.generic import until


-def test_rcompose():
-    f = rcompose(lambda x: x ** 2, str, lambda x: x * 2)
-    assert f(3) == "99"
+def test_until():
+    def f(x):
+        return x / 2
+
+    assert until(lambda x, y: x - y == 0, f, 1) == 0