def test_nsl01_scan_c():
h = np.array([[0, 0, 1], [0, 1, 1], [1, 1, 0], [1, 0, 0]])
nsl0, nsl1 = nsl01_scan(h)
expect_nsl0 = [1, np.nan, np.nan, 1]
assert_array_almost_equal(expect_nsl0, nsl0)
expect_nsl1 = [np.nan, 1, 1, np.nan]
assert_array_almost_equal(expect_nsl1, nsl1)
def test_nsl01_scan_a():
h = np.array([[0, 0, 0, 1, 1, 1], [0, 0, 0, 1, 1, 1], [0, 0, 0, 1, 1, 1],
[0, 0, 0, 1, 1, 1]])
nsl0, nsl1 = nsl01_scan(h)
expect_nsl0 = [1, 2, 3, 4]
assert_array_nanclose(expect_nsl0, nsl0)
expect_nsl1 = [1, 2, 3, 4]
assert_array_nanclose(expect_nsl1, nsl1)
