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)
