refactor: extract conversion of figure to image

src/safeds/_utils/__init__.py

@@ -7,16 +7,22 @@
 if TYPE_CHECKING:
     from ._file_io import _check_and_normalize_file_path
     from ._hashing import _structural_hash
+    from ._plotting import _figure_to_image
+    from ._random import _get_random_seed
 
 apipkg.initpkg(
     __name__,
     {
         "_check_and_normalize_file_path": "._file_io:_check_and_normalize_file_path",
         "_structural_hash": "._hashing:_structural_hash",
+        "_figure_to_image": "._plotting:_figure_to_image",
+        "_get_random_seed": "._random:_get_random_seed",
     },
 )
 
 __all__ = [
     "_check_and_normalize_file_path",
     "_structural_hash",
+    "_figure_to_image",
+    "_get_random_seed",
 ]

src/safeds/_utils/_plotting.py

@@ -0,0 +1,32 @@
+from __future__ import annotations
+
+import io
+from typing import TYPE_CHECKING
+
+from safeds.data.image.containers import Image
+
+if TYPE_CHECKING:
+    import matplotlib.pyplot as plt
+
+
+def _figure_to_image(figure: plt.Figure) -> Image:
+    """
+    Store the figure as an image and closes it.
+
+    Parameters
+    ----------
+    figure:
+        The figure to store.
+
+    Returns
+    -------
+    image:
+        The figure as an image.
+    """
+    import matplotlib.pyplot as plt
+
+    buffer = io.BytesIO()
+    figure.savefig(buffer, format="png")
+    plt.close(figure)  # Prevents the figure from being displayed directly
+    buffer.seek(0)
+    return Image.from_bytes(buffer.read())

src/safeds/data/tabular/plotting/_experimental_column_plotter.py

@@ -3,6 +3,7 @@
 import io
 from typing import TYPE_CHECKING
 
+from safeds._utils import _figure_to_image
 from safeds.data.image.containers import Image
 from safeds.exceptions import NonNumericColumnError
 
@@ -116,8 +117,4 @@ def lag_plot(self, lag: int) -> Image:
             ylabel=f"y(t + {lag})",
         )
 
-        buffer = io.BytesIO()
-        fig.savefig(buffer, format="png")
-        plt.close()  # Prevents the figure from being displayed directly
-        buffer.seek(0)
-        return Image.from_bytes(buffer.read())
+        return _figure_to_image(fig)