import unittest class Calculator: def add(self, a, b): return a + b class TestCalculator(unittest.TestCase): def test_add(self): c = Calculator() self.assertEqual(c.add(2,3), 5) if __name__ == '__main__': unittest.main()