fdfd_tools/meanas/test/utils.py
Jan Petykiewicz 9253200eaf Move j_steps and j_distribution out of conftest
- also move PRNG back into utils
2019-11-04 20:27:22 -08:00

13 lines
447 B
Python

import numpy
PRNG = numpy.random.RandomState(12345)
def assert_fields_close(x, y, *args, **kwargs):
numpy.testing.assert_allclose(x, y, verbose=False,
err_msg='Fields did not match:\n{}\n{}'.format(numpy.rollaxis(x, -1),
numpy.rollaxis(y, -1)), *args, **kwargs)
def assert_close(x, y, *args, **kwargs):
numpy.testing.assert_allclose(x, y, *args, **kwargs)