def test_get_table_list_2(self, extract_column_mock):
"""Verify extract_column() is called with the correct parameters.
extract_column() should take Column objects as well as lists.
"""
extract_column_mock.return_value = self.PhageID
querying.get_table_list(self.PhageID)
extract_column_mock.assert_called_with(self.PhageID)
def test_get_table_list_4(self):
"""Verify get_table_list() retrieves specified Table objects.
"""
columns = [self.PhageID, self.GeneID]
table_list = querying.get_table_list(columns)
self.assertTrue(self.phage in table_list)
self.assertTrue(self.gene in table_list)
def test_get_table_list_3(self):
"""Verify get_table_list() retrieves specified Table objects.
get_table_list() should exclude unspecified Tables.
"""
table_list = querying.get_table_list(self.PhageID)
self.assertTrue(self.phage in table_list)
self.assertFalse(self.gene in table_list)
def test_get_table_list_1(self):
"""Verify get_table_list() returns correct ordered Tables.
"""
columns = [self.GeneID, self.PhageID, self.TrnaID]
table_list = querying.get_table_list(columns)
self.assertEqual(table_list[0], self.gene)
self.assertEqual(table_list[1], self.phage)
self.assertEqual(table_list[2], self.trna)
def test_get_table_list_3(self):
"""Verify get_table_list() does not return table duplicates.
"""
table_list = querying.get_table_list([self.PhageID, self.Cluster])
self.assertTrue(len(table_list) == 1)
def test_get_table_list_2(self):
"""Verify get_table_list() accepts Column input.
"""
table_list = querying.get_table_list(self.AminoAcid)
self.assertEqual(table_list[0], self.trna)
def test_get_table_list_4(self):
columns = [self.PhageID, self.GeneID]
table_list = querying.get_table_list(columns)
self.assertTrue(self.phage in table_list)
self.assertTrue(self.gene in table_list)
def test_get_table_list_3(self):
table_list = querying.get_table_list(self.PhageID)
self.assertTrue(self.phage in table_list)
self.assertFalse(self.gene in table_list)
def test_get_table_list_2(self, ExtractColumn):
ExtractColumn.return_value = self.PhageID
querying.get_table_list(self.PhageID)
ExtractColumn.assert_called_with(self.PhageID)