use tomli for python 3.10, otherwise tomllib

csdms · Mar 7, 2024 · 8a4f953 · 8a4f953
1 parent 21fabae
commit 8a4f953
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/tests/test_cli.py b/tests/test_cli.py
@@ -1,6 +1,11 @@
 """Test the babelizer command-line interface"""
 
-import tomllib
+import sys
+
+if sys.version_info >= (3, 11):  # pragma: no cover (PY11+)
+    import tomllib
+else:  # pragma: no cover (<PY311)
+    import tomli as tomllib
 
 from click.testing import CliRunner