feat: add support for np.ones_like and np.zeros_like

2026-04-17 03:00:54 -04:00 · 2022-05-17 10:58:33 +02:00
parent 6c20ed7a08
commit 56085204e6
3 changed files with 32 additions and 0 deletions
--- a/tests/execution/test_others.py
+++ b/tests/execution/test_others.py
@@ -447,6 +447,20 @@ def fusable_with_one_of_the_start_nodes_is_lca_generator():
            },
            id="np.where(x < 5, x * 3, x)",
        ),
+        pytest.param(
+            lambda x: x + np.ones_like(x),
+            {
+                "x": {"status": "encrypted", "range": [0, 10]},
+            },
+            id="x + np.ones_like(x)",
+        ),
+        pytest.param(
+            lambda x: x + np.zeros_like(x),
+            {
+                "x": {"status": "encrypted", "range": [0, 10]},
+            },
+            id="x + np.zeros_like(x)",
+        ),
    ],
 )
 def test_others(function, parameters, helpers):