diff --git a/tests/test_TiO2.py b/tests/test_TiO2.py index a2652b6..95b115d 100644 --- a/tests/test_TiO2.py +++ b/tests/test_TiO2.py @@ -5,11 +5,13 @@ import os from shutil import copytree, rmtree -import elli import numpy as np -from elli.fitting import ParamsHist +import pytest from pytest import fixture +import elli +from elli.fitting import ParamsHist + @fixture def datadir(tmpdir, request): @@ -135,6 +137,8 @@ def test_solver4x4_expm(self, si_dispersion, meas_data): def test_solver4x4_expm_torch(self, si_dispersion, meas_data): """The solver4x4 with pytorch propagator is within chi square accuracy""" + + _ = pytest.importorskip("torch") sim_data = ( elli.Structure(elli.AIR, self.Layer, si_dispersion) .evaluate(