make test compatible with coverage reporting

Author: MaHaWo

tests/test_pyvista_utils.py

@@ -1,4 +1,4 @@
-import sme_contrib.pyvista_utils as pvu
+import sme_contrib as smeplots
 import numpy as np
 import matplotlib.pyplot as plt
 import matplotlib.colors as mcolors
@@ -12,22 +12,22 @@ def test_rgb_to_scalar():
             [[[0, 0, 0], [255, 255, 255]], [[0, 0, 0], [255, 255, 255]]],
         ]
     )
-    scalar_img = pvu.rgb_to_scalar(img)
+    scalar_img = smeplots.pyvista_utils.rgb_to_scalar(img)
     assert scalar_img.shape == (3, 2, 2)
     assert np.all(
         scalar_img == np.array([[[0, 1], [0, 1]], [[0, 1], [0, 1]], [[0, 1], [0, 1]]])
     )
 
 
 def test_make_discrete_colormap():
-    lt = pvu.make_discrete_colormap()
+    lt = smeplots.pyvista_utils.make_discrete_colormap()
     cm = plt.get_cmap("tab10").colors
     should = (np.array([mcolors.to_rgba(cm[0])]) * 255).astype(np.int32)
     assert lt.n_values == 1
     assert lt.scalar_range == (0, 1)
     assert np.all(lt.values == should)
 
-    lt = pvu.make_discrete_colormap("tab20", np.array([0, 1, 2, 3]))
+    lt = smeplots.pyvista_utils.make_discrete_colormap("tab20", np.array([0, 1, 2, 3]))
     assert lt.n_values == 4
     assert lt.scalar_range == (0, 4)
     cm = plt.get_cmap("tab20").colors
@@ -46,18 +46,18 @@ def test_make_discrete_colormap():
 
 
 def test_find_layout():
-    assert pvu.find_layout(1) == (1, 1)
-    assert pvu.find_layout(3) == (1, 3)
-    assert pvu.find_layout(5) == (2, 3)
-    assert pvu.find_layout(8) == (2, 4)
-    assert pvu.find_layout(10) == (2, 5)
-    assert pvu.find_layout(15) == (3, 5)
-    assert pvu.find_layout(15, portrait=True) == (5, 3)
-    assert pvu.find_layout(16) == (4, 4)
-    assert pvu.find_layout(19) == (4, 5)
-    assert pvu.find_layout(23) == (4, 6)
-    assert pvu.find_layout(25) == (5, 5)
-    assert pvu.find_layout(27) == (4, 7)
-    assert pvu.find_layout(29) == (5, 6)
-    assert pvu.find_layout(31) == (5, 7)
-    assert pvu.find_layout(31, portrait=True) == (7, 5)
+    assert smeplots.pyvista_utils.find_layout(1) == (1, 1)
+    assert smeplots.pyvista_utils.find_layout(3) == (1, 3)
+    assert smeplots.pyvista_utils.find_layout(5) == (2, 3)
+    assert smeplots.pyvista_utils.find_layout(8) == (2, 4)
+    assert smeplots.pyvista_utils.find_layout(10) == (2, 5)
+    assert smeplots.pyvista_utils.find_layout(15) == (3, 5)
+    assert smeplots.pyvista_utils.find_layout(15, portrait=True) == (5, 3)
+    assert smeplots.pyvista_utils.find_layout(16) == (4, 4)
+    assert smeplots.pyvista_utils.find_layout(19) == (4, 5)
+    assert smeplots.pyvista_utils.find_layout(23) == (4, 6)
+    assert smeplots.pyvista_utils.find_layout(25) == (5, 5)
+    assert smeplots.pyvista_utils.find_layout(27) == (4, 7)
+    assert smeplots.pyvista_utils.find_layout(29) == (5, 6)
+    assert smeplots.pyvista_utils.find_layout(31) == (5, 7)
+    assert smeplots.pyvista_utils.find_layout(31, portrait=True) == (7, 5)