def test_ssl01_scan_f(): # 4 haplotypes, h = np.array([[0, 0, 1, 1], [0, 0, 1, 1], [0, 0, 1, 1]]) expect0 = [1, 2, 3] expect1 = [1, 2, 3] actual0, actual1 = ssl01_scan(h, sum_ssl) assert_array_equal(expect0, actual0) assert_array_equal(expect1, actual1)
def test_ssl01_scan_d(): # 2 haplotypes, different h = np.array([[0, 1], [0, 1], [1, 0], [1, 0]]) expect0 = [0, 0, 0, 0] expect1 = [0, 0, 0, 0] actual0, actual1 = ssl01_scan(h, sum_ssl) assert_array_equal(expect0, actual0) assert_array_equal(expect1, actual1)
def test_ssl01_scan_e(): # 3 haplotypes, 3 pairs, identical h = np.array([[0, 0, 0], [0, 0, 0], [0, 0, 0]]) expect0 = [3, 6, 9] expect1 = [0, 0, 0] actual0, actual1 = ssl01_scan(h, sum_ssl) assert_array_equal(expect0, actual0) assert_array_equal(expect1, actual1)
def test_ssl01_scan_c(): # 2 haplotypes, identical h = np.array([[0, 0], [0, 0], [1, 1], [1, 1]]) expect0 = [1, 2, 0, 0] expect1 = [0, 0, 3, 4] actual0, actual1 = ssl01_scan(h, sum_ssl) assert_array_equal(expect0, actual0) assert_array_equal(expect1, actual1)