stress test new fftn() call with real-valued input

72b854ae · Martin Reinecke · 7890f60c · 72b854ae
Commit 72b854ae authored Jun 18, 2019 by Martin Reinecke
Hide whitespace changes
Inline Side-by-side

Showing with 5 additions and 0 deletions

stress.py stress.py +5 -0

No files found.
--- a/stress.py
+++ b/stress.py
@@ -24,6 +24,11 @@ def test():
    a=np.random.rand(*shape)-0.5 + 1j*np.random.rand(*shape)-0.5j
    b=pypocketfft.ifftn(pypocketfft.fftn(a,axes=axes,nthreads=nthreads),axes=axes,inorm=2,nthreads=nthreads)
    err = _l2error(a,b)
+    if err > cmaxerr:
+        cmaxerr = err
+        print("cmaxerr:", cmaxerr, shape, axes)
+    b=pypocketfft.ifftn(pypocketfft.fftn(a.real,axes=axes,nthreads=nthreads),axes=axes,inorm=2,nthreads=nthreads)
+    err = _l2error(a.real,b)
    if err > cmaxerr:
        cmaxerr = err
        print("cmaxerr:", cmaxerr, shape, axes)