Skip to content

Commit

Permalink
Update test_r_cluster.py
Browse files Browse the repository at this point in the history
  • Loading branch information
Ramana-Raja authored Nov 27, 2024
1 parent 4670e51 commit db1cf12
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions aeon/clustering/tests/test_r_cluster.py
Original file line number Diff line number Diff line change
Expand Up @@ -26,8 +26,8 @@ def test_r_cluster():
X_test, y_test = load_basic_motions(split="test")

r_cluster = RCluster(n_clusters=2)
r_cluster.fit(X_train[0:max_train])
test_shape_result = r_cluster.predict(X_test[0:max_train])
r_cluster._fit(X_train[0:max_train])
test_shape_result = r_cluster._predict(X_test[0:max_train])

assert np.array_equal(test_shape_result, expected_results)
assert r_cluster.n_iter_ == expected_iters
Expand Down

0 comments on commit db1cf12

Please sign in to comment.