diff --git a/libs/utils/test.py b/libs/utils/test.py index d1eb9c6ace1eec37436892350d7911c31113f74b..3edbed6db5a89f0ccd8a0878a0e221d39205201b 100644 --- a/libs/utils/test.py +++ b/libs/utils/test.py @@ -79,6 +79,13 @@ class LisaTest(unittest.TestCase): Code executed after running the experiments """ + @memoized + def get_sched_assert(self, experiment, task): + """ + Return a SchedAssert over the task provided + """ + return SchedAssert(experiment.out_dir, self.te.topology, execname=task) + @memoized def get_multi_assert(self, experiment, task_filter=""): """ @@ -99,6 +106,16 @@ class LisaTest(unittest.TestCase): start_times_dict = self.get_multi_assert(experiment).getStartTime() return min([t["starttime"] for t in start_times_dict.itervalues()]) + def get_end_time(self, experiment): + """ + Get the time at which the experiment workload finished executing + """ + end_times_dict = self.get_multi_assert(experiment).getEndTime() + return max([t["endtime"] for t in end_times_dict.itervalues()]) + + def get_window(self, experiment): + return (self.get_start_time(experiment), self.get_end_time(experiment)) + def get_end_times(self, experiment): """ Get the time at which each task in the workload finished