def test_walk_inputs_is_working_properly_in_DFS_mode(self): """testing if the walk_inputs() method is working in DFS mode correctly """ v1 = Version(task=self.test_task1) v2 = Version(task=self.test_task1) v3 = Version(task=self.test_task1) v4 = Version(task=self.test_task1) v5 = Version(task=self.test_task1) v5.inputs = [v4] v4.inputs = [v3, v2] v3.inputs = [v1] v2.inputs = [v1] expected_result = [v5, v4, v3, v1, v2, v1] visited_versions = [] for v in v5.walk_inputs(): visited_versions.append(v) self.assertEqual(expected_result, visited_versions)
def test_inputs_attribute_is_working_properly(self): """testing if the inputs attribute is working properly """ self.kwargs.pop("inputs") new_version = Version(**self.kwargs) self.assertFalse(self.test_input_link1 in new_version.inputs) self.assertFalse(self.test_input_link2 in new_version.inputs) new_version.inputs = [self.test_input_link1, self.test_input_link2] self.assertTrue(self.test_input_link1 in new_version.inputs) self.assertTrue(self.test_input_link2 in new_version.inputs)