test_divide_bad.py

test_divide_bad.py

import unittest
import divide


class TestDivide(unittest.TestCase):

    def test_divide_pass_0(self):
        try:
            self.assertEqual(divide.divide(10, 5), 2)
        except:
            assert False

    def test_divide_pass_1(self):
        try:
            self.assertEqual(divide.divide(100, 10), 10)
        except:
            assert False

    def test_divide_unexpected_fail(self):
        """
        We know this will throw a Runtime Error. But imagine we didn't.
        """
        try:
            self.assertEqual(divide.divide(10, 0), 3)
        except:
            assert False

    def test_divide_expected_fail(self):
        """Correct way to test error scenarios"""
        try:
            self.assertRaises(ZeroDivisionError, divide.divide, 8, 0)
        except:
            assert False


if __name__ == "__main__":
    unittest.main()