diff --git a/tests/test_handler.py b/tests/test_handler.py index 17304ce..6299d99 100644 --- a/tests/test_handler.py +++ b/tests/test_handler.py @@ -12,7 +12,7 @@ def test_execute_command_generate_number_success(self): "command": "generate_number", "text": "5" } - expected_file_path = "./data/nums/drawn_number_5.png" + expected_file_path = "data/nums/drawn_number_5.png" handler = Handler(MagicMock()) # Act @@ -21,10 +21,9 @@ def test_execute_command_generate_number_success(self): # Assert self.assertTrue(os.path.exists(expected_file_path)) - # Clean up + # Cleanup os.remove(expected_file_path) - def test_execute_command_generate_number_error(self): # Arrange request = {