+
    ʇi~_                     >   ^ RI t ^ RIt^ RIt^ RIt^ RIHt ^ RIHtH	t	H
t
Ht R t ! R R4      t ! R R4      t]P                  P!                  R	]P"                  ]P$                  ]P&                  ]P(                  .4      ]P                  P!                  R
RR.4      ]P                  P!                  R]P*                  P*                  ]P*                  P,                  ]P*                  P.                  ]P*                  P0                  ]P*                  P2                  ]P*                  P4                  .4      R 4       4       4       t]P                  P!                  R
RR.4      ]P                  P!                  R. RO4      R 4       4       t]P                  P;                  ]RR7       ! R R4      4       tR tR t ]P                  P!                  R]P*                  P*                  ]P*                  P0                  ]P*                  PB                  ]P*                  PD                  .4      ]P                  P!                  R]PF                  ! . RO4      ]PH                  ! ^
]PJ                  R7      ]PH                  ! ^]PL                  R7      .4      R 4       4       t'R# )    N)random)IS_WASMassert_allcloseassert_array_equalassert_raisesc                 J   \        V 4      pR\        P                  ,          \        P                  ! V4      V,          ,          p\        P                  ! V4      P	                  R^4      V,          p\        P
                  ! V \        P                  ! V4      ,          ^R7      # )y               @axisy              )lennppiarangereshapesumexp)xLphases   &  s/Users/max/.openclaw/workspace/postharvest/merge_env/lib/python3.14/site-packages/numpy/fft/tests/test_pocketfft.pyfft1r      sg    AA"%%K299Q<!+,EIIaL  Q'%/E66!bffUm#!,,    c                   &   a  ] tR t^t o R tRtV tR# )TestFFTShiftc                ^    \        \        \        P                  P                  . RO^ 4       R# )   N)r         )r   
ValueErrorr   fft)selfs   &r   
test_fft_nTestFFTShift.test_fft_n   s    j"&&**i;r    N)__name__
__module____qualname____firstlineno__r"   __static_attributes____classdictcell____classdict__s   @r   r   r      s     < <r   r   c            	       V	  a  ] tR t^t o R t]P                  P                  R]P                  ]P                  ]P                  .4      R 4       t]P                  P                  R]P                  ]P                  ]P                  .4      R 4       tR t]P                  P                  RR*4      ]P                  P                  R]]34      ]P                  P                  RR+4      R 4       4       4       t]P                  P                  R^ ^.4      R	 4       tR
 t]P                  P                  RR,4      R 4       tR tR tR tR tR tR tR tR tR tR tR tR t R t!R t"]P                  P                  R]PF                  PH                  ]PF                  PJ                  ]PF                  PL                  ]PF                  PN                  .4      R 4       t(]P                  P                  R]PF                  PH                  ]PF                  PJ                  ]PF                  PR                  ]PF                  PT                  .4      R 4       t+]P                  P                  R]PF                  PH                  ]PF                  PJ                  ]PF                  PL                  ]PF                  PN                  .4      R 4       t,]P                  P                  R]PF                  PR                  ]PF                  PT                  .4      R  4       t-]P                  P                  R]PF                  PH                  ]PF                  PJ                  ]PF                  PL                  ]PF                  PN                  ]PF                  PR                  ]PF                  PT                  .4      R! 4       t.R" t/]P                  P                  R#. R-O4      ]P                  P                  R]]34      ]P                  P                  RR+4      R$ 4       4       4       t0]P                  P                  R%]PF                  PH                  ]PF                  PJ                  ]PF                  PL                  .4      R& 4       t1]P                  P                  R'R.R/.4      R( 4       t2R)t3V t4R# )0	TestFFT1Dc           
        R p\        V4      R\        V4      ,          ,           p\        V4      p\        ^V4       F  p\        \        P                  P                  \        P                  P	                  V^ V 4      4      V^ V RR7       \        \        P                  P                  \        P                  P                  V^ V 4      V4      V^ V RR7       K  	  R# )i                 ?g-q=atolN)r   ranger   r   r    ifftirfftrfft)r!   maxlenr   xris   &    r   test_identityTestFFT1D.test_identity   s    6NR&.00F^q&!ABFFKK

