Skip to content

Commit

Permalink
Automatic pre-commit fixes
Browse files Browse the repository at this point in the history
  • Loading branch information
Ramana-Raja authored and github-actions[bot] committed Nov 27, 2024
1 parent ae51709 commit b07777d
Showing 1 changed file with 5 additions and 5 deletions.
10 changes: 5 additions & 5 deletions aeon/clustering/tests/test_r_cluster.py
Original file line number Diff line number Diff line change
Expand Up @@ -13,8 +13,9 @@

expected_results = [0, 0, 0, 0, 0]


@pytest.mark.skipif(
not _check_estimator_deps( RCluster, severity="none"),
not _check_estimator_deps(RCluster, severity="none"),
reason="skip test if required soft dependencies not available",
)
def test_kernel_k_means():
Expand All @@ -24,11 +25,10 @@ def test_kernel_k_means():
X_train, y_train = load_basic_motions(split="train")
X_test, y_test = load_basic_motions(split="test")

r_cluster = RCluster( n_clusters=2)
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])

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
assert np.array_equal( r_cluster.labels_, expected_labels)
assert np.array_equal(r_cluster.labels_, expected_labels)

0 comments on commit b07777d

Please sign in to comment.