| import pytest | |
| from omniff.models.code import CodeModel, is_code_request | |
| def test_code_model_interface(): | |
| model = CodeModel(model_id="Qwen/Qwen3-4B") | |
| assert not model.is_loaded | |
| def test_code_model_infer_not_loaded(): | |
| model = CodeModel() | |
| with pytest.raises(RuntimeError, match="not loaded"): | |
| model.infer({"prompt": "write hello world"}) | |
| def test_is_code_request_with_code_block(): | |
| assert is_code_request("fix this:\n```python\ndef foo():\n pass\n```") | |
| def test_is_code_request_with_function(): | |
| assert is_code_request("def calculate_sum(a, b):\n return a + b") | |
| def test_is_code_request_keywords(): | |
| assert is_code_request("write code to refactor this function") | |
| def test_is_code_request_plain_text(): | |
| assert not is_code_request("what is the weather today?") | |
| def test_is_code_request_single_keyword(): | |
| assert not is_code_request("implement the plan") | |