Advertisement
Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- # metoda w src.bin.model
- def find_newest_image_in_output_folder(path):
- list_dir_list = []
- for file in os.listdir(path):
- list_dir_list.append(file)
- print(f"Opened miniature of: {list_dir_list[-1]}")
- return list_dir_list[-1]
- # metoda w tests.test_model
- @mock.patch('src.bin.model.os.listdir')
- def test_find_newest_image_in_output_folder(self, mock_listdir_path):
- mock_listdir_path.return_value = ['123.jpg', '124.jpg', '125.jpg']
- list_dir_list = find_newest_image_in_output_folder('.')
- self.assertEqual('125.jpg', list_dir_list)
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement