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)