def test_contains(self): system_foo = System(self.cfg, "foo", 10) gene_in = Gene(self.cfg, 'sctJ_FLG', system_foo, self.profile_registry) gene_bank.add_gene(gene_in) self.assertIn(gene_in, gene_bank) gene_out = Gene(self.cfg, 'abc', system_foo, self.profile_registry) self.assertNotIn( gene_out, gene_bank)
def test_get_uniq_object(self): system_foo = System(self.cfg, "foo", 10) gene_in = Gene(self.cfg, 'sctJ_FLG', system_foo, self.profile_registry) gene_bank.add_gene(gene_in) gene1 = gene_bank['sctJ_FLG'] gene2 = gene_bank['sctJ_FLG'] self.assertEqual(gene1, gene2) self.assertIs(gene1, gene2)
def test_add_get_gene(self): gene_name = 'sctJ_FLG' self.assertRaises(KeyError, gene_bank.__getitem__, gene_name) system_foo = System( "foo", self.cfg, 10) gene = Gene(self.cfg, gene_name, system_foo, self.profile_registry) gene_bank.add_gene(gene) gene_from_bank = gene_bank[gene_name] self.assertTrue(isinstance(gene_from_bank, Gene)) self.assertEqual(gene_from_bank, gene)
def test_iter(self): system_foo = System(self.cfg, "foo", 10) genes = [Gene(self.cfg, 'sctJ_FLG', system_foo, self.profile_registry), Gene(self.cfg, 'abc', system_foo, self.profile_registry)] for g in genes: gene_bank.add_gene(g) i = 0 for g in gene_bank: self.assertIn(g, genes) i = i + 1 self.assertEqual(i, len(genes))