From 50debbe83d8bcd56c06de001c920fce039c623ef Mon Sep 17 00:00:00 2001 From: Valerie Tsai <87097162+vtsai881@users.noreply.github.com> Date: Tue, 5 Dec 2023 21:55:56 -0800 Subject: [PATCH] Update test_commands.py --- tests/gui/test_commands.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/tests/gui/test_commands.py b/tests/gui/test_commands.py index 2f4a13c38..652edcea6 100644 --- a/tests/gui/test_commands.py +++ b/tests/gui/test_commands.py @@ -291,14 +291,14 @@ def assert_videos_written(num_videos: int, labels_path: str = None): okay = ExportAnalysisFile_ask(context=context, params=params) assert okay == True ExportAnalysisFile.do_action(context=context, params=params) - assert_videos_written(num_videos=2, labels_path=context.state["filename"]) + assert_videos_written(num_videos=1, labels_path=context.state["filename"]) # Test with all_videos False and all_frames False - params = {"all_videos": False, "all_frames":False, "csv": csv} + params = {"all_videos": False, "all_frames": False, "csv": csv} okay = ExportAnalysisFile_ask(context=context, params=params) assert okay == True ExportAnalysisFile.do_action(context=context, params=params) - assert_videos_written(num_videos=2, labels_path=context.state["filename"]) + assert_videos_written(num_videos=1, labels_path=context.state["filename"]) # Add labels path and test with all_videos True and all_frames True (single video) context.state["filename"] = str(tmpdir.with_name("path.to.labels")) @@ -311,7 +311,7 @@ def assert_videos_written(num_videos: int, labels_path: str = None): # Add a video (no labels) and test with all_videos True and all_frames True labels.add_video(small_robot_mp4_vid) - params = {"all_videos": True, "all_frames: True, "csv": csv} + params = {"all_videos": True, "all_frames": True, "csv": csv} okay = ExportAnalysisFile_ask(context=context, params=params) assert okay == True ExportAnalysisFile.do_action(context=context, params=params)