use tomlkit

20c · Feb 5, 2024 · de29d3c · de29d3c
1 parent c85b6d2
commit de29d3c
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/tests/test_plugin_semver2.py b/tests/test_plugin_semver2.py
@@ -2,7 +2,7 @@
 import shutil
 
 import pytest
-import toml
+import tomlkit
 from util import instantiate_semver2 as instantiate
 
 import ctl
@@ -50,7 +50,7 @@ def test_tag_pyproject(tmpdir, ctlr):
 
     plugin.tag(version="2.0.0", repo="dummy_repo", prerelease="rc")
 
-    pyproject = toml.load(pyproject_path)
+    pyproject = tomlkit.load(pyproject_path)
     assert pyproject["tool"]["poetry"]["version"] == "2.0.0-rc.1"
 
 

diff --git a/tests/test_plugin_version.py b/tests/test_plugin_version.py
@@ -2,7 +2,7 @@
 import shutil
 
 import pytest
-import toml
+import tomlkit
 from util import instantiate_version as instantiate
 
 import ctl
@@ -48,7 +48,7 @@ def test_tag_pyproject(tmpdir, ctlr):
 
     plugin.tag(version="2.0.0", repo="dummy_repo")
 
-    pyproject = toml.load(pyproject_path)
+    pyproject = tomlkit.load(pyproject_path)
     assert pyproject["tool"]["poetry"]["version"] == "2.0.0"