Skip to content

Commit adb469d

Browse files
committed
fix test w correct frame dims
1 parent d60764e commit adb469d

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

test/test_encoders.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -609,6 +609,10 @@ def test_bad_input_parameterized(self, tmp_path, method):
609609
RuntimeError,
610610
match=r"Video codec invalid_codec_name not found.",
611611
):
612+
encoder = VideoEncoder(
613+
frames=torch.zeros((5, 3, 64, 64), dtype=torch.uint8),
614+
frame_rate=30,
615+
)
612616
encoder.to_file(str(tmp_path / "output.mp4"), codec="invalid_codec_name")
613617

614618
def test_bad_input(self, tmp_path):

0 commit comments

Comments
 (0)