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))
