client: use pytest in sys.modules

materialsproject · Sep 1, 2022 · e492d26 · e492d26
1 parent 913ff46
commit e492d26
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/mpcontribs-client/mpcontribs/client/__init__.py b/mpcontribs-client/mpcontribs/client/__init__.py
@@ -583,7 +583,7 @@ def __init__(
             raise ValueError(f"{self.url} not a valid URL (one of {VALID_URLS})")
 
         retries, max_retries = 0, 3
-        is_mock_test = 'unittest' in sys.modules and self.protocol == "http"
+        is_mock_test = 'pytest' in sys.modules and self.protocol == "http"
 
         if is_mock_test:
             now = datetime.datetime.now()