| from __future__ import annotations |
|
|
| import unittest |
| from typing import cast |
|
|
| from models.minicpm_vision import MiniCPMVisionConfig, MiniCPMVisionService |
| from models.model_catalog import load_model_catalog |
|
|
|
|
| class MiniCPMVisionServiceTest(unittest.TestCase): |
| def test_status_reports_transformers_availability(self) -> None: |
| status = MiniCPMVisionService.status() |
|
|
| self.assertEqual(status.name, "transformers-vision") |
| self.assertIsInstance(status.available, bool) |
|
|
| def test_unavailable_or_missing_image_path_does_not_load_weights(self) -> None: |
| catalog = load_model_catalog("config/models.yaml") |
| service = MiniCPMVisionService(catalog["minicpm_v46"]) |
|
|
| response = service.vision_chat(False, "Describe it.") |
|
|
| self.assertIn("[MiniCPM vision unavailable]", response) |
|
|
| def test_generation_kwargs_are_explicit(self) -> None: |
| catalog = load_model_catalog("config/models.yaml") |
| service = MiniCPMVisionService( |
| catalog["minicpm_v46"], |
| MiniCPMVisionConfig(max_new_tokens=64, temperature=0.1, do_sample=False), |
| ) |
|
|
| self.assertEqual( |
| service.generation_kwargs(), |
| {"max_new_tokens": 64, "temperature": 0.1, "do_sample": False}, |
| ) |
|
|
| def test_formats_image_text_prompt_with_thinking_instruction(self) -> None: |
| messages = MiniCPMVisionService.format_messages( |
| "Read the label.", |
| image="image-object", |
| thinking=True, |
| ) |
|
|
| self.assertEqual(messages[0]["role"], "user") |
| self.assertEqual(messages[0]["content"][0]["type"], "image") |
| self.assertEqual(messages[0]["content"][1]["type"], "text") |
| self.assertIn("Think carefully", messages[0]["content"][1]["text"]) |
|
|
| def test_video_support_plan_is_documented_but_not_enabled(self) -> None: |
| plan = MiniCPMVisionService.video_support_plan() |
| next_steps = cast(list[str], plan["next_steps"]) |
|
|
| self.assertFalse(plan["implemented"]) |
| self.assertTrue(any("Sample frames locally" in step for step in next_steps)) |
|
|
|
|
| if __name__ == "__main__": |
| unittest.main() |
|
|
