meanas/meanas/test/utils.py

from typing import Any

import numpy
from numpy.typing import NDArray


PRNG = numpy.random.RandomState(12345)


def assert_fields_close(
        x: NDArray,
        y: NDArray,
        *args: Any,
        **kwargs: Any,
        ) -> None:
    numpy.testing.assert_allclose(
        x, y, verbose=False,            # type: ignore
        err_msg='Fields did not match:\n{}\n{}'.format(numpy.moveaxis(x, -1, 0),
                                                       numpy.moveaxis(y, -1, 0)),
        *args,
        **kwargs,
        )

def assert_close(
        x: NDArray,
        y: NDArray,
        *args: Any,
        **kwargs: Any,
        ) -> None:
    numpy.testing.assert_allclose(x, y, *args, **kwargs)