test_utils.py 文件源码

def test_yarray():
    import numpy as np
    import astropy.units as u
    from ..utils import yarray, xarray

    y = yarray((3, 3))
    yy = np.array([[0, 0, 0], [1, 1, 1], [2, 2, 2]])
    assert np.all(y == yy)

    y = yarray((3, 3), yx=(0, 1))
    assert np.all(y == yy)

    y = yarray((3, 3), yx=(0.5, 0))
    yy = yy - 0.5
    assert np.all(y == yy)

    y = yarray((3, 3), th=-90 * u.deg)
    x = xarray((3, 3))
    assert np.allclose(y, x)