import unittest
def div(a,b):
return a / b
class TestDiv(unittest.TestCase):
def test_1_div_1(self):
self.assertEqual(div(1,1),1/1)
def test_3_div_4(self):
self.assertEqual(div(3,4),3/4)
def test_3_div_0(self):
self.assertRaises(ZeroDivisionError,div,3,0)
if __name__ == '__main__':
unittest.main()