Commit 8f35c891 authored by Raymond Hettinger's avatar Raymond Hettinger

Modernize unittest example

parent 1621d77f
...@@ -281,8 +281,10 @@ file:: ...@@ -281,8 +281,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