Skip to content

Commit 7f96462

Browse files
committed
Tell tests about output video framerate used in test data
1 parent a0d4125 commit 7f96462

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

tests/test_df3d.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -22,6 +22,7 @@
2222
TEST_DATA_LOCATION_RESULT_FILE_3D = f"{TEST_DATA_LOCATION_RESULT}/df3d_result_3d.pkl"
2323
TEST_DATA_LOCATION_REFERENCE_VIDEO_2D = f"{TEST_DATA_LOCATION_RESULT}/video_pose2d.mp4"
2424
TEST_DATA_LOCATION_REFERENCE_VIDEO_3D = f"{TEST_DATA_LOCATION_RESULT}/video_pose3d.mp4"
25+
TEST_DATA_VIDEO_FRAMERATE = 5
2526
TEST_DATA_LOCATION_WORKING = f"{TEST_DATA_LOCATION}/working"
2627
TEST_DATA_LOCATION_WORKING_RESULT = f"{TEST_DATA_LOCATION_WORKING}_df3d"
2728

@@ -254,7 +255,8 @@ def test_video_2d(self):
254255
)
255256

256257
df3d.video.make_pose2d_video(
257-
core.plot_2d, core.num_images, core.input_folder, core.output_folder
258+
core.plot_2d, core.num_images, core.input_folder,
259+
core.output_folder, fps=TEST_DATA_VIDEO_FRAMERATE
258260
)
259261

260262
video_name = "video_pose2d_" + core.input_folder.replace("/", "_") + ".mp4"
@@ -298,6 +300,7 @@ def test_video_3d(self):
298300
core.num_images,
299301
core.input_folder,
300302
core.output_folder,
303+
fps=TEST_DATA_VIDEO_FRAMERATE,
301304
)
302305

303306
video_name = "video_pose3d_" + core.input_folder.replace("/", "_") + ".mp4"

0 commit comments

Comments
 (0)