From 19cd7d0fa80ce7ccc8f7d1c856076e9b9b042a6c Mon Sep 17 00:00:00 2001
From: NAUD Maxence <maxence.naud@cea.fr>
Date: Mon, 17 Mar 2025 13:52:44 +0000
Subject: [PATCH] Upd squeeze python unit-tests
---
aidge_core/unit_tests/test_operator_squeeze.py | 4 +---
1 file changed, 1 insertion(+), 3 deletions(-)
diff --git a/aidge_core/unit_tests/test_operator_squeeze.py b/aidge_core/unit_tests/test_operator_squeeze.py
index b43605893..52120bba8 100644
--- a/aidge_core/unit_tests/test_operator_squeeze.py
+++ b/aidge_core/unit_tests/test_operator_squeeze.py
@@ -184,9 +184,7 @@ class TestSqueeze(unittest.TestCase):
input_data.set_backend("cpu")
test_squeeze_op.set_input(0, input_data)
- with self.assertRaises((RuntimeError, AssertionError)):
- test_squeeze_op.forward_dims()
- test_squeeze_op.forward()
+ self.assertFalse(test_squeeze_op.forward_dims())
return
--
GitLab