jan/fdfd_tools
2
2
Fork 1
Code Issues Pull Requests Releases Wiki Activity

add some nonuniform dxes

Browse Source 
'random' causes some tests to fail, haven't thought about why yet
This commit is contained in:
Jan Petykiewicz 2019-11-12 01:09:26 -08:00
parent d16de3802f
commit 462a8c6dbc
1 changed files with 10 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
meanas/test/conftest.py
View File

@ -62,9 +62,18 @@ def dx(request):
yield request.param
@pytest.fixture(scope='module', params=['uniform'])
@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