def test_givenNonMatchingFirstNucleotideAlleles_WhenformatAlleles_ThenReturnInput(
self):
basicRefAllele = "A"
basicAltAllele = "T"
row1 = {"ref_allele": basicRefAllele, "alt_allele": basicAltAllele}
refAllele = "ATC"
altAllele = "CTT"
row2 = {"alt_allele": altAllele, "ref_allele": refAllele}
actualRef1, actualAlt1 = AnnotationMerger.formatAlleles(row1)
actualRef2, actualAlt2 = AnnotationMerger.formatAlleles(row2)
self.assertEquals(actualRef1, basicRefAllele)
self.assertEquals(actualRef2, refAllele)
self.assertEquals(actualAlt1, basicAltAllele)
self.assertEquals(actualAlt2, altAllele)
def test_givenMatchingFirstNucleotideAlleles_WhenformatAlleles_ThenReturnFirstDroppedAlleles(
self):
refAllele = "NC"
altAllele = "NT"
row = {"alt_allele": altAllele, "ref_allele": refAllele}
refAllele2 = "NATC"
altAllele2 = "NCTT"
row2 = {"alt_allele": altAllele2, "ref_allele": refAllele2}
expectedRefAllele1 = "C"
expectedAltAllele1 = "T"
expectedRefAllele2 = "ATC"
expectedAltAllele2 = "CTT"
actualRef1, actualAlt1 = AnnotationMerger.formatAlleles(row)
actualRef2, actualAlt2 = AnnotationMerger.formatAlleles(row2)
self.assertEquals(actualRef1, expectedRefAllele1)
self.assertEquals(actualRef2, expectedRefAllele2)
self.assertEquals(actualAlt1, expectedAltAllele1)
self.assertEquals(actualAlt2, expectedAltAllele2)