test_geomedian.py 文件源码

def test_nangeomedian_axis_one_two_good():
    data = np.array([[1.0, np.nan, 1.0],
                     [2.0, 1.0, 1.0]])
    m = hd.nangeomedian(data, axis=1)
    r = np.nanmedian(data, axis=1)
    assert_array_almost_equal(m, r, decimal=3)