@@ -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