added an __iadd__, __isub__ to zero class

2026-07-31 12:40:46 +08:00 · 2015-11-24 10:05:28 -08:00
parent 815311bfec
commit f3fb1e6481
2 changed files with 9 additions and 0 deletions
@@ -20,6 +20,13 @@ class Tests(unittest.TestCase):
        assert 3*z == 0
        assert z*3 == 0
        assert z/3 == 0
+
+        a = 1
+        a += z
+        assert a == 1
+        a = 1
+        a += z
+        assert a == 1
        self.assertRaises(ZeroDivisionError, lambda:3/z)

    def test_mat_zero(self):