Skip to content
This repository has been archived by the owner on Dec 20, 2024. It is now read-only.

Commit

Permalink
fix: revise test of gaussian process
Browse files Browse the repository at this point in the history
  • Loading branch information
oesteban committed Jul 4, 2024
1 parent db1d8f9 commit 671a0eb
Showing 1 changed file with 3 additions and 6 deletions.
9 changes: 3 additions & 6 deletions test/test_model.py
Original file line number Diff line number Diff line change
Expand Up @@ -28,6 +28,7 @@
from sklearn.gaussian_process.kernels import DotProduct, WhiteKernel

from eddymotion import model
from eddymotion.model.dipy import GaussianProcessModel
from eddymotion.data.dmri import DWI
from eddymotion.data.splitting import lovo_split
from eddymotion.exceptions import ModelNotFittedError
Expand Down Expand Up @@ -107,12 +108,8 @@ def test_average_model():
assert np.all(tmodel_2000.predict([0, 0, 0]) == 1100)


def test_gp_model(datadir):
dwi = DWI.from_filename(datadir / "dwi.h5")

kernel = DotProduct() + WhiteKernel()

gp = model.GaussianProcessModel(dwi=dwi, kernel=kernel)
def test_gp_model():
gp = GaussianProcessModel(kernel="default")

assert isinstance(gp, model.GaussianProcessModel)

Expand Down

0 comments on commit 671a0eb

Please sign in to comment.