def test_get_all_allele_indices_configurations(self, num_alts_list, config,
expected):
def get_alt_bases(num_alts):
return ['C' + 'A' * i for i in range(1, num_alts + 1)]
variants = [
_var(ref='C', alt=get_alt_bases(num_alts)) for num_alts in num_alts_list
]
actual = haplotypes._get_all_allele_indices_configurations(variants, config)
self.assertEqual(list(actual), expected)
def test_get_all_allele_indices_configurations(self, num_alts_list, config,
expected):
def get_alt_bases(num_alts):
return ['C' + 'A' * i for i in xrange(1, num_alts + 1)]
variants = [
_var(ref='C', alt=get_alt_bases(num_alts)) for num_alts in num_alts_list
]
actual = haplotypes._get_all_allele_indices_configurations(variants, config)
self.assertEqual(list(actual), expected)
def test_invalid_get_all_allele_indices_configurations(self):
with six.assertRaisesRegex(self, ValueError, r'len\(variants\) must equal'):
haplotypes._get_all_allele_indices_configurations(
_resolved_compatible_outputs(), [1])
def test_invalid_get_all_allele_indices_configurations(self):
with self.assertRaisesRegexp(ValueError, r'len\(variants\) must equal'):
haplotypes._get_all_allele_indices_configurations(
_resolved_compatible_outputs(), [1])
