def test_unsupported_genotype_likelihood(self): variantcall = variants_pb2.VariantCall(genotype_likelihood=[-1, -2, -3]) with self.assertRaisesRegexp(NotImplementedError, 'only supports haploid and diploid'): variant_utils.genotype_likelihood(variantcall, [0, 1, 1])
def test_genotype_likelihood(self, gls, allele_indices, expected): variantcall = variants_pb2.VariantCall(genotype_likelihood=gls) actual = variant_utils.genotype_likelihood(variantcall, allele_indices) self.assertEqual(actual, expected)