Modernize unittest example

45072741 · Raymond Hettinger · 28b7c051 · 45072741
Commit 45072741 authored Mar 22, 2013 by Raymond Hettinger
Hide whitespace changes
Inline Side-by-side

Showing with 4 additions and 2 deletions

Doc/tutorial/stdlib.rst Doc/tutorial/stdlib.rst +4 -2

No files found.
--- a/Doc/tutorial/stdlib.rst
+++ b/Doc/tutorial/stdlib.rst
@@ -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