def test_stamp_A_B_join(T_A, T_B): m = 3 left = utils.naive_stamp(T_A, m, T_B=T_B) right = stamp.stamp(T_A, T_B, m) utils.replace_inf(left) utils.replace_inf(right) npt.assert_almost_equal(left[:, :2], right)
def test_stamp_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 = np.array( [ utils.naive_mass(Q, T_A_sub, m) for Q in core.rolling_window(T_B_sub, m) ], dtype=object, ) right = stamp.stamp(T_A_sub, T_B_sub, m) utils.replace_inf(left) utils.replace_inf(right) npt.assert_almost_equal(left[:, :2], right)
def test_stamp_A_B_join(T_A, T_B): m = 3 ref_mp = naive.stamp(T_A, m, T_B=T_B) comp_mp = stamp.stamp(T_A, T_B, m) naive.replace_inf(ref_mp) naive.replace_inf(comp_mp) npt.assert_almost_equal(ref_mp[:, :2], comp_mp)
def test_stamp_A_B_join(T_A, T_B): m = 3 left = np.array([naive_mass(Q, T_A, m) for Q in core.rolling_window(T_B, m)], dtype=object) right = stamp.stamp(T_A, T_B, m) replace_inf(left) replace_inf(right) npt.assert_almost_equal(left, right)
def test_stamp_self_join(T_A, T_B): m = 3 zone = int(np.ceil(m / 2)) ref_mp = naive.stamp(T_B, m, exclusion_zone=zone) comp_mp = stamp.stamp(T_B, T_B, m, ignore_trivial=True) naive.replace_inf(ref_mp) naive.replace_inf(comp_mp) npt.assert_almost_equal(ref_mp[:, :2], comp_mp)
def test_stamp_self_join(T_A, T_B): m = 3 zone = int(np.ceil(m / 2)) left = utils.naive_stamp(T_B, m, exclusion_zone=zone) right = stamp.stamp(T_B, T_B, m, ignore_trivial=True) utils.replace_inf(left) utils.replace_inf(right) npt.assert_almost_equal(left[:, :2], right)
def test_stamp_self_join(T_A, T_B): m = 3 zone = int(np.ceil(m/2)) left = np.array([naive_mass(Q, T_B, m, i, zone) for i, Q in enumerate(core.rolling_window(T_B, m))], dtype=object) right = stamp.stamp(T_B, T_B, m, ignore_trivial=True) replace_inf(left) replace_inf(right) npt.assert_almost_equal(left, right)
def test_stamp_nan_zero_mean_self_join(): T = np.array([-1, 0, 1, np.inf, 1, 0, -1]) m = 3 zone = int(np.ceil(m / 2)) ref_mp = naive.stamp(T, m, exclusion_zone=zone) comp_mp = stamp.stamp(T, T, m, ignore_trivial=True) naive.replace_inf(ref_mp) naive.replace_inf(comp_mp) npt.assert_almost_equal(ref_mp[:, :2], comp_mp)
def test_stamp_nan_zero_mean_self_join(): T = np.array([-1, 0, 1, np.inf, 1, 0, -1]) m = 3 zone = int(np.ceil(m / 2)) left = utils.naive_stamp(T, m, exclusion_zone=zone) right = stamp.stamp(T, T, m, ignore_trivial=True) utils.replace_inf(left) utils.replace_inf(right) npt.assert_almost_equal(left[:, :2], right)
def test_stamp_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 / 2)) left = utils.naive_stamp(T_B_sub, m, exclusion_zone=zone) right = stamp.stamp(T_B_sub, T_B_sub, m, ignore_trivial=True) utils.replace_inf(left) utils.replace_inf(right) npt.assert_almost_equal(left[:, :2], right)
def test_stamp_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 / 2)) ref_mp = naive.stamp(T_B_sub, m, exclusion_zone=zone) comp_mp = stamp.stamp(T_B_sub, T_B_sub, m, ignore_trivial=True) naive.replace_inf(ref_mp) naive.replace_inf(comp_mp) npt.assert_almost_equal(ref_mp[:, :2], comp_mp)
def test_stamp_nan_zero_mean_self_join(): T = np.array([-1, 0, 1, np.inf, 1, 0, -1]) m = 3 zone = int(np.ceil(m / 2)) 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 = stamp.stamp(T, T, m, ignore_trivial=True) utils.replace_inf(left) utils.replace_inf(right) npt.assert_almost_equal(left[:, :2], right)
def test_stamp_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 ref_mp = naive.stamp(T_A_sub, m, T_B=T_B_sub) comp_mp = stamp.stamp(T_A_sub, T_B_sub, m) naive.replace_inf(ref_mp) naive.replace_inf(comp_mp) npt.assert_almost_equal(ref_mp[:, :2], comp_mp)
def test_stamp_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.stamp(T_A_sub, m, T_B=T_B_sub) right = stamp.stamp(T_A_sub, T_B_sub, m) naive.replace_inf(left) naive.replace_inf(right) npt.assert_almost_equal(left[:, :2], right)
def test_stamp_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 / 2)) left = np.array( [ utils.naive_mass(Q, T_B_sub, m, i, zone, ignore_trivial=True) for i, Q in enumerate(core.rolling_window(T_B_sub, m)) ], dtype=object, ) right = stamp.stamp(T_B_sub, T_B_sub, m, ignore_trivial=True) utils.replace_inf(left) utils.replace_inf(right) npt.assert_almost_equal(left[:, :2], right)