1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
|
#
# This file is part of screed, http://github.com/dib-lab/screed/, and is
# Copyright (C) Michigan State University, 2009-2015. It is licensed under
# the three-clause BSD license; see doc/LICENSE.txt.
# Contact: khmer-project@idyll.org
#
# This file has been modified from the khmer project at
# https://github.com/dib-lab/khmer/blob/a8356b7abbebf8540c7656378b1459442b781f87/tests/khmer_tst_utils.py
#
import tempfile
import os
import shutil
from io import StringIO
import sys
import traceback
from importlib import resources
# Remove when we drop support for 3.8
if sys.version_info < (3, 9):
import importlib_resources as resources
def get_test_data(filename):
filepath = resources.files('screed') / 'screed' / 'tests' / filename
if not filepath.exists() or not os.path.isfile(filepath):
filepath = os.path.join(os.path.dirname(__file__), 'test-data',
filename)
return filepath
cleanup_list = []
def get_temp_filename(filename, tempdir=None):
if tempdir is None:
tempdir = tempfile.mkdtemp(prefix='screedtest_')
cleanup_list.append(tempdir)
return os.path.join(tempdir, filename)
def cleanup():
global cleanup_list
for path in cleanup_list:
shutil.rmtree(path, ignore_errors=True)
cleanup_list = []
|
