smart tdee

backend/tests/test_vision_storage.py

@@ -0,0 +1,19 @@
+import unittest
+
+from app.vision.storage import format_upload_images_markdown, upload_media_path
+
+
+class UploadMarkdownTests(unittest.TestCase):
+    def test_single_image(self) -> None:
+        md = format_upload_images_markdown(5, ["abc.jpg"])
+        self.assertIn(upload_media_path(5, "abc.jpg"), md)
+        self.assertIn("![скриншот]", md)
+
+    def test_multiple_images(self) -> None:
+        md = format_upload_images_markdown(3, ["a.jpg", "b.jpg"])
+        self.assertIn("скриншот 1/2", md)
+        self.assertIn("скриншот 2/2", md)
+
+
+if __name__ == "__main__":
+    unittest.main()