test raises

Author: pekkaklarck

sut.py

@@ -5,3 +5,7 @@ def compare(x, y):
         return 1
     else:
         return -1
+
+
+def divide(x, y):
+    return x / y

test_unittest.py

@@ -1,6 +1,6 @@
 import unittest
 
-from sut import compare
+from sut import compare, divide
 
 
 class TestCompare(unittest.TestCase):
@@ -20,6 +20,9 @@ def test_x_less_than_y_returns_minus_1(self):
     def test_failing(self):
         self.assertEquals(compare(1, 0), 0)
 
+    def test_raises(self):
+        self.assertRaises(ZeroDivisionError, divide, 1, 2)
+
 
 if __name__ == '__main__':
     unittest.main()