test_crosscorr.py 文件源码

def test_fit_3d(eng):
    reference = arange(125).reshape(5, 5, 5)
    algorithm = CrossCorr()
    deltas = [[1, 0, 2], [0, 1, 2]]
    shifted = [shift(reference, delta, mode='wrap', order=0) for delta in deltas]
    model = algorithm.fit(shifted, reference=reference)
    assert allclose(model.toarray(), deltas)