1 parent 0f0ff50 commit eb124f5Copy full SHA for eb124f5
1 file changed
tests/test_token_estimator.py
@@ -26,9 +26,9 @@ def test_cjk_estimate(self):
26
27
def test_context_window_matching(self):
28
self.assertEqual(context_window_for("deepseek-v4"), 1_000_000)
29
- self.assertEqual(context_window_for("gpt-5-mini"), 128000)
30
- self.assertEqual(context_window_for("claude-sonnet"), 200000)
31
- self.assertEqual(context_window_for("unknown-model"), 32768)
+ self.assertEqual(context_window_for("gpt-5-mini"), 128_000)
+ self.assertEqual(context_window_for("claude-sonnet"), 200_000)
+ self.assertEqual(context_window_for("unknown-model"), 128_000)
32
33
def test_calibrator(self):
34
c = TokenCalibrator()
0 commit comments