def test_case_insensitive_get_taxa_by_label(self): tns = TaxonNamespace(self.str_labels) labels_upper = [label.upper() for label in self.str_labels if label.upper() != label] assert labels_upper # default: case-insensitive t2 = tns.get_taxa(labels_upper) self.assertEqual(len(t2), len(labels_upper)) for t, label in zip(t2, labels_upper): self.assertEqual(t.label.lower(), label.lower()) # test: case sensitive tns.is_case_sensitive = True t1 = tns.get_taxa(labels_upper) self.assertEqual(len(t1), 0)
def test_get_taxa_by_label(self): tns = TaxonNamespace(self.str_labels) # label_set = set(self.str_labels) # taxa = tns.get_taxa(label_set) taxa = tns.get_taxa(self.str_labels + ["u", "x", "y"]) self.assertEqual(len(taxa), len(self.str_labels)) tx = [t.label for t in taxa] self.assertEqual(tx, self.str_labels)
def test_get_nonexistant_taxa_by_label(self): tns = TaxonNamespace(self.str_labels) check = ["u", "x", "y",] taxa = tns.get_taxa(check) self.assertEqual(len(taxa), 0)