
    Si                        S SK Jr  S SKrS SKrS SKrS SKrS SKrS SKJrJ	r	   " S S\R                  5      r\S:X  a  \R                  " SS9  gg)	    )annotationsN)helpernumpy_helperc                     \ rS rSrS3S jrS3S jrS4S jrS4S jrS4S jrS4S jr	S4S jr
S4S	 jrS4S
 jrS4S jrS4S jrS4S jrS4S jrS4S jrS rS rS rS rS rS5S6S jjr\R0                  R3                  S\R6                  R8                  4S\R6                  R:                  4S\R6                  R<                  4S\R6                  R>                  4S\R6                  R@                  4S\R6                  RB                  4S\R6                  RD                  4S\R6                  RF                  4S\R6                  RH                  4S\R6                  RJ                  4S \R6                  RL                  4S!\R6                  RN                  4S"\R6                  RP                  4S#\R6                  RR                  4S$\R6                  RT                  4S%\R6                  RV                  4S&\R6                  RX                  4S'\R6                  RZ                  4S(\R6                  R\                  4S)\R6                  R^                  4S*\R6                  R`                  4S+\R6                  Rb                  4S,\R6                  Rd                  4S-\R6                  Rf                  4S.\R6                  Rh                  4/5      S7S/ j5       r5S0 r6S1r7g2)8TestNumpyHelper   c                2   [         R                  R                  SS5      R                  U5      n[        R
                  " US5      nU R                  UR                  S5        [        R                  " U5      n[         R                  R                  X$5        g )N   %   test)nprandomrandastyper   
from_arrayassertEqualnameto_arraytestingassert_equalselfdtypea
tensor_def	a_recovers        Z/var/www/html/ai-image-ml/venv/lib/python3.13/site-packages/onnx/test/numpy_helper_test.py_test_numpy_helper_float_type-TestNumpyHelper._test_numpy_helper_float_type   si    IINN2r"))%0!,,Q7
&1 ))*5	


-    c                   [         R                  R                  [         R                  " U5      R                  [         R                  " U5      R
                  USS9n[        R                  " US5      nU R                  UR                  S5        [        R                  " U5      n[         R                  R                  X$5        g )Nr
   r   )r   sizer   )r   r   randintiinfominmaxr   r   r   r   r   r   r   r   s        r   _test_numpy_helper_int_type+TestNumpyHelper._test_numpy_helper_int_type   s    IIHHUO%!4!4E  
 ",,Q7
&1 ))*5	


