import unittest def add(x, y): return x + y class TestAdd(unittest.TestCase): def test_add_two_numbers(self): self.assertEqual(add(1, 2), 3) if __name__ == "__main__": unittest.main()