From 1183041a811ee4d450ed9c445b4aa9a109913cc3 Mon Sep 17 00:00:00 2001 From: tessier Date: Sun, 4 May 2025 00:47:48 +0900 Subject: [PATCH] feat(scripts): Fix pre-commit error 2nd try --- tests/test_cli.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/tests/test_cli.py b/tests/test_cli.py index 9d600d67..77759f59 100644 --- a/tests/test_cli.py +++ b/tests/test_cli.py @@ -1,10 +1,11 @@ import os import unittest -from unittest.mock import mock_open, patch from io import StringIO +from unittest.mock import mock_open, patch from diagrams.cli import run + class CliTest(unittest.TestCase): def setUp(self): self.test_file = "test_diagram.py" @@ -29,6 +30,7 @@ with Diagram("test_2", show=False, direction="TB"): EC2("worker4"), EC2("työntekijä 4")] >> RDS("events") """ + def tearDown(self): try: os.remove("test.png") @@ -47,7 +49,6 @@ with Diagram("test_2", show=False, direction="TB"): except FileNotFoundError: pass - def test_run_with_multiple_files(self): multiple_files = ["file1.py", "file2.py"]