def test_lag_gen_shift_pos_one(): fake_data = np.ones((3, 1)) lags = pymtrf.lag_builder(1, 1) lag_matrix = pymtrf.lag_gen(fake_data, lags) test_matrix = np.ones((3, 1)) test_matrix[0, 0] = 0 assert np.all(test_matrix == lag_matrix)
def test_lag_gen_shift_neg(): fake_data = np.ones((3, 1)) lags = pymtrf.lag_builder(-1, 0) lag_matrix = pymtrf.lag_gen(fake_data, lags) test_matrix = np.ones((3, 2)) test_matrix[2, 0] = 0 assert np.all(test_matrix == lag_matrix)
def test_lag_gen_shape_no_lags(): fake_data = np.random.rand(20, 2) lags = pymtrf.lag_builder(0, 0) shape = pymtrf.lag_gen(fake_data, lags).shape assert np.all(shape == (20, 2))
def test_lag_builder_positive_lags(): # Test lag_builder for the creation of a positive lag vector lags = pymtrf.lag_builder(1, 4) assert np.all(lags == [1, 2, 3, 4])
def test_lag_builder_only_zero(): # Test lag_builder for the creation of a negative lag vector, starting with # a positive value lags = pymtrf.lag_builder(0, 0) assert np.all(lags == [0])
def test_lag_builder_starting_from_zero(): # Test lag_builder for the creation of a negative lag vector, starting with # a positive value lags = pymtrf.lag_builder(0, 3) assert np.all(lags == [0, 1, 2, 3])
def test_lag_builder_negative_lags_reverse(): # Test lag_builder for the creation of a negative lag vector, starting with # a positive value lags = pymtrf.lag_builder(2, -2) assert np.all(lags == [2, 1, 0, -1, -2])
def test_lag_builder_negative_lags(): # Test lag_builder for the creation of a negative lag vector, starting with # a negative value lags = pymtrf.lag_builder(-2, 2) assert np.all(lags == [-2, -1, 0, 1, 2])