Skip to content

Commit b9fcf09

Browse files
committed
Update test assert messages
1 parent 1092d74 commit b9fcf09

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

tests/test_existing_images.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -38,7 +38,7 @@ def test_some_files_exist(self, mock_print, mock_gather, mock_exists):
3838
self.assertFalse(updated_df.loc[2, "in_img_dir"]) # c.jpg missing
3939
self.assertEqual(len(filtered_df), 2)
4040
mock_print.assert_called_once()
41-
self.assertIn("There are 1 files", mock_print.call_args[0][0])
41+
self.assertIn("There are 1 of desired files", mock_print.call_args[0][0])
4242

4343
@patch("cautiousrobot.utils.os.path.exists", return_value=True)
4444
@patch("cautiousrobot.utils.gather_file_paths", return_value=["test_images/a.jpg", "test_images/b.jpg", "test_images/c.jpg"])
@@ -64,7 +64,7 @@ def test_no_files_exist(self, mock_print, mock_gather, mock_exists):
6464
self.assertFalse(any(updated_df["in_img_dir"]))
6565
self.assertEqual(len(filtered_df), len(self.sample_df))
6666
mock_print.assert_called_once()
67-
self.assertIn("There are 0 files", mock_print.call_args[0][0])
67+
self.assertIn("There are 0 of desired files", mock_print.call_args[0][0])
6868

6969
@patch("cautiousrobot.utils.os.path.exists", return_value=True)
7070
@patch("cautiousrobot.utils.gather_file_paths", return_value=["test_images/species1/a.jpg", "test_images/shouldnotcount/b.jpg"])

0 commit comments

Comments
 (0)