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)
