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 a72b9ab commit e75b6e5
Showing 1 changed file with 3 additions and 7 deletions.
10 changes: 3 additions & 7 deletions test/test_model.py
Original file line number Diff line number Diff line change
Expand Up @@ -25,13 +25,13 @@
import numpy as np
import pytest
from sklearn.datasets import make_friedman2
from sklearn.gaussian_process.kernels import DotProduct, WhiteKernel

from eddymotion import model
from eddymotion.data.dmri import DWI
from eddymotion.data.splitting import lovo_split
from eddymotion.exceptions import ModelNotFittedError
from eddymotion.model.base import DEFAULT_MAX_S0, DEFAULT_MIN_S0
from eddymotion.model.dipy import GaussianProcessModel


def test_trivial_model():
Expand Down Expand Up @@ -107,12 +107,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 e75b6e5

Please sign in to comment.