add more tests for endpoint

materialsproject · Sep 28, 2024 · d2a1fbc · d2a1fbc
1 parent 5540253
commit d2a1fbc
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/tests/test_mprester.py b/tests/test_mprester.py
@@ -354,9 +354,14 @@ def test_get_api_key_from_settings(self, monkeypatch: pytest.MonkeyPatch):
         get "PMG_MAPI_KEY" from "SETTINGS".
         """
         monkeypatch.delenv("MP_API_KEY", raising=False)
+        assert os.environ.get("MP_API_KEY") is None
 
         # patch pymatgen.core.SETTINGS to contain PMG_MAPI_KEY
         monkeypatch.setitem(SETTINGS, "PMG_MAPI_KEY", FAKE_MP_API_KEY)
 
-        # create MPRester and check that it picked up the API key from SETTINGS
         assert MPRester().api_key == FAKE_MP_API_KEY
+
+    def test_get_default_endpoint(self, monkeypatch: pytest.MonkeyPatch):
+        monkeypatch.delenv("MP_API_ENDPOINT", raising=False)
+        assert os.environ.get("MP_API_ENDPOINT") is None
+        assert MPRester().endpoint == "https://api.materialsproject.org/"