class MyTestCase(unittest.TestCase): def setUp(self): self.schedule = Schedule(mid=1) def test_func(self): res = self.schedule.get_available_times() assert(len(res) != 0) def tearDown(self): self.schedule.db_close()
def get_machine_schedule_dictionaries(self): mg = Machine("temp") machines = mg.get_all_machines() types = {"Treadmill":{},"Strider":{},"Ski":{}} for machine in machines: mid = machine[0] #print(mid) mtype = machine[2] #print(mtype) schedule = Schedule() times = schedule.get_available_times(mid) #print(times) if mtype == "Treadmill": types["Treadmill"][mid] = times if mtype == "Strider": types["Strider"][mid] = times if mtype == "Ski": types["Ski"][mid] = times schedule.db_close() mg.db_close() return types