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
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
|
from nose.tools import assert_raises, ok_
from pyensembl.download_cache import (
DownloadCache,
MissingLocalFile,
MissingRemoteFile
)
import os
import tempfile
from .data import data_path
download_cache = DownloadCache(
reference_name="__test_reference",
annotation_name="__test_annotation",
copy_local_files_to_cache=False)
def test_download_cache_missing_local_file():
# clear the cache
download_cache.delete_cache_directory()
with assert_raises(MissingLocalFile):
download_cache.download_or_copy_if_necessary(
path_or_url="test_file_doesn_not_exist.file")
def test_download_cache_missing_remote_file():
# clear the cache
download_cache.delete_cache_directory()
with assert_raises(MissingRemoteFile):
download_cache.download_or_copy_if_necessary(
path_or_url="ftp://NOTAURL.NOTAURL.NOTAURL")
def test_download_cache_custom_location():
test_file = "refseq.ucsc.small.gtf"
tmp_dir = tempfile.gettempdir()
print("DIR: %s" % tmp_dir)
assert tmp_dir is not None
os.environ['PYENSEMBL_CACHE_DIR'] = tmp_dir
# We need another instance of DownloadCache
# that copies files over to cache folder
download_cache = DownloadCache(
reference_name="test_reference",
annotation_name="test_annotation",
copy_local_files_to_cache=True)
# clean up
download_cache.delete_cache_directory()
download_cache.download_or_copy_if_necessary(
download_if_missing=True,
path_or_url=data_path(test_file))
full_path = os.path.join(
tmp_dir,
"pyensembl",
"test_reference",
"test_annotation",
test_file)
print("FULL PATH: %s" % full_path)
assert len(full_path) > 0
ok_(os.path.exists(full_path))
del os.environ['PYENSEMBL_CACHE_DIR']
|