-r    c                B    U R                  [        R                  5        g Nr   r   float32r   s    r   
test_floatTestNumpyHelper.test_float!       **2::6r    c                B    U R                  [        R                  5        g r+   )r(   r   uint8r.   s    r   
test_uint8TestNumpyHelper.test_uint8$       ((2r    c                B    U R                  [        R                  5        g r+   )r(   r   int8r.   s    r   	test_int8TestNumpyHelper.test_int8'   s    ((1r    c                B    U R                  [        R                  5        g r+   )r(   r   uint16r.   s    r   test_uint16TestNumpyHelper.test_uint16*   s    ((3r    c                B    U R                  [        R                  5        g r+   )r(   r   int16r.   s    r   
test_int16TestNumpyHelper.test_int16-   r6   r    c                B    U R                  [        R                  5        g r+   )r(   r   int32r.   s    r   
test_int32TestNumpyHelper.test_int320   r6   r    c                B    U R                  [        R                  5        g r+   )r(   r   int64r.   s    r   
test_int64TestNumpyHelper.test_int643   r6   r    c                *   [         R                  " / SQ5      R                  [        5      n[        R
                  " US5      nU R                  UR                  S5        [        R                  " U5      n[         R                  R                  X5        g )N)AmyBillyCindyDavidr   )r   arrayr   objectr   r   r   r   r   r   r   r   r   r   r   s       r   test_stringTestNumpyHelper.test_string6   sd    HH78??G!,,Q7
&1 ))*5	


-r    c                6   [         R                  R                  SSS9R                  [        5      n[
        R                  " US5      nU R                  UR                  S5        [
        R                  " U5      n[         R                  R                  X5        g )N   r"   )r#   r   )r   r   r$   r   boolr   r   r   r   r   r   r   rR   s       r   	test_boolTestNumpyHelper.test_bool=   sn    IIah/66t<!,,Q7
&1 ))*5	


-r    c                B    U R                  [        R                  5        g r+   r,   r.   s    r   test_float16TestNumpyHelper.test_float16D   r1   r    c                B    U R                  [        R                  5        g r+   )r   r   	complex64r.   s    r   test_complex64TestNumpyHelper.test_complex64G   s    **2<<8r    c                B    U R                  [        R                  5        g r+   )r   r   
complex128r.   s    r   test_complex128TestNumpyHelper.test_complex128J   s    **2==9r    c                
   [         R                  " [        R                  " S5      [        R                  " S5      S.5      nU R	                  U[
        R                  5        U R                  [         R                  " UR                  R                  S   5      [        R                  " S5      5        U R                  [         R                  " UR                  R                  S   5      [        R                  " S5      5        g )N皙??r      r   ri   r   	from_dictr   rP   assertIsInstanceonnxMapProtor   r   valuestensor_valuesr   	map_protos     r   ,test_from_dict_values_are_np_arrays_of_float<TestNumpyHelper.test_from_dict_values_are_np_arrays_of_floatM   s     **rxx}#+OP	i7!!)"2"2"@"@"CDbhhsm	
 	!!)"2"2"@"@"CDbhhsm	
r    c                
   [         R                  " [        R                  " S5      [        R                  " S5      S.5      nU R	                  U[
        R                  5        U R                  [         R                  " UR                  R                  S   5      [        R                  " S5      5        U R                  [         R                  " UR                  R                  S   5      [        R                  " S5      5        g )Nri   	   rh   r   rj   rq   s     r   *test_from_dict_values_are_np_arrays_of_int:TestNumpyHelper.test_from_dict_values_are_np_arrays_of_intW   s     **rxx{rxx{+KL	i7!!)"2"2"@"@"CDbhhqk	
 	!!)"2"2"@"@"CDbhhqk	
r    c                H   [         R                  " SS/5      n[         R                  " SS/5      n[        R                  " XS.5      nU R	                  U[
        R                  5        [        R                  " UR                  R                  S   5      nU R                  US   US   5        U R                  US   US   5        [        R                  " UR                  R                  S   5      nU R                  US   US   5        U R                  US   US   5        g )Nri   rV   rv   
   rh   r   )r   rP   r   rk   rl   rm   rn   r   ro   rp   r   )r   
zero_array	one_arrayrr   
out_tensors        r   +test_from_dict_values_are_np_arrays_of_ints;TestNumpyHelper.test_from_dict_values_are_np_arrays_of_intsa   s    XXq!f%
HHaW%	 **z+HI	i7!**9+;+;+I+I!+LM
A
16A
16!**9+;+;+I+I!+LM
A	!5A	!5r    c                    U R                  [        5         [        R                  " [        R
                  " S5      [        R
                  " S5      S.5        S S S 5        g ! , (       d  f       g = f)Nrf   rg   )r   g?assertRaises	TypeErrorr   rk   r   rP   r.   s    r   "test_from_dict_differing_key_types2TestNumpyHelper.test_from_dict_differing_key_typeso   s?    y)""rxx}288C=#IJ *))   AA!!
A/c                    U R                  [        5         [        R                  " [        R
                  " S5      [        R
                  " S5      S.5        S S S 5        g ! , (       d  f       g = f)Nri   rg   rh   r   r.   s    r   $test_from_dict_differing_value_types4TestNumpyHelper.test_from_dict_differing_value_typest   s?    y)""rxx{rxx}#EF *))r   c                   [         R                  " [         R                  " [         R                  " SS/S/US9/S[         R                  " S[
        R                  R                  S/5      /[         R                  " SUS/5      /5      5      n[
        R                  R                  U5      nS[
        R                  R                  R                  U5      ;   a8  UR                  S S[        R                  " / SQ[        R                  S	905      nO7UR                  S S[        R                  " / S
Q[        R                  S	905      n[         R"                  " US   SS9nU R%                  UR&                  U5        [         R(                  " U5      nU R%                  US   R*                  UR*                  5        U(       a)  U R%                  US   R,                  UR,                  5        [         R"                  " USS9nU R%                  UR&                  UR&                  5        U R%                  UR.                  UR.                  5        U R%                  [1        UR2                  5      [1        UR2                  5      5        U R%                  [5        UR2                  5      [5        UR2                  5      5        U R%                  UR2                  UR2                  5        U R%                  [7        UR8                  5      [7        UR8                  5      5        U R%                  UR;                  5       UR;                  5       5        U R%                  UR&                  [         R<                  " UR,                  5      5        g )NCastXY)tor      UINT)r   ri   rV      )r   )r   ri   r   r   check)r   )r   
make_model
make_graph	make_nodemake_tensor_value_inform   TensorProtoFLOAT	referenceReferenceEvaluatorDataTypeNamerunr   rP   r-   r   r   r   	data_typer   shaper   r   lenraw_datalisttupledimsSerializeToStringnp_dtype_to_tensor_dtype)	r   valuecheck_dtype
onnx_modelrefstarttpbackagains	            r   _to_array_from_array$TestNumpyHelper._to_array_from_arrayy   sn   &&!!&3%#5AB..sD4D4D4J4JQCPQ..sEA3?@	

 nn//
;T%%..33E::GGD3RZZ(P"QREGGD3bjj(Q"RSE$$U1XG<u-$$R(q4U1X^^TZZ8''7;u7%**-R[[)3u~~+>?bkk*D,@Aenn5rwwuzz):;--/1H1H1JKv'F'Ftzz'RSr    r   UINT8INT8UINT16INT16INT32INT64BOOLFLOAT16DOUBLEUINT32UINT64	COMPLEX64
COMPLEX128BFLOAT16FLOAT8E4M3FNFLOAT8E4M3FNUZ
FLOAT8E5M2FLOAT8E5M2FNUZ
FLOAT8E8M0UINT4INT4UINT2INT2
FLOAT4E2M1c                &    U R                  U5        g r+   )r   )r   _r   s      r   test_to_array_from_array(TestNumpyHelper.test_to_array_from_array   s    < 	!!),r    c                X    U R                  [        R                  R                  S5        g )NF)r   rm   r   STRINGr.   s    r   test_to_array_from_array_string/TestNumpyHelper.test_to_array_from_array_string   s    !!$"2"2"9"95Ar     N)r   z	np.numberreturnNone)r   r   )T)r   intr   rW   )r   strr   zonnx.TensorProto.DataType)8__name__
__module____qualname____firstlineno__r   r(   r/   r4   r9   r=   rA   rE   rI   rS   rX   r[   r_   rc   rs   rw   r~   r   r   r   parameterizedexpandrm   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   __static_attributes__r   r    r   r   r      s   ..7324333..79:

6K
G
T<   ''d&&,,-d&&,,-T%%**+t''../d&&,,-d&&,,-d&&,,-T%%**+((001t''../t''../t''../$**4454++667))223T--::;t//>>?4++667t//>>?4++667d&&,,-T%%**+d&&,,-T%%**+4++6673	
:-;:-Br    r   __main__rV   )	verbosity)
__future__r   unittestnumpyr   r   rm   onnx.referencer   r   TestCaser   r   mainr   r    r   <module>r      sN    #      %hBh'' hBV zMMA r    