def test_deterministic_process_event_schedule(self): action = ['read_email', 'watch_video'] schedule = fnss.deterministic_process_event_schedule(20, 0, 80001, 'ms', self.event_gen, 0.5, action=action) self.assertIsNotNone(schedule) self.assertEqual(4000, len(schedule)) for time, event in schedule: self.assertTrue(event['action'] in action) self.assertTrue(time >= 0) self.assertTrue(time <= 80001)
def test_read_write_event_schedule(self): action = ['read_email', 'watch_video'] schedule = fnss.deterministic_process_event_schedule(20, 0, 801, 'ms', self.event_gen, 0.5, action=action) time, event = schedule[2] tmp_es_file = path.join(TMP_DIR, 'event-schedule.xml') fnss.write_event_schedule(schedule, tmp_es_file) read_schedule = fnss.read_event_schedule(tmp_es_file) self.assertEqual(len(schedule), len(read_schedule)) read_time, read_event = read_schedule[2] self.assertEqual(time, read_time) self.assertEqual(event, read_event)