def test_should_format_multiple_values(self):
schema = Schema()
schema.set_info_data('K', 'A', 'Float', 'K')
schema.set_info_data('K2', 'A', 'String', 'K')
schema.set_info_data('K3', '0', 'Flag', 'K')
schema.set_info_data('K4', 'A', 'String', 'K')
info_data = InfoData(schema, {
'K3': None,
'K2': ['S2'],
'K': [1.0, 2.66, 3.0],
'K4': ['S4']
})
self.assertEqual('K=1.0,2.66,3.0;K2=S2;K3;K4=S4', info_data.to_vcf())
def test_should_format_a_float_list(self):
schema = Schema()
schema.set_info_data('K', 'A', 'Integer', 'K')
info_data = InfoData(schema, {'K': [1.0, 2.66, 3.0]})
self.assertEqual('K=1.0,2.66,3.0', info_data.to_vcf())
def test_should_format_an_int_list(self):
schema = Schema()
schema.set_info_data('K', 'A', 'Integer', 'K')
info_data = InfoData(schema, {'K': [1, 2, 3]})
self.assertEqual('K=1,2,3', info_data.to_vcf())
def test_should_format_a_string_list(self):
schema = Schema()
schema.set_info_data('K', 'A', 'String', 'K')
info_data = InfoData(schema, {'K': ['V1', 'V2']})
self.assertEqual('K=V1,V2', info_data.to_vcf())
def test_should_format_a_string(self):
info_data = InfoData(None, {'K': 'V'})
self.assertEqual('K=V', info_data.to_vcf())
def test_should_format_no_data(self):
info_data = InfoData(None, {})
self.assertEqual('.', info_data.to_vcf())
def test_should_format_a_present_flag(self):
schema = Schema()
schema.set_info_data('F', '0', 'Flag', 'Flag')
info_data = InfoData(schema, {"F": None})
self.assertEqual('F', info_data.to_vcf())