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