Merge pull request #430 from robbievanleeuwen/fix-plotting-tests

Only run plotting tests on linux

Author: robbievanleeuwen

tests/geometry/test_geometry.py

@@ -3,6 +3,7 @@
 from __future__ import annotations
 
 import json
+import platform
 import sys
 from pathlib import Path
 
@@ -786,6 +787,7 @@ def test_split_section(unit_square):
         unit_square.split_section(point_i)
 
 
+@pytest.mark.skipif(platform.system() != "Linux", reason="Only test plotting on Linux")
 def test_plot_geometry(small_square_with_hole):
     """Tests plot geometry executes without errors."""
     geom = small_square_with_hole

tests/post/test_post.py

@@ -2,6 +2,8 @@
 
 from __future__ import annotations
 
+import platform
+
 import matplotlib.pyplot as plt
 import pytest
 import pytest_check as check
@@ -11,6 +13,11 @@
 from sectionproperties.pre.library import rectangular_section
 
 
+linux_only = pytest.mark.skipif(
+    platform.system() != "Linux", reason="Only test plotting on Linux"
+)
+
+
 @pytest.fixture
 def example_section() -> Section:
     """Creates an example section with geometric properties.
@@ -51,6 +58,7 @@ def test_centroidal_properties_error(example_section):
         )
 
 
+@linux_only
 def test_save_plot(example_section, tmp_path):
     """Tests saving a plot."""
     sec = example_section
@@ -61,22 +69,24 @@ def test_save_plot(example_section, tmp_path):
     plt.close("all")
 
 
-# def test_supplied_axis(example_section):
-#     """Tests supplying an axis to a plot."""
-#     sec = example_section
-#     _, ax = plt.subplots()
+@linux_only
+def test_supplied_axis(example_section):
+    """Tests supplying an axis to a plot."""
+    sec = example_section
+    _, ax = plt.subplots()
 
-#     sec.plot_mesh(ax=ax, render=False)
-#     plt.close("all")
-#     sec.plot_mesh(nrows=2, axis_index=1, render=False)
-#     plt.close("all")
+    sec.plot_mesh(ax=ax, render=False)
+    plt.close("all")
+    sec.plot_mesh(nrows=2, axis_index=1, render=False)
+    plt.close("all")
 
-#     with pytest.raises(ValueError, match="is not compatible"):
-#         sec.plot_mesh(nrows=2, ncols=2, axis_index=5, render=False)
+    with pytest.raises(ValueError, match="is not compatible"):
+        sec.plot_mesh(nrows=2, ncols=2, axis_index=5, render=False)
 
-#     plt.close("all")
+    plt.close("all")
 
 
+@linux_only
 def test_plot_centroids(example_section):
     """Tests plotting centroids."""
     sec = example_section

tests/post/test_stress_post.py

@@ -2,13 +2,20 @@
 
 from __future__ import annotations
 
+import platform
+
 import pytest
 
 from sectionproperties.analysis import Section
 from sectionproperties.pre import Material
 from sectionproperties.pre.library import rectangular_section
 
 
+linux_only = pytest.mark.skipif(
+    platform.system() != "Linux", reason="Only test plotting on Linux"
+)
+
+
 @pytest.fixture
 def example_section() -> tuple[Section, Material]:
     """Creates an example section with geometric properties.
@@ -26,6 +33,7 @@ def example_section() -> tuple[Section, Material]:
     return Section(geometry=geom), mat_b
 
 
+@linux_only
 def test_stress_plot(example_section):
     """Tests the plot_stress() method."""
     sec, mat_b = example_section
@@ -36,6 +44,7 @@ def test_stress_plot(example_section):
     stress.plot_stress(stress="n_zz", material_list=[mat_b], render=False)
 
 
+@linux_only
 def test_stress_plot_constant():
     """Tests the plot_stress() method with a constant stress."""
     geom = rectangular_section(d=1, b=1)
@@ -45,6 +54,7 @@ def test_stress_plot_constant():
     sec.calculate_stress(n=1).plot_stress(stress="n_zz", render=False)
 
 
+@linux_only
 def test_stress_vector_plot(example_section):
     """Tests the plot_stress_vector() method."""
     sec, _ = example_section
@@ -53,6 +63,7 @@ def test_stress_vector_plot(example_section):
     sec.calculate_stress(mzz=1).plot_stress_vector(stress="zxy", render=False)
 
 
+@linux_only
 def test_plot_mohrs_circles(example_section):
     """Tests the plot_mohrs_circles() method."""
     sec, _ = example_section