Skip client tests if api key not in env

materialsproject · May 24, 2021 · 26db338 · 26db338
1 parent 1db96b7
commit 26db338
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/tests/test_client.py b/tests/test_client.py
@@ -1,5 +1,5 @@
 import pytest
-from mp_api.core.client import MPRestError
+import os
 from mp_api.matproj import MPRester
 
 key_only_resters = {
@@ -33,3 +33,7 @@ def test_generic_get_methods(rester):
     elif name not in special_resters:
         doc = rester.get_document_by_id(key_only_resters[name], fields=[rester.primary_key])
         assert isinstance(doc, rester.document_model)
+
+
+if os.environ.get("MP_API_KEY", None) is None:
+    pytest.mark.skip(test_generic_get_methods)