diff --git a/tests/test_base_client.py b/tests/test_base_client.py index dcfc00af..81b689b8 100644 --- a/tests/test_base_client.py +++ b/tests/test_base_client.py @@ -325,8 +325,8 @@ def inner(h: BaseHTTPRequestHandler): content_length = int(h.headers.get('Content-Length', 0)) if content_length > 0: requests.append(h.rfile.read(content_length)) - - h.send_response(429) + + h.send_response(429) h.send_header('Retry-After', '1') h.end_headers() @@ -335,7 +335,7 @@ def inner(h: BaseHTTPRequestHandler): with http_fixture_server(inner) as host: client = _BaseClient() - + # Should raise error immediately without retry. with pytest.raises(DatabricksError): client.do('POST', f'{host}/foo', data=stream)