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