@@ -87,6 +87,39 @@ async def test_process_video_to_video() -> None:
8787 assert result == b"fake video data"
8888
8989
90+ @pytest .mark .asyncio
91+ async def test_process_video_to_video_fast () -> None :
92+ client = DecartClient (api_key = "test-key" )
93+
94+ with patch ("aiohttp.ClientSession" ) as mock_session_cls :
95+ mock_response = MagicMock ()
96+ mock_response .ok = True
97+ mock_response .read = AsyncMock (return_value = b"fake video data" )
98+
99+ mock_session = MagicMock ()
100+ mock_session .__aenter__ = AsyncMock (return_value = mock_session )
101+ mock_session .__aexit__ = AsyncMock (return_value = None )
102+ mock_session .post = MagicMock ()
103+ mock_session .post .return_value .__aenter__ = AsyncMock (return_value = mock_response )
104+ mock_session .post .return_value .__aexit__ = AsyncMock (return_value = None )
105+
106+ mock_session_cls .return_value = mock_session
107+
108+ result = await client .process (
109+ {
110+ "model" : models .video ("lucy-fast-v2v" ),
111+ "prompt" : "Change the car to a motorcycle" ,
112+ "data" : b"fake input video" ,
113+ "resolution" : "480p" ,
114+ "enhance_prompt" : True ,
115+ "num_inference_steps" : 50 ,
116+ "seed" : 42 ,
117+ }
118+ )
119+
120+ assert result == b"fake video data"
121+
122+
90123@pytest .mark .asyncio
91124async def test_process_max_prompt_length () -> None :
92125 client = DecartClient (api_key = "test-key" )
0 commit comments