Commit 45072741 authored by Raymond Hettinger's avatar Raymond Hettinger

Modernize unittest example

parent 28b7c051
...@@ -278,8 +278,10 @@ file:: ...@@ -278,8 +278,10 @@ file::
def test_average(self): def test_average(self):
self.assertEqual(average([20, 30, 70]), 40.0) self.assertEqual(average([20, 30, 70]), 40.0)
self.assertEqual(round(average([1, 5, 7]), 1), 4.3) self.assertEqual(round(average([1, 5, 7]), 1), 4.3)
self.assertRaises(ZeroDivisionError, average, []) with self.assertRaises(ZeroDivisionError):
self.assertRaises(TypeError, average, 20, 30, 70) average([])
with self.assertRaises(TypeError):
average(20, 30, 70)
unittest.main() # Calling from the command line invokes all tests unittest.main() # Calling from the command line invokes all tests
......
Markdown is supported
0%
or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment