def test_get_gene_list_nested_and_or(self):
genes = '(gene_a or gene_d) or (gene_b and gene_c)'
self.assertEqual(psammotate.get_gene_list(genes),
frozenset(['gene_a', 'gene_b', 'gene_c', 'gene_d']))
def test_get_gene_list_multiple_nests(self):
genes = '(gene_a or (gene_b and gene_c)) or gene_d'
self.assertEqual(psammotate.get_gene_list(genes),
frozenset(['gene_a', 'gene_b', 'gene_c', 'gene_d']))
def test_get_gene_list_or(self):
genes = 'gene_a or gene_b'
self.assertEqual(psammotate.get_gene_list(genes),
frozenset(['gene_a', 'gene_b']))