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
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
|
"""
Module containing fixtures and hooks used by the pytest test suite.
"""
import re
import numpy as np
import pytest
from brian2.core.clocks import defaultclock
from brian2.core.functions import DEFAULT_FUNCTIONS, Function
from brian2.devices import get_device, reinit_devices
from brian2.devices.device import reinit_and_delete, set_device
from brian2.units import ms
def pytest_ignore_collect(collection_path, config):
if config.option.doctestmodules:
if "tests" in collection_path.parts:
return True # Ignore tests package for doctests
# Do not test brian2.hears bridge (needs Brian1)
if collection_path.name == "hears.py":
return True
# The "random" values are always 0.5
def fake_randn(vectorisation_idx):
return 0.5 * np.ones_like(vectorisation_idx)
fake_randn = Function(
fake_randn, arg_units=[], return_unit=1, auto_vectorise=True, stateless=False
)
fake_randn.implementations.add_implementation(
"cpp",
"""
double randn(int vectorisation_idx)
{
return 0.5;
}
""",
)
fake_randn.implementations.add_implementation(
"cython",
"""
cdef double randn(int vectorisation_idx):
return 0.5
""",
)
@pytest.fixture
def fake_randn_randn_fixture():
orig_randn = DEFAULT_FUNCTIONS["randn"]
DEFAULT_FUNCTIONS["randn"] = fake_randn
yield None
DEFAULT_FUNCTIONS["randn"] = orig_randn
# Fixture that is used for all tests
@pytest.fixture(autouse=True)
def setup_and_teardown(request):
# Set preferences before each test
import brian2
if hasattr(request.config, "workerinput"):
config = request.config.workerinput
for key, value in config["brian_prefs"].items():
if isinstance(value, tuple) and value[0] == "TYPE":
matches = re.match(r"<(type|class) 'numpy\.(.+)'>", value[1])
if matches is None or len(matches.groups()) != 2:
raise TypeError(
f"Do not know how to handle {value[1]} in preferences"
)
t = matches.groups()[1]
if t == "float64":
value = np.float64
elif t == "float32":
value = np.float32
elif t == "int64":
value = np.int64
elif t == "int32":
value = np.int32
brian2.prefs[key] = value
set_device(config["device"], **config["device_options"])
else:
for k, v in request.config.brian_prefs.items():
brian2.prefs[k] = v
set_device(request.config.device, **request.config.device_options)
brian2.prefs._backup()
# Print output changed in numpy 1.14, stick with the old format to
# avoid doctest failures
try:
np.set_printoptions(legacy="1.13")
except TypeError:
pass # using a numpy version < 1.14
yield # run test
# Reset defaultclock.dt to be sure
defaultclock.dt = 0.1 * ms
# (Optionally) mark tests raising NotImplementedError as skipped (mostly used
# for testing Brian2GeNN)
@pytest.hookimpl(hookwrapper=True)
def pytest_runtest_makereport(item, call):
if hasattr(item.config, "workerinput"):
fail_for_not_implemented = item.config.workerinput["fail_for_not_implemented"]
else:
fail_for_not_implemented = item.config.fail_for_not_implemented
outcome = yield
rep = outcome.get_result()
if rep.outcome == "failed":
project_dir = getattr(get_device(), "project_dir", None)
if project_dir is not None:
rep.sections.append(("Standalone project directory", f"{project_dir}"))
reinit_devices()
if not fail_for_not_implemented:
exc_cause = getattr(call.excinfo.value, "__cause__", None)
if call.excinfo.errisinstance(NotImplementedError) or isinstance(
exc_cause, NotImplementedError
):
rep.outcome = "skipped"
r = call.excinfo._getreprcrash()
rep.longrepr = (str(r.path), r.lineno, r.message)
else:
# clean up after the test (delete directory for standalone)
reinit_and_delete()
|
