test_crosscorr.py 文件源码

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