Fix polygon plots in docs

src/data_morph/shapes/factory.py

@@ -144,7 +144,7 @@ def plot_available_shapes(self) -> Axes:
             num_rows,
             num_cols,
             layout='constrained',
-            figsize=(12, 2 * num_rows),
+            figsize=(2 * num_cols, 2 * num_rows),
         )
         fig.get_layout_engine().set(w_pad=0.2, h_pad=0.2)
 

src/data_morph/shapes/lines/diamond.py

@@ -15,7 +15,7 @@ class Diamond(LineCollection):
 
         import matplotlib.pyplot as plt
         from data_morph.data.loader import DataLoader
-        from data_morph.shapes.polygons import Diamond
+        from data_morph.shapes.lines import Diamond
 
         _ = Diamond(DataLoader.load_dataset('panda')).plot()
 

src/data_morph/shapes/lines/rectangle.py

@@ -15,7 +15,7 @@ class Rectangle(LineCollection):
 
         import matplotlib.pyplot as plt
         from data_morph.data.loader import DataLoader
-        from data_morph.shapes.polygons import Rectangle
+        from data_morph.shapes.lines import Rectangle
 
         _ = Rectangle(DataLoader.load_dataset('panda')).plot()
 

src/data_morph/shapes/lines/star.py

@@ -15,7 +15,7 @@ class Star(LineCollection):
 
         import matplotlib.pyplot as plt
         from data_morph.data.loader import DataLoader
-        from data_morph.shapes.polygons import Star
+        from data_morph.shapes.lines import Star
 
         _ = Star(DataLoader.load_dataset('panda')).plot()
 

tests/shapes/test_polygons.py

@@ -1,4 +1,4 @@
-"""Test polygons module."""
+"""Test polygons in the lines module."""
 
 from numbers import Number
 
@@ -8,7 +8,7 @@
 pytestmark = [pytest.mark.shapes, pytest.mark.lines, pytest.mark.polygons]
 
 
-class PolygonsModuleTestBase:
+class PolygonsLineModuleTestBase:
     """Base for testing polygon shapes."""
 
     shape_name: str
@@ -47,7 +47,7 @@ def test_lines_form_polygon(self, shape):
         assert np.unique(endpoints, axis=0).shape[0] == self.expected_line_count
 
 
-class TestDiamond(PolygonsModuleTestBase):
+class TestDiamond(PolygonsLineModuleTestBase):
     """Test the Diamond class."""
 
     shape_name = 'diamond'
@@ -59,7 +59,7 @@ def test_slopes(self, slopes):
         np.testing.assert_array_equal(np.sort(slopes).flatten(), [-1.5, -1.5, 1.5, 1.5])
 
 
-class TestRectangle(PolygonsModuleTestBase):
+class TestRectangle(PolygonsLineModuleTestBase):
     """Test the Rectangle class."""
 
     shape_name = 'rectangle'
@@ -71,7 +71,7 @@ def test_slopes(self, slopes):
         np.testing.assert_array_equal(np.sort(slopes).flatten(), [0, 0, np.inf, np.inf])
 
 
-class TestStar(PolygonsModuleTestBase):
+class TestStar(PolygonsLineModuleTestBase):
     """Test the Star class."""
 
     shape_name = 'star'