import calculator
import unittest


class TestCalculator(unittest.TestCase):

    def test_subtraction(self):
        result = calculator.subtract(5,2)
        self.assertEqual(result, 3)


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