def test_linear_model_project_vector():
data = np.zeros((3, 120))
data[0, 0] = 1
data[1, 1] = 1
data[2, 2] = 1
linear_model = LinearVectorModel(data)
sample = np.random.random(120)
weights = linear_model.project(sample)
assert_allclose(weights, sample[:3])
def test_linear_model_project_vector():
data = np.zeros((3, 120))
data[0, 0] = 1
data[1, 1] = 1
data[2, 2] = 1
linear_model = LinearVectorModel(data)
sample = np.random.random(120)
weights = linear_model.project(sample)
assert_allclose(weights, sample[:3])
