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

index 1343649faafffab22f7a8a4228bd107fc2770761..f6239d6f51c7fefe27959884f5bfaa35995577a8 100644 (file)
@@ -278,8 +278,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