import unittest from src.ex2_tv import startRecordingAt class MyTestCase(unittest.TestCase): def test_simple1(self): self.assertEqual( startRecordingAt([12, 22, 32, 42, 25, 12], 52), 2 ) def test_simple2(self): self.assertEqual( startRecordingAt([12, 22, 32, 42, 25, 12], 80), 3 ) if __name__ == '__main__': unittest.main()