| import unittest |
| from unittest.mock import patch |
|
|
| from shop_ledger.llama_backend import LlamaLedgerBackend |
| from shop_ledger.processor import LedgerProcessor, resolve_model_path_from_env |
|
|
|
|
| class ProcessorTests(unittest.TestCase): |
| def test_mock_processor_returns_rows(self): |
| processor = LedgerProcessor(mode="mock") |
| result = processor.process("paid Ravi 1200 for rice bags") |
|
|
| self.assertEqual(result.model_used, "mock heuristic") |
| self.assertEqual(len(result.entries), 1) |
| self.assertEqual(result.entries[0].amount, 1200) |
|
|
| def test_llama_mode_falls_back_without_model(self): |
| processor = LedgerProcessor(mode="llama", model_path="/missing/model.gguf") |
| result = processor.process("customer Nimal owes 750") |
|
|
| self.assertIn("fallback", result.model_used) |
| self.assertEqual(result.entries[0].amount, 750) |
|
|
| def test_llama_backend_uses_readable_model_label(self): |
| label = "unsloth/gemma-4-12b-it-GGUF / gemma-4-12b-it-UD-Q4_K_XL.gguf / llama.cpp" |
| with patch.dict("os.environ", {"LLAMA_MODEL_LABEL": label}): |
| backend = LlamaLedgerBackend(model_path="/models/model.gguf") |
|
|
| self.assertEqual(backend.model_label, label) |
|
|
| def test_env_model_path_wins_for_modal_mounts(self): |
| with patch.dict( |
| "os.environ", |
| { |
| "LLAMA_GGUF_PATH": "/models/model.gguf", |
| "LLAMA_GGUF_REPO": "unused/repo", |
| "LLAMA_GGUF_FILE": "unused.gguf", |
| }, |
| ): |
| self.assertEqual(resolve_model_path_from_env(), "/models/model.gguf") |
|
|
|
|
| if __name__ == "__main__": |
| unittest.main() |
|
|
