add tests

Lightning-AI · Sep 27, 2024 · 894ed6a · 894ed6a
1 parent c9a00a7
commit 894ed6a
Showing 1 changed file with 15 additions and 0 deletions.
diff --git a/tests/test_utils.py b/tests/test_utils.py
@@ -0,0 +1,15 @@
+import pickle
+
+from fastapi import HTTPException
+
+from litserve.utils import dump_exception
+
+
+def test_dump_exception():
+    e1 = dump_exception(HTTPException(status_code=404, detail="Not Found"))
+    assert isinstance(e1, bytes)
+
+    exc = HTTPException(400, "Custom Lit error")
+    isinstance(pickle.loads(dump_exception(exc)), HTTPException)
+    assert pickle.loads(dump_exception(exc)).detail == "Custom Lit error"
+    assert pickle.loads(dump_exception(exc)).status_code == 400