auto set version

pyproject.toml

@@ -28,6 +28,7 @@ scipy = "^1.13.1"
 cloudpickle = "^3.0.0"
 nptyping = "^2.5.0"
 librosa = "^0.10.2.post1"
+toml = "^0.10.2"
 
 
 [tool.poetry.group.docs.dependencies]

src/new_fave/__init__.py

@@ -10,8 +10,17 @@
 
 from importlib.metadata import version
 
-# this is bugged for testing
-#__version__ = version("new_fave")
+from pathlib import Path
+import toml
+
+__version__ = "unknown"
+# adopt path to your pyproject.toml
+pyproject_toml_file = Path(__file__).parent.parent.parent / "pyproject.toml"
+if pyproject_toml_file.exists() and pyproject_toml_file.is_file():
+    data = toml.load(pyproject_toml_file)
+    # check project.version
+    if "tool" in data and "poetry" in data["tool"] and "version" in data["tool"]["poetry"]:
+        __version__ = data["tool"]["poetry"]["version"]
 
 __all__ = [
     "VowelMeasurement", 

tests/test_version.py

@@ -0,0 +1,18 @@
+from pathlib import Path
+import toml
+import new_fave
+
+def test_version():
+    pyproject_toml_file = Path(__file__).parent.parent / "pyproject.toml"
+    toml_v = "unknown"
+    if pyproject_toml_file.exists() and pyproject_toml_file.is_file():
+        data = toml.load(pyproject_toml_file)
+        # check project.version
+        if "project" in data and "version" in data["project"]:
+            toml_v = data["project"]["version"]
+        # check tool.poetry.version
+        elif "tool" in data and "poetry" in data["tool"] and "version" in data["tool"]["poetry"]:
+            toml_v = data["tool"]["poetry"]["version"]
+
+    assert new_fave.__version__ == toml_v
+