You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
41 lines
1.1 KiB
41 lines
1.1 KiB
3 years ago
|
import pickle
|
||
|
|
|
||
|
|
import pytest
|
||
|
|
|
||
|
|
import openai
|
||
|
|
|
||
|
|
EXCEPTION_TEST_CASES = [
|
||
|
|
openai.InvalidRequestError(
|
||
|
|
"message",
|
||
|
|
"param",
|
||
|
|
code=400,
|
||
|
|
http_body={"test": "test1"},
|
||
|
|
http_status="fail",
|
||
|
|
json_body={"text": "iono some text"},
|
||
|
|
headers={"request-id": "asasd"},
|
||
|
|
),
|
||
|
|
openai.error.AuthenticationError(),
|
||
|
|
openai.error.PermissionError(),
|
||
|
|
openai.error.RateLimitError(),
|
||
|
|
openai.error.ServiceUnavailableError(),
|
||
|
|
openai.error.SignatureVerificationError("message", "sig_header?"),
|
||
|
|
openai.error.APIConnectionError("message!", should_retry=True),
|
||
|
|
openai.error.TryAgain(),
|
||
|
|
openai.error.Timeout(),
|
||
|
|
openai.error.APIError(
|
||
|
|
message="message",
|
||
|
|
code=400,
|
||
|
|
http_body={"test": "test1"},
|
||
|
|
http_status="fail",
|
||
|
|
json_body={"text": "iono some text"},
|
||
|
|
headers={"request-id": "asasd"},
|
||
|
|
),
|
||
|
|
openai.error.OpenAIError(),
|
||
|
|
]
|
||
|
|
|
||
|
|
|
||
|
|
class TestExceptions:
|
||
|
|
@pytest.mark.parametrize("error", EXCEPTION_TEST_CASES)
|
||
|
|
def test_exceptions_are_pickleable(self, error) -> None:
|
||
|
|
assert error.__repr__() == pickle.loads(pickle.dumps(error)).__repr__()
|