test_divide_good.py

test_divide_good.py

import unittest
import divide


class TestDivide(unittest.TestCase):

    def test_divide_pass_0(self):
        self.assertEqual(divide.divide(10, 5), 2)

    def test_divide_pass_1(self):
        self.assertEqual(divide.divide(100, 10), 10)

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

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


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