Skip to content
This repository was archived by the owner on Apr 26, 2024. It is now read-only.

Commit b26a203

Browse files
committed
Fixup types
1 parent 6d82dd3 commit b26a203

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

tests/api/test_errors.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -21,10 +21,12 @@
2121
class ErrorsTestCase(unittest.TestCase):
2222
def test_limit_exceeded_header(self) -> None:
2323
err = LimitExceededError(retry_after_ms=100)
24-
self.assertEqual(err.error_dict({}).get("retry_after_ms"), 100)
24+
self.assertEqual(err.error_dict(None).get("retry_after_ms"), 100)
25+
assert err.headers
2526
self.assertEqual(err.headers.get("Retry-After"), "1")
2627

2728
def test_limit_exceeded_rounding(self) -> None:
2829
err = LimitExceededError(retry_after_ms=3001)
29-
self.assertEqual(err.error_dict({}).get("retry_after_ms"), 3001)
30+
self.assertEqual(err.error_dict(None).get("retry_after_ms"), 3001)
31+
assert err.headers
3032
self.assertEqual(err.headers.get("Retry-After"), "4")

0 commit comments

Comments
 (0)