def test_find_xputs(self): a = self.a b = self.b c = self.c d = self.d e = self.e desired_output = { 1:{'originated_by':set([a]), 'modified_by':set(), 'finalized_by':set([b,d]) }, 2:{'originated_by':set(), 'modified_by':set([b]), 'finalized_by':set([c]) }, 3:{'originated_by':set([b]), 'modified_by':set([c]), 'finalized_by':set([e]) }, 4:{'originated_by':set([c]), 'modified_by':set(), 'finalized_by':set([e]) }, 5:{'originated_by':set([d]), 'modified_by':set(), 'finalized_by':set([e]) }} output = scheduler.find_xputs(self.operations) for key, value in desired_output.items(): self.assertEqual(output[key], value) self.assertEqual(output, desired_output)
def test_find_dependencies(self): a = self.a b = self.b c = self.c d = self.d e = self.e desired_output = set([(a, b), (a, d), (b, c), (b, c), (c, e), (d, e)]) xputs = scheduler.find_xputs(self.operations) output = scheduler.find_dependencies(xputs) self.assertEqual(output, desired_output)