mirror of
https://github.com/wassname/scikit-image.git
synced 2026-06-28 20:24:18 +08:00
François Boulogne
24 lines
494 B
Python
24 lines
494 B
Python
import numpy as np
|
|
|
|
from skimage.transform import frt2, ifrt2
|
|
|
|
|
|
def test_frt():
|
|
SIZE = 59
|
|
try:
|
|
import sympy.ntheory as sn
|
|
assert sn.isprime(SIZE) == True
|
|
except ImportError:
|
|
pass
|
|
|
|
# Generate a test image
|
|
L = np.tri(SIZE, dtype=np.int32) + np.tri(SIZE, dtype=np.int32)[::-1]
|
|
f = frt2(L)
|
|
fi = ifrt2(f)
|
|
assert len(np.nonzero(L - fi)[0]) == 0
|
|
|
|
if __name__ == '__main__':
|
|
from numpy.testing import run_module_suite
|
|
run_module_suite()
|
|
|