def setUp(self):
# logging.basicConfig(level=logging.DEBUG)
self.haplo_diff = HaploBuilder("test_haplos.txt")
self.all_haplo_diff = HaploBuilder("HLAB.txt")
class TestHaploBuilder(unittest.TestCase):
def setUp(self):
# logging.basicConfig(level=logging.DEBUG)
self.haplo_diff = HaploBuilder("test_haplos.txt")
self.all_haplo_diff = HaploBuilder("HLAB.txt")
def test_filter_haplos_include(self):
filtered = self.haplo_diff.filter_haplos(["B*51240301"], [{"snp": "0:A"}])
self.assertEqual(1, len(filtered))
filtered = self.haplo_diff.filter_haplos(self.haplo_diff.haplos.keys(), [{"snp": "0:A"}])
self.assertEqual(2, len(filtered))
def test_filter_haplos_2_include(self):
filtered = self.haplo_diff.filter_haplos(self.haplo_diff.haplos.keys(), [{"snp": "0:A"}, {"snp": "2:C"}])
self.assertEqual(1, len(filtered))
def test_filter_haplos(self):
filtered = self.haplo_diff.filter_haplos(["B*51240301"], [{"snp": "0:A"}])
self.assertEqual(1, len(filtered))
filtered = self.haplo_diff.filter_haplos(self.haplo_diff.haplos.keys(), [{"snp": "0:A"}])
self.assertEqual(2, len(filtered))
filtered = self.haplo_diff.filter_haplos(self.haplo_diff.haplos.keys(), [{"snp": "11:G"}])
self.assertEqual(2, len(filtered))
def test_filter_haplos_2_exclude(self):
filtered = self.haplo_diff.filter_haplos(self.haplo_diff.haplos.keys(), [{"snp": "0:A"}, {"snp": "2:C"}], False)
self.assertEqual(4, len(filtered))
def test_check_1_hap(self):
important_haplos = ["B*55070101"]
unimportant_haplos = [hap for hap in self.haplo_diff.haplos.keys() if hap not in important_haplos]
checked = self.haplo_diff.check_haplos(important_haplos, unimportant_haplos, [])
self.assertEqual(2, len(checked[0]["snps"]))
def test_check_2_haplos(self):
important_haplos = ["B*55070101", "B*07420101"]
unimportant_haplos = [hap for hap in self.haplo_diff.haplos.keys() if hap not in important_haplos]
checked = self.haplo_diff.check_haplos(important_haplos, unimportant_haplos, [])
self.assertEqual(2, len(checked))
def test_check_4_haplos(self):
important_haplos = ["B*51240301", "B*55070101", "B*07420101", "B*27090101"]
unimportant_haplos = [hap for hap in self.haplo_diff.haplos.keys() if hap not in important_haplos]
checked = self.haplo_diff.check_haplos(important_haplos, unimportant_haplos, [])
self.assertEqual(4, len(checked))
def test_all_check_1(self):
important_haplos = ["B*55070101"]
# important_haplos = ['B*51240301']
unimportant_haplos = [hap for hap in self.all_haplo_diff.haplos.keys() if hap not in important_haplos]
checked = self.all_haplo_diff.check_haplos(important_haplos, unimportant_haplos, [])
self.assertEqual(1, len(checked))
def test_all_check_4(self):
important_haplos = ["B*51240301", "B*55070101", "B*07420101", "B*27090101"]
unimportant_haplos = [hap for hap in self.all_haplo_diff.haplos.keys() if hap not in important_haplos]
checked = self.all_haplo_diff.check_haplos(important_haplos, unimportant_haplos, [])
self.assertEqual(4, len(checked))
