Added tests for gcd_ and lcm_ to improve code coverage

helmholtz-analytics · Dec 6, 2023 · 98d15f6 · 98d15f6 · github-actions · Dec 6, 2023
1 parent 0b9c622
commit 98d15f6
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/heat/core/tests/test_arithmetics.py b/heat/core/tests/test_arithmetics.py
@@ -2382,6 +2382,9 @@ def test_gcd_(self):
         with self.assertRaises(TypeError):
             a_tensor.gcd_(an_int_tensor)
         a_tensor = self.a_tensor.copy()  # reset
+        with self.assertRaises(TypeError):
+            a_tensor.gcd_(a_tensor)
+        a_tensor = self.a_tensor.copy()  # reset
         with self.assertRaises(ValueError):
             an_int_tensor.gcd_(another_int_vector)
         an_int_tensor = self.an_int_tensor.copy()  # reset
@@ -2806,6 +2809,9 @@ def test_lcm_(self):
         with self.assertRaises(TypeError):
             a_tensor.lcm_(an_int_tensor)
         a_tensor = self.a_tensor.copy()  # reset
+        with self.assertRaises(TypeError):
+            a_tensor.lcm_(a_tensor)
+        a_tensor = self.a_tensor.copy()  # reset
         with self.assertRaises(ValueError):
             an_int_tensor.lcm_(another_int_vector)
         an_int_tensor = self.an_int_tensor.copy()  # reset