patch tests to reflect mprestwarning

Author: esoteric-ephemera

tests/client/core/test_oxygen_evolution.py

@@ -9,6 +9,7 @@
 from pymatgen.core import Composition
 from pymatgen.entries.computed_entries import ComputedEntry
 
+from mp_api.client.core.exceptions import MPRestWarning
 from mp_api.client.core._oxygen_evolution import (
     DEFAULT_CACHE_FILE,
     NIST_JANAF_O2_MU_T,
@@ -39,7 +40,7 @@ def test_interp():
 
     # ensure user is warned when data is outside fit range
     for badval in (1.0, -50.0):
-        with pytest.warns(UserWarning, match="outside the fitting range"):
+        with pytest.warns(MPRestWarning, match="outside the fitting range"):
             oxyevo.mu_to_temp_spline(badval)
 
 

tests/client/molecules/test_jcesr.py

@@ -4,6 +4,7 @@
 import pytest
 from pymatgen.core.periodic_table import Element
 
+from mp_api.client.core.exceptions import MPRestWarning
 from mp_api.client.routes.molecules.jcesr import JcesrMoleculesRester
 
 from ..conftest import requires_api_key
@@ -51,5 +52,5 @@ def test_client(rester):
 
 
 def test_warning():
-    with pytest.warns(UserWarning, match="unmaintained legacy molecules"):
+    with pytest.warns(MPRestWarning, match="unmaintained legacy molecules"):
         JcesrMoleculesRester()

tests/client/test_mprester.py

@@ -165,7 +165,7 @@ def test_get_entries(self, mpr):
         syms = ["Li", "Fe", "O"]
         chemsys = "Li-Fe-O"
         with pytest.warns(
-            UserWarning, match="The `inc_structure` argument is deprecated"
+            DeprecationWarning, match="The `inc_structure` argument is deprecated"
         ):
             entries = mpr.get_entries(chemsys, inc_structure=False)
 
@@ -558,7 +558,9 @@ def test_get_stability(self, chemsys, thermo_type):
                 ]
 
             if no_compound_entries:
-                with pytest.warns(UserWarning, match="No phase diagram data available"):
+                with pytest.warns(
+                    MPRestWarning, match="No phase diagram data available"
+                ):
                     mpr.get_stability(modified_entries, thermo_type=thermo_type)
                 return