daniel543x/artcraft
1
0 Fork 0
mirror of https://github.com/storytold/artcraft.git synced 2026-11-04 17:38:10 +00:00
Code Issues Packages Projects Releases Wiki Activity

update tests

Browse Source
This commit is contained in:
Brandon Thomas committed 2026-02-23 23:24:16 +07:00
1 parent d73ddfa5ad
commit 2a5f9eded8
1 file changed
+6 -9
Download Patch File Download Diff File Expand all files Collapse all files
crates/lib/artcraft_router/tests/artcraft_seedance_2p0.rs
+6 -9
View File
@@ -132,19 +132,16 @@ mod plan_tests {
#[test]
fn batch_count_direct_mapping() {
let VideoGenerationPlan::ArtcraftSeedance2p0(plan) = begin_video_generation(
&GenerateVideoRequest { video_batch_count: Some(1), ..base_request() }
).unwrap();
let req = GenerateVideoRequest { video_batch_count: Some(1), ..base_request() };
let VideoGenerationPlan::ArtcraftSeedance2p0(plan) = begin_video_generation(&req).unwrap();
assert!(matches!(plan.batch_count, Seedance2p0BatchCount::One));
let VideoGenerationPlan::ArtcraftSeedance2p0(plan) = begin_video_generation(
&GenerateVideoRequest { video_batch_count: Some(2), ..base_request() }
).unwrap();
let req = GenerateVideoRequest { video_batch_count: Some(2), ..base_request() };
let VideoGenerationPlan::ArtcraftSeedance2p0(plan) = begin_video_generation(&req).unwrap();
assert!(matches!(plan.batch_count, Seedance2p0BatchCount::Two));
let VideoGenerationPlan::ArtcraftSeedance2p0(plan) = begin_video_generation(
&GenerateVideoRequest { video_batch_count: Some(4), ..base_request() }
).unwrap();
let req = GenerateVideoRequest { video_batch_count: Some(4), ..base_request() };
let VideoGenerationPlan::ArtcraftSeedance2p0(plan) = begin_video_generation(&req).unwrap();
assert!(matches!(plan.batch_count, Seedance2p0BatchCount::Four));
}