def test_process_genotype_data_4(self):
"""Does this work when given correct input?"""
data = {}
genotypes = ['0|0:0.015,0.000', '1|2:0.000,0.000', '0|0:0.002,0.000', '0|0:0.000,0.052', '0|0:0.000,0.000']
observation_id_index = 1
with self.assertRaises(ValueError):
process_genotype_data(data, genotypes, observation_id_index)
def test_process_genotype_data_3(self):
"""Does this work when given correct input?"""
data = {(1, 1):1, (2, 2):2, (2,4):1, (2, 9):1}
genotypes = ['0|0:0.000:-0.03,-1.23,-5.00', '0|0:0.000:-0.02,-1.33,-5.00', '1|1:0.000:-0.02,-1.33,-5.00', '0|0:0.000:-0.03,-1.21,-5.00', '0|1:0.000:-0.06,-0.87,-5.00', '0|0:0.000:-0.48,-0.48,-0.48', '0|0:0.000:-0.10,-0.69,-4.70', '0|0:0.000:-0.03,-1.22,-5.00', '0|0:0.000:-0.01,-1.49,-5.00', '1|0:0.000:-0.00,-2.05,-5.00', '0|0:0.000:-0.06,-0.87,-5.00', '0|0:0.000:-0.06,-0.90,-5.00']
observation_id_index = 2
expected = {(1, 1):1, (2, 2):2, (2,4):1, (2, 9):1}
self.assertEqual(process_genotype_data(data, genotypes, observation_id_index), expected)
def test_process_genotype_data(self):
"""Does this work when given correct input?"""
data = {}
genotypes = ['0|0:0.015,0.000', '0|0:0.000,0.000', '0|0:0.002,0.000', '0|0:0.000,0.052', '0|0:0.000,0.000']
observation_id_index = 0
expected = {}
self.assertEqual(process_genotype_data(data, genotypes, observation_id_index), expected)