1Qq6(:;QqV!&(BFFLLR!W)=qAqG%1 "r   dtypec                   ^p^\         P                  ! \         P                  ! RVR7      4      ,          p\        V4      P	                  V4      R\        V4      P	                  V4      ,          ,           p\         P
                  ! V\         P                  ! V4      .4      p\        V4      P	                  V4      p\         P
                  ! V\         P                  ! V4      .4      p\        ^V^,          4       F  p\         P                  P                  \         P                  P                  WHR7      VR7      p	V	P                  P                  V8X  g   Q h\        W^ V V^ R7       \         P                  P                  \         P                  P                  WhR7      VR7      p
V
P                  V8X  g   Q h\        W^ V V^ R7       K  	  R#          ?r<   r0   nr2   rtolN)r   spacingarrayr   astypeconcatenate
zeros_liker3   r    r4   realr<   r   r5   r6   )r!   r<   r7   r2   r   xxr8   xxrr9   check_ccheck_rs   &&         r   test_identity_long_short"TestFFT1D.test_identity_long_short$   sM    2::bhhr7886N!!%(2v0E0Ee0L+LL^^Qa 012F^""5)nnb"--"345q&1*%Affkk"&&**Q*"4k:G<<%%...G!W4a@ffll266;;r;#71l=G==E)))G1XDqA &r   c                2   ^p^\         P                  ! \         P                  ! RVR7      4      ,          p\        V4      P	                  V4      R\        V4      P	                  V4      ,          ,           p\         P
                  ! V\         P                  ! V4      .4      p\        ^V^,          4       EF]  p\         P                  P                  \         P                  P                  WFR7      VR7      pVP                  VP                  8X  g   Q h\        Wu^ V V^ R7       VP                  4       pV^,          ^,           p	^ VP                  ^ &   V^,          ^ 8X  d   ^ VP                  V	^,
          R% \         P
                  ! V\         P                  ! V4      .4      p
