def test_simple_linear_structure(self): [grandparent_id, parent_id, child_id] = all_cases = ['grandparent', 'parent', 'child'] tree = IndexTree(indices={ child_id: convert_list_to_dict([parent_id]), parent_id: convert_list_to_dict([grandparent_id]), }) cases = tree.get_all_cases_that_depend_on_case(grandparent_id) self.assertEqual(cases, set(all_cases))
def test_multiple_children(self): [grandparent_id, parent_id, child_id_1, child_id_2] = all_cases = ['rickard', 'ned', 'bran', 'arya'] tree = IndexTree(indices={ child_id_1: convert_list_to_dict([parent_id]), child_id_2: convert_list_to_dict([parent_id]), parent_id: convert_list_to_dict([grandparent_id]), }) cases = tree.get_all_cases_that_depend_on_case(grandparent_id) self.assertEqual(cases, set(all_cases))
def test_simple_linear_structure(self): [grandparent_id, parent_id, child_id] = all_cases = ['grandparent', 'parent', 'child'] tree = IndexTree( indices={ child_id: convert_list_to_dict([parent_id]), parent_id: convert_list_to_dict([grandparent_id]), }) cases = tree.get_all_cases_that_depend_on_case(grandparent_id) self.assertEqual(cases, set(all_cases))
def test_multiple_children(self): [grandparent_id, parent_id, child_id_1, child_id_2] = all_cases = ['rickard', 'ned', 'bran', 'arya'] tree = IndexTree( indices={ child_id_1: convert_list_to_dict([parent_id]), child_id_2: convert_list_to_dict([parent_id]), parent_id: convert_list_to_dict([grandparent_id]), }) cases = tree.get_all_cases_that_depend_on_case(grandparent_id) self.assertEqual(cases, set(all_cases))