
    g                     D    d dl Zd dlmZmZmZ d dlmZ d Zd Z	d Z
d Zy)    N)CenterStandardizecenter)atleast_2d_column_defaultc                 *   t        j                  |      }t        j                  |      }|g|f||f|D cg c]  }|g c}|f|D cg c]  }t        j                  |       c}|ft        j                  |      g|f|D cg c]  }t        j                  |g       c}|ft        j                  |      d d d f   gt        |      f|D cg c]  }t        j                  |gg       c}t        |      fg}|r|t        j                  ||d d d   f      gt        j                  ||d d d   f      ft        t        |            D cg c]$  }t        j                  ||   || dz
     gg      & c}t        j                  ||d d d   f      fgz  }ddlm}	 |	ridd l	}
|
j                  |
j                  f}t        j                  ddt        |            }|j                  dk(  r|
j                  ||      }n|
j                  ||      }||
j                  ||      g|ft        ||      D cg c]  \  }}|
j                  |g|g       c}}|fgz  }|rt        j                  ||d d d   f      }t        j                  ||d d d   f      }|
j                  ||      }||
j                  ||      g|ft        t        |            D cg c]!  }|
j                  ||d d f   g||   g      # c}|fgz  }|D ]  \  }}t        |        |        }|D ]  } |j                   |g|i |  |j#                          g }|D ]p  } |j$                  |g|i |}|j                  |j                  k(  r.|j                  t        j                  |      j                  k(  sJ |j'                  |       r |	rBt)        |d         r3
j+                  |      }t        j,                  |j.                        sgJ |d   j                  dk(  rt        j                  |      }n=|d   j                  dk(  rt        j0                  |      }nt        j2                  |      }|j4                  d   t        |      k(  sJ t        j6                  ||      sJ t        j                  |d         j                  dk(  rt        j                  |      }ns|	r!t)        |d         r
j+                  |      }nPt        j                  |d         j                  dk(  rt        j0                  |      }nt        j2                  |      } |j$                  |g|i |}|	r1t)        |d         r"t        j,                  |j.                        sJ |j                  |j                  k(  r|j                  |j                  k(  sJ t        j6                  ||      rJ  y c c}w c c}w c c}w c c}w c c}w c c}}w c c}w )N   r   have_pandas)numindex)npasarrayarrayr   column_stackrangelen
patsy.utilr   pandasSeries	DataFramelinspacendimzipprintmemorize_chunkmemorize_finish	transformappend
isinstanceconcatarray_equalr   concatenatevstackshapeallclose)clsaccepts_multicolumninputoutputargskwargsn
test_casesir   r   pandas_typepandas_index	output_1dxidxinput_2d_2coloutput_2d_2coloutput_2col_dataframe	input_obj
output_objtinput_chunkall_outputsoutput_chunkall_output1	all_inputall_output2s                               M/var/www/dash_apps/app1/venv/lib/python3.12/site-packages/patsy/test_state.pycheck_statefulrC      s   JJuEZZF &		!1#	v&$	%!"((1+	%v.
((5/	F#!&	'A"((A3-	'0
((5/!T'
"	#%>v%FG#(	)a"((QC5/	)+DV+LM!J$ %tt!567" 67 AFc%j@QR1E!HeQBFm456R" 67

 
	

 '}}f&6&67{{1aSZ8;;!fLAI((|(DImmEm67C
 %(|$< C MM1#cUM3 	
 	

 OOUE$B$K,@AM__ffTrTl-CDN$*$4$4^<$4$X! %%m<%HI) "'s5z!2 ((-1*=)>|TUFW(X * J ", )4	:iE$ 	;KA[:4:6:	;	$ 	-K&1;;{DTDVDLzzV[[(#((BJJ{,C,H,HHHH|,		-
 :k!nkB --4K>>+"3"3\BBB^  A%((;/K^  A%..5K))K0K  #s5z111 {{;
333::il#((A-+IZ	!kBi0IZZ	!%**a/ y1I		),I!akk)=d=f=:ilK@>>+"3"3\BBB::$##y~~555{{;
333S)4I 
	% 
( 
* S*&s)   
W1W6
W;X )X<X

&Xc                      t        t        dg dg d       t        t        dg dg d       t        t        dg dg d       y )NTr	         r   r   r	   r	   rF   r	   rF               ?rK   rL   )g?g333333$g      @      (@)g      gL)g@gfffff"@)rC   r        rB   test_CenterrP      s/    64J764/EF64!8:UVrO   c                     t        j                  t        g d      g d      sJ t        j                  t        g d      g d      sJ t        g d      j                  t        j                  t              k(  sJ t        t        j
                  g dt         j                              j                  t        j                  t         j                        k(  sJ t        g d      j                  t        j                  t              k(  sJ ddlm}  | r*dd l	}|j                  g dg d	
      }|j                  ddgddgddggddgg d      }t        |      }t        ||j                        sJ t        j                  |j                  g d	      sJ t        j                  |g d      sJ t        |      }t        ||j                        sJ t        j                  |j                  g d      sJ t        j                  |j                  ddg      sJ t        j                  |ddgddgddgg      sJ y y )NrE   rH   rI   rJ   )g      ?g       @g      @)dtyper   r
   )abcr   r	   rF      rG      x1x2)
         )columnsr   r   )r   r'   r   rR   floatr   float32r   r   r   r   r   r!   r#   r   r]   )r   r   sdfs_cdf_cs         rB   test_stateful_transform_wrapperre      s   ;;vi(*555;;vl+-CDDD/"((BHHUO;;;"((?"**=>DD


I    )""bhhuo555&MM)?M;VaVaV$tTl,  
 Qi#v}}---~~cii999{{3
+++bz$ 0 0111~~djj,777~~dllT4L999{{42r(QFQF!;<<< rO   c            
         t        t        dddgddg       t        t        dddgddg       t        t        dg dt        j                  d      dt        j                  d       g       t        t        dg d	t        j                  d      dt        j                  d       g       t	        t        d
            } t        t        dddgt        j                  d      dz  t        j                  d       dz  gd       t        t        d| t	        t        j                  d
      dz
  dz        d       t        t        d| t	        t        j                  d
      dz
  dz        d       t        t        d| t	        t        j                  d
      dz
        dd       t        t        d| t	        t        j                  d
      dz        dd       t        t        d| | ddd       y )NTr	   r      rZ   )rg      rZ   g      ?r   )rM   g      &@g      $@r[   rF   )ddofg      #@g2^^@g-@F)rescaleri   )r   ri   )r   rj   ri   )rC   r   r   sqrtlistr   arange)r20s    rB   test_Standardizero      s   ;q"g2w7;r2hB8T<"'''*:A@P?P)Q T-0@!bgggFVEV/W uRy/CTAr7RWWQZ!^bggaj[1_$EA T3biimc&9=O%O PWX T3biimc&9=O%O PWX T3biimc&9 ;UQR RYYr]//0 ;c3ueRSTrO   )numpyr   patsy.stater   r   r   r   r   rC   rP   re   ro   rN   rO   rB   <module>rr      s+   
  3 3 0v4rW=:(UrO   