def test_get_tiles():
"""just checking method runs at the moment"""
def objective(array):
return np.sum(array**2, axis=1)
# 2D input
x0 = np.linspace(-1, 1, 10)
x1 = np.linspace(-1, 1, 10)
X = np.array([x0, x1]).T
y = objective(X)
# tests
g = Golem(forest_type='rf',
ntrees=5,
goal='max',
random_state=42,
verbose=True)
g.fit(X=X, y=y)
for i in range(5):
_ = g.get_tiles(tree_number=i)
