def testMakeExampleMultiAllelic(self):
alts = ['AA', 'CC', 'GG']
self.variant.alternate_bases[:] = alts
# Providing GG, AA checks that we're sorting the indices.
example = tf_utils.make_example(self.variant, ['GG', 'AA'], 'foo',
self.default_shape, self.default_format)
self.assertEqual([0, 2], tf_utils.example_alt_alleles_indices(example))
self.assertEqual(['AA', 'GG'], tf_utils.example_alt_alleles(example))
self.assertEqual('1:11:C->AA/GG', tf_utils.example_key(example))
def testMakeExample(self):
example = tf_utils.make_example(self.variant, self.alts, self.encoded_image,
self.default_shape, self.default_format)
self.assertEqual(self.encoded_image,
tf_utils.example_encoded_image(example))
self.assertEqual(
'raw', example.features.feature['image/format'].bytes_list.value[0])
self.assertEqual(self.variant, tf_utils.example_variant(example))
self.assertEqual('1:11-11', tf_utils.example_locus(example))
self.assertEqual([0], tf_utils.example_alt_alleles_indices(example))
self.assertEqual('1:11:C->A', tf_utils.example_key(example))
