mirror of
https://github.com/kovidgoyal/kitty
synced 2026-07-27 18:51:41 +02:00
DRYer
This commit is contained in:
@@ -6,6 +6,7 @@ import io
|
|||||||
import os
|
import os
|
||||||
import select
|
import select
|
||||||
import shlex
|
import shlex
|
||||||
|
import shutil
|
||||||
import signal
|
import signal
|
||||||
import struct
|
import struct
|
||||||
import sys
|
import sys
|
||||||
@@ -179,6 +180,12 @@ class BaseTest(TestCase):
|
|||||||
maxDiff = 2048
|
maxDiff = 2048
|
||||||
is_ci = os.environ.get('CI') == 'true'
|
is_ci = os.environ.get('CI') == 'true'
|
||||||
|
|
||||||
|
def rmtree_ignoring_errors(self, tdir):
|
||||||
|
try:
|
||||||
|
shutil.rmtree(tdir)
|
||||||
|
except FileNotFoundError as err:
|
||||||
|
print('Failed to delete the directory:', tdir, 'with error:', err, file=sys.stderr)
|
||||||
|
|
||||||
def tearDown(self):
|
def tearDown(self):
|
||||||
set_options(None)
|
set_options(None)
|
||||||
|
|
||||||
|
|||||||
@@ -7,7 +7,7 @@ import shutil
|
|||||||
import stat
|
import stat
|
||||||
import tempfile
|
import tempfile
|
||||||
from collections import namedtuple
|
from collections import namedtuple
|
||||||
from contextlib import contextmanager, suppress
|
from contextlib import contextmanager
|
||||||
from pathlib import Path
|
from pathlib import Path
|
||||||
|
|
||||||
from kittens.transfer.rsync import Differ, Hasher, Patcher, parse_ftc
|
from kittens.transfer.rsync import Differ, Hasher, Patcher, parse_ftc
|
||||||
@@ -188,8 +188,7 @@ class TestFileTransmission(BaseTest):
|
|||||||
self.orig_home = os.environ.get('HOME')
|
self.orig_home = os.environ.get('HOME')
|
||||||
|
|
||||||
def tearDown(self):
|
def tearDown(self):
|
||||||
with suppress(FileNotFoundError):
|
self.rmtree_ignoring_errors(self.tdir)
|
||||||
shutil.rmtree(self.tdir)
|
|
||||||
self.responses = []
|
self.responses = []
|
||||||
if self.orig_home is None:
|
if self.orig_home is None:
|
||||||
os.environ.pop('HOME', None)
|
os.environ.pop('HOME', None)
|
||||||
|
|||||||
@@ -2,11 +2,9 @@
|
|||||||
# License: GPL v3 Copyright: 2017, Kovid Goyal <kovid at kovidgoyal.net>
|
# License: GPL v3 Copyright: 2017, Kovid Goyal <kovid at kovidgoyal.net>
|
||||||
|
|
||||||
import os
|
import os
|
||||||
import shutil
|
|
||||||
import sys
|
import sys
|
||||||
import tempfile
|
import tempfile
|
||||||
import unittest
|
import unittest
|
||||||
from contextlib import suppress
|
|
||||||
from functools import partial
|
from functools import partial
|
||||||
|
|
||||||
from kitty.constants import is_macos, read_kitty_resource
|
from kitty.constants import is_macos, read_kitty_resource
|
||||||
@@ -35,8 +33,7 @@ class Rendering(BaseTest):
|
|||||||
def tearDown(self):
|
def tearDown(self):
|
||||||
self.test_ctx.__exit__()
|
self.test_ctx.__exit__()
|
||||||
del self.sprites, self.cell_width, self.cell_height, self.test_ctx
|
del self.sprites, self.cell_width, self.cell_height, self.test_ctx
|
||||||
with suppress(FileNotFoundError):
|
self.rmtree_ignoring_errors(self.tdir)
|
||||||
shutil.rmtree(self.tdir)
|
|
||||||
super().tearDown()
|
super().tearDown()
|
||||||
|
|
||||||
def test_sprite_map(self):
|
def test_sprite_map(self):
|
||||||
|
|||||||
Reference in New Issue
Block a user