You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
80 lines
2.4 KiB
Python
80 lines
2.4 KiB
Python
"""
|
|
|
|
Test fixtures
|
|
|
|
"""
|
|
import numpy # type: ignore
|
|
import pytest # type: ignore
|
|
|
|
from .utils import PRNG
|
|
|
|
@pytest.fixture(scope='module',
|
|
params=[(5, 5, 1),
|
|
(5, 1, 5),
|
|
(5, 5, 5),
|
|
# (7, 7, 7),
|
|
])
|
|
def shape(request):
|
|
yield (3, *request.param)
|
|
|
|
|
|
@pytest.fixture(scope='module', params=[1.0, 1.5])
|
|
def epsilon_bg(request):
|
|
yield request.param
|
|
|
|
|
|
@pytest.fixture(scope='module', params=[1.0, 2.5])
|
|
def epsilon_fg(request):
|
|
yield request.param
|
|
|
|
|
|
@pytest.fixture(scope='module', params=['center', '000', 'random'])
|
|
def epsilon(request, shape, epsilon_bg, epsilon_fg):
|
|
is3d = (numpy.array(shape) == 1).sum() == 0
|
|
if is3d:
|
|
if request.param == '000':
|
|
pytest.skip('Skipping 000 epsilon because test is 3D (for speed)')
|
|
if epsilon_bg != 1:
|
|
pytest.skip('Skipping epsilon_bg != 1 because test is 3D (for speed)')
|
|
if epsilon_fg not in (1.0, 2.0):
|
|
pytest.skip('Skipping epsilon_fg not in (1, 2) because test is 3D (for speed)')
|
|
|
|
epsilon = numpy.full(shape, epsilon_bg, dtype=float)
|
|
if request.param == 'center':
|
|
epsilon[:, shape[1] // 2, shape[2] // 2, shape[3] // 2] = epsilon_fg
|
|
elif request.param == '000':
|
|
epsilon[:, 0, 0, 0] = epsilon_fg
|
|
elif request.param == 'random':
|
|
epsilon[:] = PRNG.uniform(low=min(epsilon_bg, epsilon_fg),
|
|
high=max(epsilon_bg, epsilon_fg),
|
|
size=shape)
|
|
|
|
yield epsilon
|
|
|
|
|
|
@pytest.fixture(scope='module', params=[1.0]) # 1.5
|
|
def j_mag(request):
|
|
yield request.param
|
|
|
|
|
|
@pytest.fixture(scope='module', params=[1.0, 1.5])
|
|
def dx(request):
|
|
yield request.param
|
|
|
|
|
|
@pytest.fixture(scope='module', params=['uniform', 'centerbig'])
|
|
def dxes(request, shape, dx):
|
|
if request.param == 'uniform':
|
|
dxes = [[numpy.full(s, dx) for s in shape[1:]] for _ in range(2)]
|
|
elif request.param == 'centerbig':
|
|
dxes = [[numpy.full(s, dx) for s in shape[1:]] for _ in range(2)]
|
|
for eh in (0, 1):
|
|
for ax in (0, 1, 2):
|
|
dxes[eh][ax][dxes[eh][ax].size // 2] *= 1.1
|
|
elif request.param == 'random':
|
|
dxe = [PRNG.uniform(low=1.0 * dx, high=1.1 * dx, size=s) for s in shape[1:]]
|
|
dxh = [(d + numpy.roll(d, -1)) / 2 for d in dxe]
|
|
dxes = [dxe, dxh]
|
|
yield dxes
|
|
|