
    Mh                         S SK rS SKrS SKJrJr  S SKJr   " S S5      r	 " S S\	5      r
 " S S\	5      r\R                  R                  S	/ S
Q5      S 5       rg)    N)	DataFrameSeriesc                       \ rS rSr\R
                  S 5       r\R
                  S\4S j5       r\R
                  S\4S j5       r	\
S 5       rS rS rS	 rS
 rS rS rS rS rSrg)DotSharedTests   c                     [         eNNotImplementedErrorselfs    U/var/www/html/env/lib/python3.13/site-packages/pandas/tests/frame/methods/test_dot.pyobjDotSharedTests.obj   s    !!    returnc                     [         e)zF
other is a DataFrame that is indexed so that obj.dot(other) is valid
r
   r   s    r   otherDotSharedTests.other   
    
 "!r   c                     [         e)z'
The expected result of obj.dot(other)
r
   r   r   r   s      r   expectedDotSharedTests.expected   r   r   c                     [         e)>
Assertion about results with 1 fewer dimension that self.obj
r
   clsresultr   s      r   reduced_dim_assert!DotSharedTests.reduced_dim_assert   r   r   c                 R    UR                  U5      n[        R                  " XC5        g r	   )dottmassert_equalr   r   r   r   r   s        r   test_dot_equiv_values_dot(DotSharedTests.test_dot_equiv_values_dot%   s    
)r   c                     UR                  UR                  5      n[        R                  " XCR                  :H  5      (       d   eg r	   )r#   valuesnpallr&   s        r   test_dot_2d_ndarray"DotSharedTests.test_dot_2d_ndarray*   s2     &vvf/0000r   c                     UR                   S:X  a  UR                  S   OUnUR                  UR                  5      nUR                  U5      nU R	                  XB5        g )N   r   )ndimilocr#   r*   r    )r   r   r   rowr   s        r   test_dot_1d_ndarray"DotSharedTests.test_dot_1d_ndarray0   sH     XX]chhqk$773<1r   c                 T    UR                  US   5      nU R                  XCS   5        g )N1)r#   r    r&   s        r   test_dot_seriesDotSharedTests.test_dot_series8   s&    s$6r   c                 t    UR                  UR                  S S S2   S   5      nU R                  XCS   5        g )Nr7   )r#   r2   r    r&   s        r   test_dot_series_alignment(DotSharedTests.test_dot_series_alignment=   s4    DbD)#./6r   c                 v    UR                   S S S2   nUR                  U5      n[        R                  " XS5        g )Nr;   )r2   r#   r$   r%   )r   r   r   r   other2r   s         r   test_dot_alignsDotSharedTests.test_dot_alignsA   s-    DbD!
)r   c                     Sn[         R                  " [        US9   UR                  UR                  S S 5        S S S 5        g ! , (       d  f       g = f)NzDot product shape mismatchmatch   )pytestraises	Exceptionr#   r*   )r   r   msgs      r   test_dot_shape_mismatch&DotSharedTests.test_dot_shape_mismatchG   s7    *]]9C0GGCJJrN# 100s   A
Ac                     Sn[         R                  " [        US9   UR                  UR                  5        S S S 5        g ! , (       d  f       g = f)Nzmatrices are not alignedrC   )rF   rG   
ValueErrorr#   T)r   r   r   rI   s       r   test_dot_misaligned"DotSharedTests.test_dot_misalignedM   s1    (]]:S1GGEGG 211s   A
A N)__name__
__module____qualname____firstlineno__rF   fixturer   r   r   r   classmethodr    r'   r-   r4   r8   r<   r@   rJ   rO   __static_attributes__rQ   r   r   r   r      s    ^^" " ^^"y " " ^^"i " " " "*
127
7*$r   r   c                       \ rS rSr\R
                  S 5       r\R
                  S 5       r\R
                  S 5       r\	S 5       r
Srg)TestSeriesDotS   c                 r    [        [        R                  R                  S5      R	                  S5      / SQS9$ )Nr0      pqrsindex)r   r+   randomdefault_rngstandard_normalr   s    r   r   TestSeriesDot.objT   s/    II!!!$44Q7?S
 	
r   c                     [        [        R                  R                  S5      R	                  S5      / SQ/ SQS9R
                  $ )Nr0   rE   r]   )r7   23r^   rd   columns)r   r+   re   rf   rg   rN   r   s    r   r   TestSeriesDot.otherZ   s9    II!!!$44V<!(
 !		r   c                 |    [        [        R                  " UR                  UR                  5      UR                  S9$ )Nrc   )r   r+   r#   r*   rn   r   s      r   r   TestSeriesDot.expectedb   s&    bffSZZ6emmLLr   c                 0    [         R                  " X5        g)r   N)r$   assert_almost_equalr   s      r   r     TestSeriesDot.reduced_dim_assertf   s    
 	v0r   rQ   NrR   rS   rT   rU   rF   rV   r   r   r   rW   r    rX   rQ   r   r   rZ   rZ   S   s^    ^^
 

 ^^  ^^M M 1 1r   rZ   c                       \ rS rSr\R
                  S 5       r\R
                  S 5       r\R
                  S 5       r\	S 5       r
Srg)TestDataFrameDotn   c                 x    [        [        R                  R                  S5      R	                  S5      / SQ/ SQS9$ )Nr0   rj   )abcr^   rm   r   r+   re   rf   rg   r   s    r   r   TestDataFrameDot.objo   s2    II!!!$44V<!(
 	
r   c                 x    [        [        R                  R                  S5      R	                  S5      / SQSS/S9$ )Nr0   )r]   r0   r^   r7   rk   rm   r}   r   s    r   r   TestDataFrameDot.otherw   s6    II!!!$44V<&#J
 	
r   c                     [        [        R                  " UR                  UR                  5      UR                  UR
                  S9$ )Nrm   )r   r+   r#   r*   rd   rn   r   s      r   r   TestDataFrameDot.expected   s1    FF3::u||,CIIu}}
 	
r   c                 L    [         R                  " XSS9  UR                  b   eg)r   F)check_namesN)r$   assert_series_equalnamer   s      r   r    #TestDataFrameDot.reduced_dim_assert   s$    
 	vUC{{"""r   rQ   Nru   rQ   r   r   rw   rw   n   s\    ^^
 
 ^^
 
 ^^
 

 # #r   rw   zdtype,exp_dtype))Float32Float64)Int16Int32)zfloat[pyarrow]zdouble[pyarrow]c                     [         R                  " S5        SS/n[        SS/SS/SS	//US
S9n[        SS/SS//X S9nUR                  U5      n[        SS/SS/SS	//US9n[        R
                  " XV5        g )Npyarrowrz   r{      r0   rE   r]         int32)rn   dtyper   )rd   r   )r   )rF   importorskipr   r#   r$   assert_frame_equal)r   	exp_dtypecolsdf_adf_br   r   s          r   test_arrow_dtyper      s    
 	":Dq!fq!fq!f-t7KDq!fq!f%T?DXXd^F1a&1a&1a&1CH&+r   )numpyr+   rF   pandasr   r   pandas._testing_testingr$   r   rZ   rw   markparametrizer   rQ   r   r   <module>r      sd      E EP1N 16#~ #@ W	,		,r   