Function test_subspace_include_discords

tests/test_mstump.py:196–208

(T, m)

@pytest.mark.parametrize("T, m", test_data)
def test_subspace_include_discords(T, m):
    discord_idx = 1
    nn_idx = 4
    for width in range(T.shape[0]):
        for i in range(T.shape[0] - width):
            include = np.asarray(range(i, i + width + 1))

            for k in range(T.shape[0]):
                ref_S = naive.subspace(
                    T, m, discord_idx, nn_idx, k, include, discords=True
                )
                comp_S = subspace(T, m, discord_idx, nn_idx, k, include, discords=True)
                npt.assert_almost_equal(ref_S, comp_S)


@pytest.mark.parametrize("T, m", test_data)