The syntax of with assertRaises changes across python versions

sbuss · Jul 30, 2015 · 5e3a231 · 5e3a231
1 parent 9b49a5f
commit 5e3a231
Showing 1 changed file with 7 additions and 3 deletions.
diff --git a/tests/test_bip32.py b/tests/test_bip32.py
@@ -325,10 +325,14 @@ def test_child_too_big(self):
 
     def test_no_prime_children_of_public_copy(self):
         pub = self.wallet.public_copy()
-        with self.assertRaises(ValueError) as exc:
+        try:
             pub.get_child(0, is_prime=True)
-        self.assertIn("prime child", exc.exception.message)
-        self.assertIn("private key", exc.exception.message)
+            assert False
+        except ValueError as e:
+            err_msg = str(e)
+            self.assertTrue(
+                "Cannot compute a prime child without a private key"
+                in err_msg)
         self.assertTrue(pub.get_child(0))
 
     def test_path_bigger_than_boundary(self):