test_calculator.py

import unittest
from calculator import Calculator

class TestCalculator(unittest.TestCase):

    def setUp(self):
        self.calc = Calculator()

    def test_add(self):
        self.assertEqual(self.calc.add(1, 2), 3)
        self.assertEqual(self.calc.add(0, 0), 0)
        self.assertEqual(self.calc.add(-1, -1), -2)

    def test_subtract(self):
        self.assertEqual(self.calc.subtract(1, 2), 1)
        self.assertEqual(self.calc.subtract(5, 3), -2)
        self.assertEqual(self.calc.subtract(-1, -5), -4)

    def test_multiply(self):
        self.assertEqual(self.calc.multiply(3, 2), 6)
        self.assertEqual(self.calc.multiply(0, 4), 0)

    def test_divide(self):
        self.assertEqual(self.calc.divide(3, 1), 3)
        self.assertEqual(self.calc.divide(9, 3), 3)

    def test_modulo(self):
        self.assertEqual(self.calc.modulo(10, 3), 1)
        self.assertEqual(self.calc.modulo(9, 2), 1)


    # Add the following test methods to the TestCalculator class:

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