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_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))