def test_point_operations(self): scheduler.point_operations(self.operations) self.assertEqual(self.a.points_at, set([self.b,self.d])) self.assertEqual(self.b.points_at, set([self.c])) self.assertEqual(self.c.points_at, set([self.e])) self.assertEqual(self.d.points_at, set([self.e])) self.assertEqual(self.e.points_at, set())
def test_find_ready_lists(self): desired_output = [list(self.operations)] output = scheduler.find_ready_sets(self.operations) for i in range(len(desired_output)): d = desired_output[i] o = output[i] self.assertEqual(o, set(d)) scheduler.point_operations(self.operations) desired_output = [[self.a], [self.b, self.d], [self.c], [self.e]] output = scheduler.find_ready_sets(self.operations) self.assertEqual(len(output), len(desired_output)) for i in range(len(desired_output)): d = desired_output[i] o = output[i] self.assertEqual(o, set(d))