\         P                  P                  \         P                  P                  WFR7      VR7      pVP                  VP                  8X  g   Q h\        W^ V	 V^ R7       EK`  	  R# r>   )r   rF   rG   r   rH   rI   rJ   r3   r    r4   r<   r   copyimagr6   r5   )r!   r<   r7   r2   r   rL   r9   check_via_cyrC   yycheck_via_rs   &&          r   !test_identity_long_short_reversed+TestFFT1D.test_identity_long_short_reversed6   s    2::bhhr7886N!!%(2v0E0Ee0L+LL^^Qa 012q&1*%A&&**RVV[[[%8A*>K$$///KAat!D AQ
AAFF1I1uz!"q1uvBMM!$4 56B&&++bffll1l&:a+@K$$///KAat!DD &r   c                :   \        ^4      R\        ^4      ,          ,           p\        \        V4      \        P                  P	                  V4      RR7       \        \        V4      \        P                  P	                  VRR7      RR7       \        \        V4      \        P
                  ! ^4      ,          \        P                  P	                  VRR7      RR7       \        \        V4      R,          \        P                  P	                  VRR7      RR7       R	# )
   r0   ư>r1   backwardnormortho      >@forwardN)r   r   r   r   r    sqrtr!   r   s   & r   test_fftTestFFT1D.test_fftN   s    2JfRj(QAT:QAJ!?dKQ"''"+-

17
3$	@Q#

19
5D	Br   r
   	transposec                  a V3R  lpV\         J dW   \        R4      R\        R4      ,          ,           p\        P                  P                  \        P                  P                  rvM>\        R4      p\        P                  P
                  \        P                  P                  rvV! WSR7      pV! V4      p	V! WYVR7      p
WJ g   Q h\        W4       V! WR7      pV\         J d   T	MV! V4      pV! WVR7      pWJ g   Q h\        W4       R# )c                    < S'       d+   \         P                  ! V P                  4      P                  # \         P                  ! V 4      # Nr   rJ   Tr   rh   s   &r   rJ   3TestFFT1D.test_fft_out_argument.<locals>.zeros_like[   .    }}QSS)+++}}Q''r   r0   r	   )outr
   N)
      )complexr   r   r    r4   r6   r5   r   )r!   r<   rh   r
   rJ   rV   r    r4   expectedrq   result	expected2out2result2s   &&f&          r   test_fft_out_argumentTestFFT1D.test_fft_out_argumentW   s    	( Gx 2x(8#88A

BFFKKx ARVV\\q$"Qd+}}6,-	w&sJy,As407.r   c                   \        R4      R\        R4      ,          ,           pVP                  4       p\        P                  P                  W1R7      p\        P                  P                  W1VR7      pWSJ g   Q h\	        WT4       VP                  4       pV^ 8X  d
   VR,          MVR,          p\        P                  P                  V^
VR7      p\        P                  P                  V^
WR7      p	WJ g   Q h\	        W4       V^ 8X  d   \	        VR,          VR,          4       M\	        VR,          VR,          4       VP                  4       p
V^ 8X  d
   V
R	,          MV
R,          pV^ 8X  d
   V
R
,          MV
R,          p\        P                  P                  V^
VR7      p\        P                  P                  V^
WR7      pWJ g   Q h\	        W4       V^ 8X  d6   \	        V
R,          VR,          4       \	        V
R,          VR,          4       M4\	        V
R,          VR,          4       \	        V
R,          VR,          4       VP                  4       pV^ 8X  d
   VR,          MVR,          pV^ 8X  d
   VR,          MVR,          p\        P                  P                  V^VR7      p\        P                  P                  V^VVR7      pVVJ g   Q h\	        VV4       V^ 8X  d   \	        VR,          VR,          4       M\	        VR,          VR,          4       VP                  4       pVP
                  p\        P                  P                  VVVR7      pVVJ g   Q h\	        VV4       VP                  4       pV^ 8X  d   VRRR1,          MVRRRR13,          p\        P                  P                  VVVR7      pVVJ g   Q h\	        VV4       R# )rs   r0   r	   )r
   rq   Nrr   NNNN)rC   r
   )rC   r
   rq   rr   NN   NNr      NNr   Nr   NNNr   NN)rs   rs   )r~   r}   )r~   r   )r~   r   )r~   r   )r~   r   )r~   r   )r~   r   r   )r   rS   r   r    r   rm   )r!   r
   rV   y1	expected1result1y2rx   rw   ry   y3y3_selout3	expected3result3y4y4_selout4	expected4result4y5out5result5y6out6result6s   &&                        r   test_fft_inplace_outTestFFT1D.test_fft_inplace_outu   s    8rF8$444VVXFFJJrJ-	&&**R*3}}7.VVX!)r#wFFFJJrRdJ3	&&**R2D*;7.19r#w#/r&z1V95VVX19B"U)19r$x"W+FFJJv$J7	&&**Vr*?7.19r"vqu-r#w#/r%y!E(3r&z1V95VVX AIC2f:!)r#wFFFJJv$J7	&&**Vr$*?$7I.19r#w#/r&z1V95VVXtt&&**Rd*5$7I.VVX19r$B$x"Q"W+&&**Rd*5$7I.r   c           
        \         P                  ! R 4      p\        P                  ! \        RR7      ;_uu_ 4        \         P
                  P                  VRR7       RRR4       \        P                  ! \        RR7      ;_uu_ 4        \         P
                  P                  V\         P                  ! V4      P                  ^R	4      R7       RRR4       \        P                  ! \        RR7      ;_uu_ 4        \         P
                  P                  V\         P                  ! V\        R7      R7       RRR4       R#   + '       g   i     L; i  + '       g   i     L; i  + '       g   i     R# ; i)
rb   zmust be of ArrayTypematch )rq   Nhas wrong shapezCannot castrA   r   )
r   r   pytestraises	TypeErrorr    r   rJ   r   floatre   s   & r   test_fft_bad_outTestFFT1D.test_fft_bad_out   s    IIcN]]9,BCCFFJJqbJ! D]]:->??FFJJqbmmA.66q"=J> @]]9M::FFJJqbmmAU;J< ;:	 DC??:::s$   "E	AE;<E'E	E$	'E8	r`   Nc           
        \        ^4      R\        ^4      ,          ,           p\        V\        P                  P	                  \        P                  P                  W!R7      VR7      RR7       \
        P                  ! \        RR7      ;_uu_ 4        \        P                  P	                  . VR7       RRR4       R#   + '       g   i     R# ; i)r\   r0   r_   r]   r1   z!Invalid number of FFT data pointsr   N)r   r   r   r    r4   r   r   r   )r!   r`   r   s   && r   	test_ifftTestFFT1D.test_ifft   s    2JfRj(rvv{{266::a:3${?	 ]]:!DF FFFKKK&F F F Fs   "B>>C	c           	        \        R
4      R\        R
4      ,          ,           p\        \        P                  P                  \        P                  P                  V^R7      ^ R7      \        P                  P	                  V4      RR7       \        \        P                  P	                  V4      \        P                  P	                  VRR7      RR7       \        \        P                  P	                  V4      \        P
                  ! R4      ,          \        P                  P	                  VRR7      RR7       \        \        P                  P	                  V4      R,          \        P                  P	                  VRR7      RR7       R	# r\   r0   r	   r]   r1   r^   r_   ra   rc   Nr\   rs   X       @)r   r   r   r    fft2rd   re   s   & r   	test_fft2TestFFT1D.test_fft2   s    8rF8$444

