def test_count_unknown_base_raises_exception(self):
snp = SNP(chromosome='chr1', position=1000, genotype='AA', counts={'A': 95, 'C': 1, 'G': 3, 'T': 1})
with self.assertRaises(UnknownBaseError):
snp.count('N')
def test_count(self):
snp = SNP(chromosome='chr19', position=230, genotype='AC', counts={'A': 30, 'C': 22, 'G': 7, 'T': 9})
self.assertEqual(snp.count('A'), 30)
self.assertEqual(snp.count('C'), 22)
self.assertEqual(snp.count('G'), 7)
self.assertEqual(snp.count('T'), 9)