feat(scripts): Fix pre-commit error 2nd try

pull/524/head
tessier 5 months ago
parent 50abea690a
commit 1183041a81

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

Loading…
Cancel
Save