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
|
import os
from os import path as op
from mne import datasets
from mne.externals.six import string_types
from mne.utils import _TempDir, run_tests_if_main, requires_good_network
def test_datasets():
"""Test simple dataset functions."""
# XXX 'hf_sef' and 'misc' do not conform to these standards
for dname in ('sample', 'somato', 'spm_face', 'testing', 'opm',
'bst_raw', 'bst_auditory', 'bst_resting', 'multimodal',
'bst_phantom_ctf', 'bst_phantom_elekta', 'kiloword',
'mtrf', 'phantom_4dbti',
'visual_92_categories', 'fieldtrip_cmc'):
if dname.startswith('bst'):
dataset = getattr(datasets.brainstorm, dname)
check_name = 'brainstorm.%s' % (dname,)
else:
dataset = getattr(datasets, dname)
check_name = dname
if dataset.data_path(download=False) != '':
assert isinstance(dataset.get_version(), string_types)
assert datasets.utils.has_dataset(check_name)
else:
assert dataset.get_version() is None
assert not datasets.utils.has_dataset(check_name)
print('%s: %s' % (dname, datasets.utils.has_dataset(check_name)))
tempdir = _TempDir()
# don't let it read from the config file to get the directory,
# force it to look for the default
os.environ['_MNE_FAKE_HOME_DIR'] = tempdir
try:
assert (datasets.utils._get_path(None, 'foo', 'bar') ==
op.join(tempdir, 'mne_data'))
finally:
del os.environ['_MNE_FAKE_HOME_DIR']
@requires_good_network
def test_megsim():
"""Test MEGSIM URL handling."""
data_dir = _TempDir()
paths = datasets.megsim.load_data(
'index', 'text', 'text', path=data_dir, update_path=False)
assert len(paths) == 1
assert paths[0].endswith('index.html')
@requires_good_network
def test_downloads():
"""Test dataset URL handling."""
# Try actually downloading a dataset
data_dir = _TempDir()
path = datasets._fake.data_path(path=data_dir, update_path=False)
assert op.isfile(op.join(path, 'bar'))
assert datasets._fake.get_version() is None
run_tests_if_main()
|
