def test_no_perturbation_1d():
"""Test with uniform data in 1D."""
observations = 5
# given all the values are the same, there should not be perturbations
ts = np.ones(observations)
pert_zero = 0
assert_array_equal(pert_zero, get_perturbation(ts))
def test_no_perturbation_1d():
"""Test with uniform data in 1D."""
observations = 5
# given all the values are the same, there should not be perturbations
ts = np.ones(observations)
pert_zero = 0
assert_array_equal(pert_zero, get_perturbation(ts))
def test_no_perturbation_2d_axis_first():
"""Test with uniform data in 2D along the first axis."""
observations = 5
instruments = 2
# given all the values are the same, there should not be perturbations
ts = np.ones((observations, instruments))
pert_zero = np.zeros((observations, instruments))
assert_array_equal(pert_zero, get_perturbation(ts, axis=0))
def test_no_perturbation_2d_axis_first():
"""Test with uniform data in 2D along the first axis."""
observations = 5
instruments = 2
# given all the values are the same, there should not be perturbations
ts = np.ones((observations, instruments))
pert_zero = np.zeros((observations, instruments))
assert_array_equal(pert_zero, get_perturbation(ts, axis=0))
def test_known_perturbation_nonzero_mean_2d_axis_first(pert_nonzero_mean):
"""Test with non-zero-mean data in 2D along the first axis."""
ts, pert_known = pert_nonzero_mean
ts = np.array([ts, ts, ts]).transpose()
pert_known = np.array([pert_known, pert_known, pert_known]).transpose()
assert_array_equal(pert_known, get_perturbation(ts, axis=0))
def test_known_perturbation_nonzero_mean_1d(pert_nonzero_mean):
"""Test with non-zero-mean data in 1D."""
ts, pert_known = pert_nonzero_mean
assert_array_equal(pert_known, get_perturbation(ts))
def test_known_perturbation_zero_mean_2d_axis_last(pert_zero_mean):
"""Test with zero-mean data in 2D along the last axis."""
ts, pert_known = pert_zero_mean
ts = np.array([ts, ts, ts])
pert_known = np.array([pert_known, pert_known, pert_known])
assert_array_equal(pert_known, get_perturbation(ts, axis=-1))
def test_known_perturbation_nonzero_mean_2d_axis_first(pert_nonzero_mean):
"""Test with non-zero-mean data in 2D along the first axis."""
ts, pert_known = pert_nonzero_mean
ts = np.array([ts, ts, ts]).transpose()
pert_known = np.array([pert_known, pert_known, pert_known]).transpose()
assert_array_equal(pert_known, get_perturbation(ts, axis=0))
def test_known_perturbation_nonzero_mean_1d(pert_nonzero_mean):
"""Test with non-zero-mean data in 1D."""
ts, pert_known = pert_nonzero_mean
assert_array_equal(pert_known, get_perturbation(ts))
def test_known_perturbation_zero_mean_2d_axis_last(pert_zero_mean):
"""Test with zero-mean data in 2D along the last axis."""
ts, pert_known = pert_zero_mean
ts = np.array([ts, ts, ts])
pert_known = np.array([pert_known, pert_known, pert_known])
assert_array_equal(pert_known, get_perturbation(ts, axis=-1))
