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