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))