From a88e0f5f3dfdc7fb432b811885aef805cd2e7323 Mon Sep 17 00:00:00 2001 From: yuwk <1729065730@qq.com> Date: Fri, 11 Sep 2026 13:56:50 +0800 Subject: [PATCH] test(bedrock): cover every video extension added to content type inference The parametrized case list only exercised five of the nine video extensions, so codecov flagged the remaining map entries as uncovered Add flv, mpeg, mpg and wmv so every added entry has a case --- .../test_litellm_core_utils_prompt_templates_common_utils.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/tests/test_litellm/litellm_core_utils/prompt_templates/test_litellm_core_utils_prompt_templates_common_utils.py b/tests/test_litellm/litellm_core_utils/prompt_templates/test_litellm_core_utils_prompt_templates_common_utils.py index b3b5b08e8da..c84bcbd6ae9 100644 --- a/tests/test_litellm/litellm_core_utils/prompt_templates/test_litellm_core_utils_prompt_templates_common_utils.py +++ b/tests/test_litellm/litellm_core_utils/prompt_templates/test_litellm_core_utils_prompt_templates_common_utils.py @@ -1824,6 +1824,10 @@ class TestEncryptedReasoningReplay: ("https://example.com/clip.mov", "video/mov"), ("https://example.com/clip.mkv", "video/mkv"), ("https://example.com/clip.3gp", "video/3gp"), + ("https://example.com/clip.flv", "video/flv"), + ("https://example.com/clip.mpeg", "video/mpeg"), + ("https://example.com/clip.mpg", "video/mpg"), + ("https://example.com/clip.wmv", "video/wmv"), ], ) def test_infer_video_content_type_from_url_extension(url, expected):