]> granicus.if.org Git - python/commitdiff
Modernize unittest example
authorRaymond Hettinger <python@rcn.com>
Fri, 22 Mar 2013 14:26:18 +0000 (07:26 -0700)
committerRaymond Hettinger <python@rcn.com>
Fri, 22 Mar 2013 14:26:18 +0000 (07:26 -0700)
Doc/tutorial/stdlib.rst

index 1ebf792f75390274fdbf153b520d2f54335eb70e..7e7a1545eec59bb6df1524ab2238e2994dc03f62 100644 (file)
@@ -281,8 +281,10 @@ file::
        def test_average(self):
            self.assertEqual(average([20, 30, 70]), 40.0)
            self.assertEqual(round(average([1, 5, 7]), 1), 4.3)
-           self.assertRaises(ZeroDivisionError, average, [])
-           self.assertRaises(TypeError, average, 20, 30, 70)
+           with self.assertRaises(ZeroDivisionError):
+               average([])
+           with self.assertRaises(TypeError):
+               average(20, 30, 70)
 
    unittest.main() # Calling from the command line invokes all tests