266::aa:#8q
AAT	3AAJ7d	DA)99AG44	AA)4AI6T	Cr   c           	        \        R
4      R\        R
4      ,          ,           p\        \        P                  P	                  \        P                  P	                  V^R7      ^ R7      \        P                  P                  V4      RR7       \        \        P                  P                  V4      \        P                  P                  VRR7      RR7       \        \        P                  P                  V4      \        P                  ! R4      ,          \        P                  P                  VRR7      RR7       \        \        P                  P                  V4      R,          \        P                  P                  VRR7      RR7       R	# r   )r   r   r   r    r4   ifft2rd   re   s   & r   
test_ifft2TestFFT1D.test_ifft2   s    8rF8$444BFFKKK$:CQd	4QQZ8t	EQ"'''*::QW5D	BQ95QY7d	Dr   c                Z   \        R
4      R\        R
4      ,          ,           p\        \        P                  P                  \        P                  P                  \        P                  P                  V^R7      ^R7      ^ R7      \        P                  P	                  V4      RR7       \        \        P                  P	                  V4      \        P                  P	                  VRR7      RR7       \        \        P                  P	                  V4      \        P
                  ! R4      ,          \        P                  P	                  VRR7      RR7       \        \        P                  P	                  V4      R,          \        P                  P	                  VRR7      RR7       R	# r\   r0   r	   r]   r1   r^   r_   ra   rc   Nr\   rs   rr   p       p@)r   r   r   r    fftnrd   re   s   & r   	test_fftnTestFFT1D.test_fftn   s   < 2|(<#<<FFJJrvvzz"&&**QQ*"7az@qJIFFKKN	' 	AAJ7d	DA)>>AG44	AA/:AI6T	Cr   c                Z   \        R
4      R\        R
4      ,          ,           p\        \        P                  P	                  \        P                  P	                  \        P                  P	                  V^R7      ^R7      ^ R7      \        P                  P                  V4      RR7       \        \        P                  P                  V4      \        P                  P                  VRR7      RR7       \        \        P                  P                  V4      \        P                  ! R4      ,          \        P                  P                  VRR7      RR7       \        \        P                  P                  V4      R,          \        P                  P                  VRR7      RR7       R	# r   )r   r   r   r    r4   ifftnrd   re   s   & r   
test_ifftnTestFFT1D.test_ifftn   s   < 2|(<#<<FFKKBFFKKK$:C!KLFFLLO$	( 	QQZ8t	EQ"'',*??QW5D	BQ?;QY7d	Dr   c                >   \        ^4      pVP                  ^VP                  ,          3 EFm  pR	 F`  p\        \        P                  P	                  WVR7      RV^,          ^,            \        P                  P                  WVR7      RR7       Kb  	  \        \        P                  P                  WR7      \        P                  P                  WRR7      RR7       \        \        P                  P                  WR7      \        P                  ! V4      ,          \        P                  P                  WRR7      RR7       \        \        P                  P                  WR7      V,          \        P                  P                  WRR7      RR7       EKp  	  R# )
r\   Nr^   ra   rc   rC   r`   r]   r1   rB   Nr^   ra   rc   )r   sizer   r   r    r6   rd   )r!   r   rC   r`   s   &   r   	test_rfftTestFFT1D.test_rfft   s!   2J&&!aff*%A>FFJJqDJ1-AFQJ@FFKKTK2? ? A#A44A A#bggaj0A1> A#a'A3$@ @ &r   c                    \         P                  ! ^4      p^p\         P                  P                  W4      p\	        V\         P                  P                  VRV 4      RV^,          ^,            RR7       R# )   N+=rE   )r   r   r    r6   r   )r!   r   rC   rV   s   &   r   test_rfft_evenTestFFT1D.test_rfft_even  sT    IIaLFFKK266::ae,[a1fqj9Fr   c                    \         P                  ! . RO4      p\         P                  P                  V4      p\	        V\         P                  P                  V4      R,          RR7       R# )r   :Nr   Nr   r   N)r   r   r   r   )r   rG   r    r6   r   )r!   r   rV   s   &  r   test_rfft_oddTestFFT1D.test_rfft_odd
  s?    HH%&FFKKN266::a=,59r   c           
     t   \        ^4      p\        V\        P                  P	                  \        P                  P                  V4      4      RR7       \        V\        P                  P	                  \        P                  P                  VRR7      RR7      RR7       \        V\        P                  P	                  \        P                  P                  VRR7      RR7      RR7       \        V\        P                  P	                  \        P                  P                  VRR7      RR7      RR7       R# )r\   r]   r1   r^   r_   ra   rc   N)r   r   r   r    r5   r6   re   s   & r   
test_irfftTestFFT1D.test_irfft  s    2J266<<A7dC266<<AJ(G' ( )/3	5266<<AG(D$ ( &,0	2266<<AI(F& ( (.2	4r   c                   \        R4      p\        \        P                  P	                  V4      R	,          \        P                  P                  V4      RR7       \        \        P                  P                  V4      \        P                  P                  VRR7      RR7       \        \        P                  P                  V4      \        P                  ! R
4      ,          \        P                  P                  VRR7      RR7       \        \        P                  P                  V4      R,          \        P                  P                  VRR7      RR7       R# )r\   r]   r1   r^   r_   ra   rc   Nr   )r~   :N   Nr   r   )r   r   r   r    r   rfft2rd   re   s   & r   
test_rfft2TestFFT1D.test_rfft2  s    8Av.QdKQQZ8t	EQ"'''*::QW5D	BQ95QY7d	Dr   c           
     t   \        R4      p\        V\        P                  P	                  \        P                  P                  V4      4      RR7       \        V\        P                  P	                  \        P                  P                  VRR7      RR7      RR7       \        V\        P                  P	                  \        P                  P                  VRR7      RR7      RR7       \        V\        P                  P	                  \        P                  P                  VRR7      RR7      RR7       R# )	r\   r]   r1   r^   r_   ra   rc   Nr   )r   r   r   r    irfft2r   re   s   & r   test_irfft2TestFFT1D.test_irfft2#  s    8266==a9E266==aj)I' ) )/3	5266==ag)F$ ) &,0	2266==ai)H& ) (.2	4r   c           
     T   \        R
4      p\        \        P                  P	                  V4      R,          \        P                  P                  V4      RR7       \        \        P                  P                  V4      \        P                  P                  VRR7      RR7       \        \        P                  P                  V4      \        P                  ! R4      ,          \        P                  P                  VRR7      RR7       \        \        P                  P                  V4      R,          \        P                  P                  VRR7      RR7       \        P                  ! R4      p\        P                  P                  VRRR7      pVP                  R8X  g   Q h\        P                  P                  \        P                  P                  \        P                  P                  V^^(R7      ^ ^R7      ^ ^
R7      pVP                  R8X  g   Q h\        W#RR7       R	# )r\   r]   r1   r^   r_   ra   rc   )axess)r
   rC   Nr   )r~   r~   :N   Nr   r   )r   r   )r   r   r   )rr   rs   (   )rr      )
