Skip to content

Commit b44cf74

Browse files
update tests
1 parent db9a0f5 commit b44cf74

1 file changed

Lines changed: 2 additions & 2 deletions

File tree

tests/unit/test_cli.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -70,7 +70,7 @@ def test_image_command_success(
7070
mock_get_settings.assert_called_once()
7171
mock_initialize_ee.assert_called_once_with("test-project")
7272
mock_export_image.assert_called_once_with(
73-
"PROJECTS/test/assets/test_image", "test-bucket", "test/prefix", False
73+
"PROJECTS/test/assets/test_image", "test-bucket", "test/prefix/", False
7474
)
7575

7676
# Verify the output and exit code
@@ -107,7 +107,7 @@ def test_image_command_with_quantize_flag(
107107
mock_export_image.assert_called_once_with(
108108
"PROJECTS/test/assets/embedding_image",
109109
"embedding-bucket",
110-
"quantized/prefix",
110+
"quantized/prefix/",
111111
True,
112112
)
113113

0 commit comments

Comments
 (0)