Change test exception catch

materialsproject · Sep 6, 2023 · ec82542 · ec82542
1 parent ee8ff1b
commit ec82542
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/tests/api/test_api.py b/tests/api/test_api.py
@@ -2,6 +2,7 @@
 from random import choice, randint
 from typing import Any, Tuple
 from urllib.parse import urlencode
+import json
 
 import pytest
 from fastapi.encoders import jsonable_encoder
@@ -129,8 +130,8 @@ def search_helper(payload, base: str = "/?", debug=True) -> Tuple[Response, Any]
     res = client.get(url)
     try:
         data = res.json().get("data", [])
-    except Exception:
-        data = res.reason
+    except json.decoder.JSONDecodeError:
+        data = res.text
 
     return res, data