r   r   r   r    r   rfftnrd   onesshaper6   )r!   r   rv   ru   s   &   r   
test_rfftnTestFFT1D.test_rfftn-  sn   < Ax0"&&,,q/MQQZ8t	EQ"'',*??QW5D	BQ?;QY7d	D GGFOai<@||x'''66::bffjjQQ")E!"b ) *01R  9~~)))t4r   c           
     t   \        R4      p\        V\        P                  P	                  \        P                  P                  V4      4      RR7       \        V\        P                  P	                  \        P                  P                  VRR7      RR7      RR7       \        V\        P                  P	                  \        P                  P                  VRR7      RR7      RR7       \        V\        P                  P	                  \        P                  P                  VRR7      RR7      RR7       R# )	r\   r]   r1   r^   r_   ra   rc   Nr   )r   r   r   r    irfftnr   re   s   & r   test_irfftnTestFFT1D.test_irfftn?  s    < 266==a9E266==aj)I' ) )/3	5266==ag)F$ ) &,0	2266==ai)H& ) (.2	4r   c                   \        ^4      R\        ^4      ,          ,           p\        P                  ! \        ^4      V\        ^4      34      p\        P                  ! W!RRR
1,          P                  4       34      p\	        \        P
                  P                  V4      \        P
                  P                  V4      RR7       \	        \        P
                  P                  V4      \        P
                  P                  VRR7      RR7       \	        \        P
                  P                  V4      \        P                  ! ^4      ,          \        P
                  P                  VRR7      RR7       \	        \        P
                  P                  V4      R,          \        P
                  P                  VR	R7      RR7       R# )   r0   Nr]   r1   r^   r_   ra   rb   rc   r   )r   r   rI   conjr   r    hfftrd   r!   r   x_herms   &  r   	test_hfftTestFFT1D.test_hfftI  s   2JfRj(Avay 9:NNFddGLLN34

1rvv{{6':FF+F<4	IF+bggbk9F9	FF+c1F;$	Hr   c           
     V   \        ^4      R\        ^4      ,          ,           p\        P                  ! \        ^4      V\        ^4      34      p\        P                  ! W!RRR	1,          P                  4       34      p\	        V\        P
                  P                  \        P
                  P                  V4      4      RR7       \	        V\        P
                  P                  \        P
                  P                  VRR7      RR7      RR7       \	        V\        P
                  P                  \        P
                  P                  VRR7      RR7      RR7       \	        V\        P
                  P                  \        P
                  P                  VRR7      RR7      RR7       R# )
r   r0   Nr]   r1   r^   r_   ra   rc   r   )r   r   rI   r   r   r    ihfftr   r   s   &  r   
test_ihfftTestFFT1D.test_ihfftU  s4   2JfRj(Avay 9:NNFddGLLN34RVV[[-@ AMRVV[[' .9 .)/9 !- !;AE	GRVV[[$ .9 .&,3 !- !5;?	ARVV[[& .9 .(.7 !- !9?C	Er   opc                    \        R4      p. ROpV FJ  pV! \        P                  ! W$4      4      p\        P                  ! V! W$R7      V4      p\        WVRR7       KL  	  R# )r\   r   r]   r1   Nr   )r   r   r   )r   r   r   )r   r   r   )r   r   r   )r   r   r   )r   r   r   )r   r   rh   r   )r!   r   r   r   aop_trtr_ops   &&     r   	test_axesTestFFT1D.test_axesa  sN     < QAr||A)*ELLA2EEt4 r   c                    \         P                  ! ^d4      P                  ^
^
4      pV! VRRR7      P                  R8X  g   Q hR# )d   r   r   Nr   r   r   r   )rr   r   )r   r   r   r   r!   r   r   s   && r   test_s_negative_1TestFFT1D.test_s_negative_1k  s;     IIcN""2r*!wV,22g===r   c                    \         P                  ! ^d4      P                  ^
^
4      p\        P                  ! RR7      ;_uu_ 4        V! VRR7       RRR4       R#   + '       g   i     R# ; i)r  "`axes` should not be `None` if `s`r   )r   Nr  r   r   r   r   warnsr  s   && r   test_s_axes_noneTestFFT1D.test_s_axes_noner  sH     IIcN""2r*\\ DEEqG FEEEs   AA,	c                    \         P                  ! ^d4      P                  ^
^
4      p\        P                  ! RR7      ;_uu_ 4        V! VRRR7       RRR4       R#   + '       g   i     R# ; i)r  r  r   Nr  r  r  r  s   && r   test_s_axes_none_2DTestFFT1D.test_s_axes_none_2Dy  sH    IIcN""2r*\\ DEEqG$' FEEEs   AA-	c                    \        R4      p\        P                  ! RR7      ;_uu_ 4        V! VRRR7       RRR4       R#   + '       g   i     R# ; i)r\   z%array containing `None` values to `s`r   Nr  r   )rr   Nrr   r   )r   r   r  r  s   && r   test_s_contains_noneTestFFT1D.test_s_contains_none  s:     < \\ GHHqN3 IHHHs   AA	c           	        \        ^4      p\        P                  P                  V4      pVP                  ^,          p\        P
                  P
                  \        P
                  P                  3\        P
                  P                  \        P
                  P                  3\        P
                  P                  \        P
                  P                  3.pV Fs  w  rVVP                  ^VP                  ,          3 FK  pR FB  pV! WVR7      pV! WVR7      p\        V\        P                  P                  V4      RR7       KD  	  KM  	  Ku  	  R# )r\   Nr   r]   r1   r   )r   r   linalgr`   r   r    r4   r6   r5   r   r   r   )	r!   r   x_normrC   
func_pairsforwbackr`   tmps	   &        r   test_all_1d_norm_preserving%TestFFT1D.test_all_1d_norm_preserving  s    2J"FFQJvvzz266;;/vv{{BFFLL1 vv||RVV[[1	
 %JDffa!&&j)BDqD1Csd3C#F$&IINN3$7dD C * %r   r   c                  a V3R  lpV\         J dW   \        R4      R\        R4      ,          ,           p\        P                  P                  \        P                  P
                  rvM>\        R4      p\        P                  P                  \        P                  P                  rvV! WSR7      pV! V4      p	V! WYVR7      p
WJ g   Q h\        W4       V! WR7      pV\         J d   T	MV! V4      pV! WVR7      pWJ g   Q h\        W4       R# )c                    < S'       d+   \         P                  ! V P                  4      P                  # \         P                  ! V 4      # rk   rl   rn   s   &r   rJ   4TestFFT1D.test_fftn_out_argument.<locals>.zeros_like  rp   r   r0   r   )rq   r   Nrr   r   r   )	rt   r   r   r    r   r   r   r   r   )r!   r<   rh   r   rJ   r   r    r4   ru   rq   rv   rw   rx   ry   s   &&f&          r   test_fftn_out_argument TestFFT1D.test_fftn_out_argument  s    	( Gz"R&*<%<<ARVV\\z"Abffmmq$"Qd+}}6,-	w&sJy,As407.r   r    c           	        V\         P                  P                  J d   \        R4      pM"\        R4      R\        R4      ,          ,           p\        P
                  ! \        RR7      ;_uu_ 4        V! V\         P                  ! V4      R	R
R7       RRR4       RpV! W#R
R7      p\         P                  ! V4      pV! W#R
VR7      pWeJ g   Q h\        Wd4       R#   + '       g   i     LQ; i)rr   r0   r   r   )rq   r   r   Nr  r   r   rq   r#  r   r   r   r   )rr   r   r   )	r   r    r   r   r   r   r   rJ   r   )r!   r    r   r   ru   rq   rv   s   &&     r   test_fftn_out_and_s_interaction)TestFFT1D.test_fftn_out_and_s_interaction  s     "&&,,z"Az"R&*<%<<A]]:->??r}}Q'99E @ qI.mmH%Q)5}}6, @?s   1!CC+	r   c                    \        R4      R\        R4      ,          ,           p\        P                  P                  W!RR7      p\        P                  ! V4      p\        P                  P                  W!RVR7      pWTJ g   Q h\        WS4       R# )	   r0   r  r'  N)r,  r   r   r   r   )r   r   r    r   rJ   r   )r!   r   r   ru   rq   rv   s   &&    r   !test_irfftn_out_and_s_interaction+TestFFT1D.test_irfftn_out_and_s_interaction  sk     < 2|(<#<<66==i=8mmH%qI3?}}6,r   r$   r  )TFr   )r  r   r   N)r,  r   r   r(  )5r%   r&   r'   r(   r:   r   markparametrizer   singledouble
longdoublerP   rY   rf   rt   r   rz   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r    r   r   r   r   r  r   r   r	  r  r  r  r  r$  r)  r-  r)   r*   r+   s   @r   r.   r.      s    1 [[Wryy"))R]]&KLB MB" [[Wryy"))R]]&KLE ME.B [[VV,[[Ww&67[[[-8/ 9 8 -/6 [[VaV,9/ -9/v= [[V%KL' M'	C	D
C
D@"G:
4D45$4
H
E [[TBFFKK$&FFLL"&&--$A B5B5 [[TBFFKK$&FFKK$? @>@>
 [[TBFFKK$&FFLL"&&--$A BB
 [[TBFFKK#>?( @(
 [[TBFFKK$&FFLL"&&--$&FFKK$? @4@4
D& [[V%;<[[Ww&67[[[-8/ 9 8 =/6 [[URVV[["&&,,$MN- O-  [[S9i"89- :-r   r.   r<   orderFznon-contiguousr    c                    \         P                  P                  ^*4      pVP                  ^^^4      P	                  V RR7      pR\         P
                  ! \         P                  ! VP                  4      4      ,          \         P                  ! VP                  4      P                  ,          pVR8X  d   \         P                  ! V4      pM,VRRR1,          p\         P                  ! VRRR1,          4      pVP                  P                  R4      '       d4   \        ^4       F"  pV! WGR7      pV! WgR7      p	\!        WWUR7       K$  	  R# VP                  P                  R4      '       dc   . ROp
VP                  P                  R	4      '       d   V
P#                  . RO4       V
 F"  pV! WKR
7      pV! WkR
7      p	\!        WWUR7       K$  	  R# \$        h)*   F)rS   g       @r6  Nr    r	   rD   r   r   r   )r   r   )r  )r   r   r/  ))r   )r   )r   N)r   r   RandomStaterandrH   rd   log2r   finfor<   epsasfortranarrayascontiguousarrayr%   endswithr3   r   extendr   )r<   r5  r    rngX_tolYr
   X_resY_resr   axs   &&&         r   test_fft_with_orderrI    s]    ))


#CAr!!%e!4A))BHHQWW,=,A,AAD|a  ddG  4R4)
||U##!HD%E%EEt?  
		/	0	0'<<  ((KK01BOEOEEt? 
 r   CrC   c                    \         P                  P                  ^*4      pVP                  ^
4      p\         P                  ! V\         P
                  V R7      p\         P                  P                  W1R7      pVP                  P                  VP                  P                  8X  g   Q hVP                  P                  VP                  P                  8X  g   Q hR# )r8  )r<   r5  rB   N)
r   r   r9  r:  asarray	complex64r    flagsc_contiguousf_contiguous)r5  rC   rB  r   ress   &&   r   test_fft_output_orderrR    s     ))


#CA


1BLL6A
&&**Q*
C99!!QWW%9%999999!!QWW%9%9999r   zCannot start thread)reasonc                   F   a  ] tR tRt o ^tR	tR tR tR tR t	R t
RtV tR# )
TestFFTThreadSafei  c                  a V3R  lp\         P                  ! 4       pS! V!  p\        V P                  4       Uu. uF  p\        P
                  ! W2V3R7      NK  	  ppV Uu. uF  qP                  4       NK  	   pV Uu. uF  qP                  4       NK  	   p\        V P                  4       F   p\        VP                  ^R7      VR4       K"  	  R# u upi u upi u upi )c                 2   < VP                  S! V !  4       R # rk   )put)argsqfuncs   &&r   worker.TestFFTThreadSafe._test_mtsame.<locals>.worker  s    EE$+r   )targetrY  )timeoutz6Function returned wrong value in multithreaded contextN)
queueQueuer3   threads	threadingThreadstartjoinr   get)	r!   r[  rY  r\  rZ  ru   r9   tr   s	   &f*      r   _test_mtsameTestFFTThreadSafe._test_mtsame  s    	 KKM; DLL)+) V)<) 	
 +AqA1a1t||$AquuQu/HJ %+s   "C"C Cc                    \         P                  ! V P                  4      ^,          R,           pV P                  \         P                  P                  V4       R# r   y                N)r   r   input_shaperi  r    r!   r   s   & r   rf   TestFFTThreadSafe.test_fft  s8    GGD$$%)B."&&**a(r   c                    \         P                  ! V P                  4      ^,          R,           pV P                  \         P                  P
                  V4       R# rl  )r   r   rm  ri  r    r4   rn  s   & r   r   TestFFTThreadSafe.test_ifft!  s8    GGD$$%)B."&&++q)r   c                    \         P                  ! V P                  4      pV P                  \         P                  P
                  V4       R # rk   )r   r   rm  ri  r    r6   rn  s   & r   r   TestFFTThreadSafe.test_rfft%  s.    GGD$$%"&&++q)r   c                    \         P                  ! V P                  4      ^,          R,           pV P                  \         P                  P
                  V4       R# rl  )r   r   rm  ri  r    r5   rn  s   & r   r   TestFFTThreadSafe.test_irfft)  s8    GGD$$%)B."&&,,*r   r$   N)i      )r%   r&   r'   r(   rb  rm  ri  rf   r   r   r   r)   r*   r+   s   @r   rU  rU    s.     GKJ$)**+ +r   rU  c                  ,   \         P                  ! ^
4      p \         P                  P                  V ^R7       \         P                  P	                  V ^R7       \         P                  P                  \         P
                  ! ^ .\        4      ^
R7       R# )rr   rB   N)r   r   r    r5   r   rG   rt   )r   s    r   test_irfft_with_n_1_regressionrx  .  sX    
		"AFFLLaLFFKKQKFFLL1#w'2L.r   c                      \         P                  ! ^4      R,          p \         P                  P                  V ^
R7      p\         P                  ! . RO4      p\        W4       R# )r   rB   Ny      ?      ?)
g      4@gk "#@g'g$ pn]k@grg      @gG& gdƍĿgt=M?g@L>M4)r   r   r    r   rG   r   )r   rv   ru   s      r   "test_irfft_with_n_large_regressionrz  6  sG    
		!AVV[[b[!Fxx 2 3H F%r   datarA   c                     V! V 4      pV P                  \        P                  ! V R 4      4      pV! V4      p\        W$4       R# )r@   N)rH   r   result_typer   )r{  r    rv   
float_dataru   s   &&   r   #test_fft_with_integer_or_bool_inputr  @  s7     YFR^^D"56J:Hv(r   )N      )FTF)(r`  rc  r   numpyr   numpy.randomr   numpy.testingr   r   r   r   r   r   r.   r0  r1  float32float64rM  
complex128r    r   r   r4   r   r   rI  rR  skipifrU  rx  rz  r6   r5   rG   r   uint8int16r  r$   r   r   <module>r     s        U U-< <z- z-z 	RZZr}}=? 3(8"9:	RVV[["&&++	bffllBFFLL	23	3 ;?> 3*-m,: - .: G$9:$+ $+ ;$+N/& FFJJRVV[["&&,,!  HH!"IIb!IIarxx " 
))r   