Skip to content

Commit efdd92f

Browse files
committed
Chore: Add tests for new fast model
1 parent 3a1c9f4 commit efdd92f

File tree

1 file changed

+33
-0
lines changed

1 file changed

+33
-0
lines changed

tests/test_process.py

Lines changed: 33 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -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
91124
async def test_process_max_prompt_length() -> None:
92125
client = DecartClient(api_key="test-key")

0 commit comments

Comments
 (0)