Use assertRaises in test_iter

jwg4 · web-flow · commit b60b9b7103de · 2021-12-22T18:07:04.000Z
diff --git a/tests/chainz_test.py b/tests/chainz_test.py
@@ -15,12 +15,8 @@ def test_iter(self):
         it = iter(a)
         self.assertEqual(next(it), 0)
         self.assertEqual(next(it), 1)
-        try:
+        with self.assertRaises(StopIteration):
             next(it)
-        except StopIteration:
-            pass
-        else:
-            self.fail()
 
     def test_next(self):
         a = Chain(range(2))
