remove 'contextlib' imports

This commit is contained in:
Mike Fährmann
2024-04-06 03:43:12 +02:00
parent 9a8403917a
commit 40bd145637
2 changed files with 10 additions and 4 deletions

View File

@@ -13,7 +13,6 @@ import unittest
from unittest.mock import patch
import io
import contextlib
sys.path.insert(0, os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
from gallery_dl import job, config, text # noqa E402
@@ -32,8 +31,13 @@ class TestJob(unittest.TestCase):
jobinstance = extr_or_job
with io.StringIO() as buffer:
with contextlib.redirect_stdout(buffer):
stdout = sys.stdout
sys.stdout = buffer
try:
jobinstance.run()
finally:
sys.stdout = stdout
return buffer.getvalue()