import unittest from increment import add_one class Tester(unittest.TestCase): def test_add_one(self): self.assertEqual(add_one(1), 2) self.assertEqual(add_one(-1), 0) self.assertEqual(add_one(0), 1) if __name__ == "__main__": unittest.main()