test_utils.py 文件源码

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

    x = xarray((3, 3))
    xx = np.array([[0, 1, 2], [0, 1, 2], [0, 1, 2]])
    assert np.all(x == xx)

    x = xarray((3, 3), yx=(1, 0))
    assert np.all(x == xx)

    x = xarray((3, 3), yx=(0, 0.5))
    xx = xx - 0.5
    assert np.all(x == xx)

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