Merge pull request #3 from cgahr/improved_setup

Improved setup

src/latexplotlib/__init__.py

@@ -1,2 +1,18 @@
 from ._latexplotlib import *
 from ._version import __version__
+
+
+def _make_styles_available() -> None:
+    from pathlib import Path
+
+    import matplotlib.pyplot as plt
+
+    lpl_styles = plt.style.core.read_style_directory(
+        Path(__path__[0]) / "styles"  # noqa: F405
+    )
+
+    plt.style.core.update_nested_dict(plt.style.library, lpl_styles)
+    plt.style.core.available[:] = sorted(plt.style.library.keys())
+
+
+_make_styles_available()

src/latexplotlib/_latexplotlib.py

@@ -4,6 +4,7 @@
 from pathlib import Path
 from typing import TYPE_CHECKING, Any, Dict, Iterator, Optional, Sequence, Tuple, Union
 
+import matplotlib as mpl
 import matplotlib.pyplot as plt
 from appdirs import user_config_dir
 
@@ -327,3 +328,25 @@ def subplots(
     )
 
     return plt.subplots(nrows, ncols, figsize=_figsize, **kwargs)
+
+
+def delete_styles_from_previous_installation() -> None:
+    """A helper function to remove old style files."""
+    old = [
+        "latex10pt-minimal.mplstyle",
+        "latex10pt.mplstyle",
+        "latex11pt-minimal.mplstyle",
+        "latex11pt.mplstyle",
+        "latex12pt-minimal.mplstyle",
+        "latex12pt.mplstyle",
+        "latex9pt-minimal.mplstyle",
+        "latex9pt.mplstyle",
+    ]
+
+    styledir = Path(mpl.get_configdir()) / "stylelib"
+
+    for name in old:
+        style = styledir / name
+
+        if style.exists() and input(f"delete {style}? [y/N]") == "y":
+            style.unlink()

src/latexplotlib/_version.py

@@ -1 +1 @@
-__version__ = "0.5.3"
+__version__ = "0.6.0"

tests/test_styles.py

@@ -3,7 +3,7 @@
 import matplotlib.pyplot as plt
 import pytest
 
-styles_folder = Path("styles/")
+styles_folder = Path("src/latexplotlib/styles/")
 
 
 class TestStyles: