def test_invalid_allele_indices_with_num_alts(self, alt_bases, num_alts, ploidy): variant = variants_pb2.Variant(alternate_bases=alt_bases) with self.assertRaises((NotImplementedError, ValueError)): variant_utils.allele_indices_with_num_alts(variant, num_alts, ploidy)
def test_allele_indices_with_num_alts(self, alt_bases, num_alts, expected): variant = variants_pb2.Variant(alternate_bases=alt_bases) actual = variant_utils.allele_indices_with_num_alts( variant, num_alts, ploidy=2) self.assertEqual(actual, expected)