def test_aamp_nan_inf_A_B_join(T_A, T_B, substitute_A, substitute_B,
substitution_locations):
m = 3
T_A_sub = T_A.copy()
T_B_sub = T_B.copy()
for substitution_location_B in substitution_locations:
for substitution_location_A in substitution_locations:
T_A_sub[:] = T_A[:]
T_B_sub[:] = T_B[:]
T_A_sub[substitution_location_A] = substitute_A
T_B_sub[substitution_location_B] = substitute_B
left = naive.aamp(T_A_sub, m, T_B=T_B_sub)
right = aamp(T_A_sub, m, T_B_sub, ignore_trivial=False)
naive.replace_inf(left)
naive.replace_inf(right)
npt.assert_almost_equal(left, right)
right = aamp(pd.Series(T_A_sub),
m,
pd.Series(T_B_sub),
ignore_trivial=False)
naive.replace_inf(right)
npt.assert_almost_equal(left, right)
def test_aamp_two_constant_subsequences_A_B_join():
T_A = np.concatenate(
(np.zeros(10, dtype=np.float64), np.ones(10, dtype=np.float64)))
T_B = np.concatenate(
(np.zeros(20, dtype=np.float64), np.ones(5, dtype=np.float64)))
m = 3
left = naive.aamp(T_A, m, T_B=T_B)
right = aamp(T_A, m, T_B, ignore_trivial=False)
naive.replace_inf(left)
naive.replace_inf(right)
npt.assert_almost_equal(left[:, 0], right[:, 0]) # ignore indices
right = aamp(pd.Series(T_A), m, pd.Series(T_B), ignore_trivial=False)
naive.replace_inf(right)
npt.assert_almost_equal(left[:, 0], right[:, 0]) # ignore indices
# Swap inputs
left = naive.aamp(T_B, m, T_B=T_A)
right = aamp(T_B, m, T_A, ignore_trivial=False)
naive.replace_inf(left)
naive.replace_inf(right)
npt.assert_almost_equal(left[:, 0], right[:, 0]) # ignore indices
right = aamp(pd.Series(T_B), m, pd.Series(T_A), ignore_trivial=False)
naive.replace_inf(right)
npt.assert_almost_equal(left[:, 0], right[:, 0]) # ignore indices
def test_aamp_identical_subsequence_A_B_join():
identical = np.random.rand(8)
T_A = np.random.rand(20)
T_B = np.random.rand(20)
T_A[1:1 + identical.shape[0]] = identical
T_B[11:11 + identical.shape[0]] = identical
m = 3
left = naive.aamp(T_A, m, T_B=T_B)
right = aamp(T_A, m, T_B, ignore_trivial=False)
naive.replace_inf(left)
naive.replace_inf(right)
npt.assert_almost_equal(left[:, 0], right[:, 0],
config.STUMPY_TEST_PRECISION) # ignore indices
right = aamp(pd.Series(T_A), m, pd.Series(T_B), ignore_trivial=False)
naive.replace_inf(right)
npt.assert_almost_equal(left[:, 0], right[:, 0],
config.STUMPY_TEST_PRECISION) # ignore indices
# Swap inputs
left = naive.aamp(T_B, m, T_B=T_A)
right = aamp(T_B, m, T_A, ignore_trivial=False)
naive.replace_inf(left)
naive.replace_inf(right)
npt.assert_almost_equal(left[:, 0], right[:, 0],
config.STUMPY_TEST_PRECISION) # ignore indices
def test_aamp_A_B_join(T_A, T_B):
m = 3
left = naive.aamp(T_A, m, T_B=T_B)
right = aamp(T_A, m, T_B, ignore_trivial=False)
naive.replace_inf(left)
naive.replace_inf(right)
npt.assert_almost_equal(left, right)
right = aamp(pd.Series(T_A), m, pd.Series(T_B), ignore_trivial=False)
naive.replace_inf(right)
npt.assert_almost_equal(left, right)
def test_aamp_self_join(T_A, T_B):
m = 3
zone = int(np.ceil(m / 4))
left = naive.aamp(T_B, m)
right = aamp(T_B, m)
naive.replace_inf(left)
naive.replace_inf(right)
npt.assert_almost_equal(left, right)
right = aamp(pd.Series(T_B), m)
naive.replace_inf(right)
npt.assert_almost_equal(left, right)
def test_aamp_constant_subsequence_self_join():
T_A = np.concatenate((np.zeros(20, dtype=np.float64), np.ones(5, dtype=np.float64)))
m = 3
left = naive.aamp(T_A, m)
right = aamp(T_A, m, ignore_trivial=True)
naive.replace_inf(left)
naive.replace_inf(right)
npt.assert_almost_equal(left[:, 0], right[:, 0]) # ignore indices
right = aamp(pd.Series(T_A), m, ignore_trivial=True)
naive.replace_inf(right)
npt.assert_almost_equal(left[:, 0], right[:, 0]) # ignore indices
def test_aamp_identical_subsequence_self_join():
identical = np.random.rand(8)
T_A = np.random.rand(20)
T_A[1:1 + identical.shape[0]] = identical
T_A[11:11 + identical.shape[0]] = identical
m = 3
left = naive.aamp(T_A, m)
right = aamp(T_A, m, ignore_trivial=True)
naive.replace_inf(left)
naive.replace_inf(right)
npt.assert_almost_equal(left[:, 0], right[:, 0],
decimal=naive.PRECISION) # ignore indices
right = aamp(pd.Series(T_A), m, ignore_trivial=True)
naive.replace_inf(right)
npt.assert_almost_equal(left[:, 0], right[:, 0],
decimal=naive.PRECISION) # ignore indices
def test_aamp_nan_inf_self_join(T_A, T_B, substitute_B, substitution_locations):
m = 3
T_B_sub = T_B.copy()
for substitution_location_B in substitution_locations:
T_B_sub[:] = T_B[:]
T_B_sub[substitution_location_B] = substitute_B
zone = int(np.ceil(m / 4))
left = naive.aamp(T_B_sub, m)
right = aamp(T_B_sub, m, ignore_trivial=True)
naive.replace_inf(left)
naive.replace_inf(right)
npt.assert_almost_equal(left, right)
right = aamp(pd.Series(T_B_sub), m, ignore_trivial=True)
naive.replace_inf(right)
npt.assert_almost_equal(left, right)
def test_aamp_nan_zero_mean_self_join():
T = np.array([-1, 0, 1, np.inf, 1, 0, -1])
m = 3
zone = int(np.ceil(m / 4))
left = naive.aamp(T, m)
right = aamp(T, m, ignore_trivial=True)
naive.replace_inf(left)
naive.replace_inf(right)
npt.assert_almost_equal(left, right)
