DazedAnon
a1af8fb985
- Queue only untranslated rows from mixed skip-check batches - Let .env batchsize and frequency_penalty override model defaults - Cover skip behavior and pricing batch size override in tests
22 lines
724 B
Python
22 lines
724 B
Python
import os
|
|
import unittest
|
|
from unittest.mock import patch
|
|
|
|
from util.translation import getPricingConfig
|
|
|
|
|
|
class PricingBatchSizeOverrideTests(unittest.TestCase):
|
|
def test_env_batchsize_overrides_model_default(self):
|
|
with patch.dict(os.environ, {"batchsize": "2"}, clear=False):
|
|
cfg = getPricingConfig("claude-sonnet-4-6")
|
|
self.assertEqual(cfg["batchSize"], 2)
|
|
|
|
def test_missing_env_keeps_model_default(self):
|
|
env = {k: v for k, v in os.environ.items() if k != "batchsize"}
|
|
with patch.dict(os.environ, env, clear=True):
|
|
cfg = getPricingConfig("claude-sonnet-4-6")
|
|
self.assertEqual(cfg["batchSize"], 30)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|