def test_get_table_pathing_1(self, ShortestPath):
ShortestPath.return_value = self.table_path_1
for index in range(len(self.table_names_1)):
with self.subTest(table_index=index):
querying.get_table_pathing(self.graph, self.table_list_1)
ShortestPath.assert_any_call(self.graph, "center",
self.table_names_1[index])
ShortestPath.clear()
def test_get_table_pathing_1(self, shortest_path_mock):
"""Verify shortest_path() is called with correct parameters.
"""
shortest_path_mock.return_value = self.table_path_1
for index in range(len(self.table_names_1)):
with self.subTest(table_index=index):
querying.get_table_pathing(self.graph, self.table_list_1)
shortest_path_mock.assert_any_call(self.graph, "center",
self.table_names_1[index])
shortest_path_mock.clear()
def test_get_table_pathing_2(self, shortest_path_mock):
"""Verify center_table parameter influences shortest_path() parameters.
"""
shortest_path_mock.return_value = self.table_path_1
for index in range(len(self.table_names_2)):
with self.subTest(table_index=index):
querying.get_table_pathing(self.graph,
self.table_list_2,
center_table=self.center)
shortest_path_mock.assert_any_call(self.graph, "center",
self.table_names_2[index])
shortest_path_mock.clear()
def test_get_table_pathing_2(self):
"""Verify get_table_pathing() builds from distal center table.
"""
table_list = [self.gene, self.phage, self.trna]
table_pathing = querying.get_table_pathing(self.graph, table_list)
self.assertEqual(table_pathing[0], self.gene)
self.assertEqual(table_pathing[1][0], ["gene", "phage"])
self.assertEqual(table_pathing[1][1], ["gene", "phage", "trna"])
def test_get_table_pathing_3(self):
"""Verify get_table_pathing() correctly utilizes center_table input.
"""
table_list = [self.gene, self.phage, self.trna]
table_pathing = querying.get_table_pathing(self.graph,
table_list,
center_table=self.phage)
self.assertEqual(table_pathing[0], self.phage)
self.assertEqual(table_pathing[1][0], ["phage", "gene"])
self.assertEqual(table_pathing[1][1], ["phage", "trna"])