importskip torch

Author: domna

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(