diff --git a/halfapi/lib/responses.py b/halfapi/lib/responses.py index 0ed2f1f..7fb0acc 100644 --- a/halfapi/lib/responses.py +++ b/halfapi/lib/responses.py @@ -62,7 +62,7 @@ class ORJSONResponse(JSONResponse): decimal.Decimal } list_types = { - type(set) + set } if type(x) in str_types: @@ -70,7 +70,7 @@ class ORJSONResponse(JSONResponse): if type(x) in list_types: return list(x) - raise TypeError(f'Type {type(x)} is not handled in ORJSONResponse') + raise TypeError(f'Type {type(x)} is not handled by ORJSONResponse') class HJSONResponse(ORJSONResponse): diff --git a/tests/test_lib_responses.py b/tests/test_lib_responses.py index c63fea8..3f66441 100644 --- a/tests/test_lib_responses.py +++ b/tests/test_lib_responses.py @@ -8,7 +8,7 @@ def test_orjson(): test_obj = { "ok": "ko", "dec": decimal.Decimal(42), - "set": set([0,4,2]) + "set": {0, 4, 2} } resp = ORJSONResponse(test_obj) @@ -18,4 +18,4 @@ def test_orjson(): assert 'ok' in test_obj_dec.keys() assert isinstance(test_obj_dec['ok'], str) assert isinstance(test_obj_dec['dec'], str) - assert isinstance(test_obj_dec['set'], liststr) + assert isinstance(test_obj_dec['set'], list)