meanas/meanas/test/utils.py

from typing import Any
import numpy        # type: ignore


PRNG = numpy.random.RandomState(12345)


def assert_fields_close(x: numpy.ndarray,
                        y: numpy.ndarray,
                        *args: Any,
                        **kwargs: Any,
                        ) -> None:
    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: numpy.ndarray,
                 y: numpy.ndarray,
                 *args: Any,
                 **kwargs: Any,
                 ) -> None:
    numpy.testing.assert_allclose(x, y, *args, **kwargs)