Assert that subplot layout is correct.

stefmolin · May 15, 2023 · f08c850 · f08c850
1 parent ae6610e
commit f08c850
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/tests/shapes/test_factory.py b/tests/shapes/test_factory.py
@@ -30,6 +30,11 @@ def test_plot_available_shapes(self, shape_factory, monkeypatch, subset):
             )
 
         axs = shape_factory.plot_available_shapes()
+        if subset is None or subset > 5:
+            assert len(axs) > 1
+        else:
+            assert len(axs) == axs.size
+
         populated_axs = [ax for ax in axs.flatten() if ax.get_figure()]
         assert len(populated_axs) == len(shape_factory.AVAILABLE_SHAPES)
         assert all([ax.get_xlabel() == ax.get_ylabel() == '' for ax in populated_axs])