def test_indiv_snp_variation(self):
"""Does the function return correct output when given correct output"""
input = '10 89674917 rs182708158 T G 100 PASS ERATE=0.0004;LDAF=0.0024;AA=T;AN=2184;THETA=0.0008;VT=SNP;AVGPOST=0.9993;RSQ=0.9008;AC=5;SNPSOURCE=LOWCOV;AF=0.0023;ASN_AF=0.01 GT:DS:GL 0|0:0.000:-0.03,-1.22,-5.00 0|0:0.000:-0.01,-1.62,-5.00 0|0:0.000:-0.01,-1.87,-5.00 0|0:0.000:-0.18,-0.48,-2.19 0|0:0.000:-0.12,-0.62,-3.85 0|0:0.000:-0.14,-0.56,-3.22 0|0:0.000:-0.03,-1.15,-5.00 0|0:0.000:-0.01,-1.74,-5.00 0|0:0.000:-0.03,-1.19,-5.00 0|0:0.000:-0.05,-0.99,-5.00 0|0:0.000:-0.08,-0.75,-5.00 0|0:0.000:-0.02,-1.28,-5.00 0|0:0.000:-0.00,-2.07,-5.00 0|0:0.000:-0.01,-1.54,-5.00 0|0:0.000:-0.03,-1.15,-5.00 0|0:0.000:-0.03,-1.20,-5.00 0|0:0.000:-0.06,-0.91,-5.00 0|0:0.000:-0.00,-1.97,-5.00 0|0:0.000:-0.01,-1.51,-5.00'
expected = [[0, 0], [0, 0], [0, 0], [0, 0], [0, 0], [0, 0], [0, 0],
[0, 0], [0, 0], [0, 0], [0, 0], [0, 0], [0, 0], [0, 0],
[0, 0], [0, 0], [0, 0], [0, 0], [0, 0]]
self.assertEqual(indiv_snp_variation(input), expected)
def test_indiv_snp_variation_3(self):
"""Does the function return correct output when given correct output"""
input = '10 89623323 rs1044322 G A 100 PASS . GT:AP 1|0:0.015,0.000 0|0:0.000,0.000 0|0:0.002,0.000 1|1:0.000,0.052 0|0:0.000,0.000'
expected = [[1, 0], [0, 0], [0, 0], [1, 1], [0, 0]]
self.assertEqual(indiv_snp_variation(input), expected)
def test_indiv_snp_variation_3(self):
"""Does the function return correct output when given correct output"""
input = '10 89623323 rs1044322 G A 100 PASS . GT:AP 1|0:0.015,0.000 0|0:0.000,0.000 0|0:0.002,0.000 1|1:0.000,0.052 0|0:0.000,0.000'
expected = [[1, 0], [0, 0], [0, 0], [1, 1], [0, 0]]
self.assertEqual(indiv_snp_variation(input), expected)
def test_indiv_snp_variation(self):
"""Does the function return correct output when given correct output"""
input = '10 89674917 rs182708158 T G 100 PASS ERATE=0.0004;LDAF=0.0024;AA=T;AN=2184;THETA=0.0008;VT=SNP;AVGPOST=0.9993;RSQ=0.9008;AC=5;SNPSOURCE=LOWCOV;AF=0.0023;ASN_AF=0.01 GT:DS:GL 0|0:0.000:-0.03,-1.22,-5.00 0|0:0.000:-0.01,-1.62,-5.00 0|0:0.000:-0.01,-1.87,-5.00 0|0:0.000:-0.18,-0.48,-2.19 0|0:0.000:-0.12,-0.62,-3.85 0|0:0.000:-0.14,-0.56,-3.22 0|0:0.000:-0.03,-1.15,-5.00 0|0:0.000:-0.01,-1.74,-5.00 0|0:0.000:-0.03,-1.19,-5.00 0|0:0.000:-0.05,-0.99,-5.00 0|0:0.000:-0.08,-0.75,-5.00 0|0:0.000:-0.02,-1.28,-5.00 0|0:0.000:-0.00,-2.07,-5.00 0|0:0.000:-0.01,-1.54,-5.00 0|0:0.000:-0.03,-1.15,-5.00 0|0:0.000:-0.03,-1.20,-5.00 0|0:0.000:-0.06,-0.91,-5.00 0|0:0.000:-0.00,-1.97,-5.00 0|0:0.000:-0.01,-1.51,-5.00'
expected = [[0, 0], [0, 0], [0, 0], [0, 0], [0, 0],[0, 0], [0, 0], [0, 0], [0, 0], [0, 0], [0, 0], [0, 0], [0, 0], [0, 0],[0, 0], [0, 0], [0, 0], [0, 0], [0, 0]]
self.assertEqual(indiv_snp_variation(input), expected)
