Update plugin.py

klebster2 · Nov 2, 2024 · d3e6af6 · d3e6af6
1 parent 0f73fd6
commit d3e6af6
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/python/plugin.py b/python/plugin.py
@@ -30,8 +30,11 @@ def install(package: str):
     for dataset_name, item in wn.config.index.items()
     if item.get("language")
 }
+if "PYTEST_CURRENT_TEST" in os.environ:
+    ARTEFACT_NAME = LANGUAGE_TO_WORDNET_ARTEFACT["mul"]
+else:
+    ARTEFACT_NAME = LANGUAGE_TO_WORDNET_ARTEFACT.get(vim.eval("g:wn_cmp_language"), "mul")
 
-ARTEFACT_NAME = LANGUAGE_TO_WORDNET_ARTEFACT.get(vim.eval("g:wn_cmp_language"), "mul")
 try:
     spec = wn.Wordnet(ARTEFACT_NAME)