fix(frontend-python): mimic summing like numpy exactly

2026-04-17 03:00:54 -04:00 · 2023-07-06 16:30:46 +02:00
parent 1215a57752
commit 9ef1229c51
2 changed files with 18 additions and 0 deletions
--- a/frontends/concrete-python/tests/execution/test_sum.py
+++ b/frontends/concrete-python/tests/execution/test_sum.py
@@ -17,6 +17,18 @@ from concrete import fhe
                "x": {"shape": (3, 2), "range": [0, 10], "status": "encrypted"},
            },
        ),
+        pytest.param(
+            lambda x: np.sum(x, 0),
+            {
+                "x": {"shape": (3, 2), "range": [0, 10], "status": "encrypted"},
+            },
+        ),
+        pytest.param(
+            lambda x: np.sum(x, 1),
+            {
+                "x": {"shape": (3, 2), "range": [0, 10], "status": "encrypted"},
+            },
+        ),
        pytest.param(
            lambda x: np.sum(x, axis=None),  # type: ignore
            {