def test_scrimp_self_join(T):
m = 3
zone = int(np.ceil(m / 4))
left = np.array(
[
utils.naive_mass(Q, T, m, i, zone, True)
for i, Q in enumerate(core.rolling_window(T, m))
],
dtype=object,
)
right = scrimp(T, m, percentage=1.0)
utils.replace_inf(left)
utils.replace_inf(right)
npt.assert_almost_equal(left[:, 0], right[:, 0])
right = scrimp(pd.Series(T), m, percentage=1.0)
utils.replace_inf(right)
npt.assert_almost_equal(left[:, 0], right[:, 0])
def test_scrimp_self_join_larger_window(T):
for m in [8, 16, 32]:
if len(T) > m:
zone = int(np.ceil(m / 4))
left = np.array(
[
utils.naive_mass(Q, T, m, i, zone, True)
for i, Q in enumerate(core.rolling_window(T, m))
],
dtype=object,
)
right = scrimp(T, m)
utils.replace_inf(left)
utils.replace_inf(right)
npt.assert_almost_equal(left[:, 0], right[:, 0])
right = scrimp(pd.Series(T), m)
utils.replace_inf(right)
npt.assert_almost_equal(left[:, 0], right[:, 0])
def test_constant_subsequence_self_join():
T = np.concatenate((np.zeros(20,
dtype=np.float64), np.ones(5,
dtype=np.float64)))
m = 3
zone = int(np.ceil(m / 4))
left = np.array(
[
utils.naive_mass(Q, T, m, i, zone, True)
for i, Q in enumerate(core.rolling_window(T, m))
],
dtype=object,
)
right = scrimp(T, m, percentage=1.0)
utils.replace_inf(left)
utils.replace_inf(right)
npt.assert_almost_equal(left[:, 0], right[:, 0]) # ignore indices
right = scrimp(pd.Series(T), m, percentage=1.0)
utils.replace_inf(right)
npt.assert_almost_equal(left[:, 0], right[:, 0]) # ignore indices
def test_scrimp_self_join(T):
m = 3
left = np.zeros(T.shape[0])
right = scrimp(T, m, percentage=0.0)
utils.replace_inf(right)
npt.assert_almost_equal(left, right[:, 0])