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)