Fix py2 incompatible usage of assertRaises

Officially, assertRaises can be called as context manager starting in python3, even though it seems the tests are passing, so I guess it was backported some time later, but we can't really rely on that.
memo-off · Sep 6, 2018 · 31e103f · 31e103f
1 parent ab548e3
commit 31e103f
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/tests/test_magic.py b/tests/test_magic.py
@@ -58,8 +58,9 @@ def test_hash_obj(self):
         self.assertEqual(hash(mod.Foo()), 1234)
         self.assertEqual(hash(mod.Bar()), 1234)
         if is_py3:
-            with self.assertRaises(TypeError):
-                hash(mod.Mux())
+            # py3 implicitly adds '__hash__=None' during class construction
+            # if '__eq__ is defined:
+            self.assertRaises(TypeError, lambda: hash(mod.Mux()))
         else:
             self.assertEqual(hash(mod.Mux()), 1234)