test_crosscorr.py 文件源码

def test_fit_axis(eng):
    reference = arange(60).reshape(2, 5, 6)
    algorithm = CrossCorr(axis=0)
    a = shift(reference[0], [1, 2], mode='wrap', order=0)
    b = shift(reference[1], [-2, 1], mode='wrap', order=0)
    c = shift(reference[0], [2, 1], mode='wrap', order=0)
    d = shift(reference[1], [1, -2], mode='wrap', order=0)
    shifted = [asarray([a, b]), asarray([c, d]),]
    model = algorithm.fit(shifted, reference=reference)
    assert allclose(model.toarray(), [[[1, 2], [-2, 1]], [[2, 1], [1, -2]]])