Skip to content

Commit

Permalink
test constrcution
Browse files Browse the repository at this point in the history
  • Loading branch information
Remi-Gau committed Jun 28, 2023
1 parent ea20e4e commit 30b9b14
Showing 1 changed file with 11 additions and 9 deletions.
20 changes: 11 additions & 9 deletions nilearn/regions/tests/test_parcellations.py
Original file line number Diff line number Diff line change
Expand Up @@ -230,24 +230,26 @@ def test_parcellations_transform_with_multi_confounds_multi_images(
assert signals[0].shape == (10, n_parcel)


@pytest.mark.parametrize("method", METHODS)
@pytest.mark.parametrize("n_parcel", [5])
def test_constructor(method, n_parcel, test_image_2):
parcellator = Parcellations(method=method, n_parcels=n_parcel, verbose=0)
assert parcellator.labels_img_ is not None
if method not in ["kmeans", "rena", "hierarchical_kmeans"]:
assert parcellator.connectivity_ is not None
assert parcellator.masker_ is not None


@pytest.mark.skip("Skip to see if CI passes without it.")
@pytest.mark.parametrize("method", METHODS)
@pytest.mark.parametrize("n_parcel", [5])
def test_fit_transform(method, n_parcel, test_image_2):
def test_fit_transform_with_confounds(method, n_parcel, test_image_2):
rng = np.random.RandomState(42)
fmri_imgs = [test_image_2] * 3
confounds = rng.standard_normal(size=(10, 3))
confounds_list = [confounds] * 3

parcellator = Parcellations(method=method, n_parcels=n_parcel, verbose=0)
signals = parcellator.fit_transform(fmri_imgs)

assert parcellator.labels_img_ is not None
if method not in ["kmeans", "rena", "hierarchical_kmeans"]:
assert parcellator.connectivity_ is not None
assert parcellator.masker_ is not None

# fit_transform with confounds
signals = parcellator.fit_transform(fmri_imgs, confounds=confounds_list)

assert isinstance(signals, list)
Expand Down

0 comments on commit 30b9b14

Please sign in to comment.