import unittest from time_interval import to_minute class TimeIntervalTestCase(unittest.TestCase): def test_to_minute(self): time = "9:20" minutes = to_minute(time) expected = 9*60 + 20 self.assertEqual(minutes, expected) def test_to_minute2(self): time = "9" minutes = to_minute(time) expected = 9*60 self.assertEqual(minutes, expected) if __name__ == '__main__': unittest.main()