U
|
O ±dm ã @ sò d Z ddlZddlmZ ddlmZ ddlZddlm Z m
|
Z
|
m Z ej dfej dfejdfejd fejd
|
fejd fejd fejd fejdfejdfejdfejdfejdfejdfejdfejdfej dfej dfgZ e e \Z Z!G dd dZ"dS )z2
|
Test scalar buffer interface adheres to PEP 3118
|
é N)Úrational)Úget_buffer_info)Úassert_Ú assert_equalÚ assert_raisesú?ÚbÚhÚiÚlÚqÚBÚHÚIÚLÚQÚeÚfÚdÚgZZfZZdZZgc @ sÎ e Zd Zejjdeeddd Zejjdeeddd Z ejjde
|
eddd Z ejjdeedd
|
d Z d d Z dd Zdd Zej dejdddejdddejdddg¡dd Zd d Zd S ) ÚTestScalarPEP3118Úscalar)Úidsc C s< | }t jg t |¡d}t|}t|}t|j|j d S )N©Údtype)ÚnpÚarrayr Ú
|
memoryviewr Úformat)Úselfr ÚxÚaÚmv_xÚmv_a© r$ úYd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\numpy/core/tests/test_scalarbuffer.pyÚtest_scalar_match_array% s
|
z)TestScalarPEP3118.test_scalar_match_arrayc C sV | }t |}t|jt |¡j t|jd t|jd t|jd t|jd d S )Nr r$ ) r r Úitemsizer r ÚndimÚshapeÚstridesÚ
|
suboffsets)r r r r" r$ r$ r% Útest_scalar_dim- s z!TestScalarPEP3118.test_scalar_dimz scalar, codec C s< | }t d|jjdd|dd}t|}| |¡|ks8td S )Nr$ r T©r* r' r( r) r Úreadonly)Údictr r' r Ú_as_dictÚAssertionError)r r Úcoder Úexpectedr" r$ r$ r% Ú test_scalar_code_and_properties7 s ÿz1TestScalarPEP3118.test_scalar_code_and_propertiesc C s0 | }t jtdd t|dg W 5 Q R X d S )Núscalar buffer is readonly©ÚmatchÚWRITABLE)ÚpytestÚraisesÚ BufferErrorr )r r r r$ r$ r% Ú test_scalar_buffers_readonly@ s z.TestScalarPEP3118.test_scalar_buffers_readonlyc C s t dt jdfdt jdfg¡}t jd|dd }tt|t j t|}dt t jdf¡j }|d t t j¡j 7 }t
|
|j | t
|
|j d
|
t
|
|j d t
|
|j d t
|
|jd t jd d g|d}tt|t j t|}t
|
|j |j t
|
|j|j tjtd d t|dg W 5 Q R X d S )NÚnameé Zgrades)é )Zndarray_scalar)g333333ó?g @r r$ é r? r )ZSarah)g @ç @)ZJohn)g @rA r5 r6 r8 )r r Zunicode_Úfloat64r r Ú
|
isinstanceZvoidr r' r r( r) r* r+ Zndarrayr r9 r: r; r )r Údtr r" Z expected_sizer! r# r$ r$ r% Ú test_void_scalar_structured_dataF s$ z2TestScalarPEP3118.test_void_scalar_structured_datac C s t |j|j|j|j|j|jdS )N)r* r) r' r( r r. )r/ r* r) r' r( r r. )r Úmr$ r$ r% r0 _ s
|
ÿzTestScalarPEP3118._as_dictc C s¸ t d¡}t d¡}tddddddd}t|}| |¡|ksBtt|| }| |¡|ks`tt d d
|
g¡}t d|¡}tt t
|
ft|d t j t
|
d d t |dg W 5 Q R X d S )Nz
|
2016-01-01z
|
2017-01-01)r@ r@ )é r Tr- )r! Zuint16)r zM8[s]r r5 r6 r8 )r Z
|
datetime64r/ r r0 r1 r Úemptyr Ú
|
ValueErrorr; r9 r: r )r Zdt1Zdt2r3 ÚvrD r! r$ r$ r% Útest_datetime_memoryviewc s
|
|
|
ÿ z*TestScalarPEP3118.test_datetime_memoryviewÚsZ22Úascii)Úidu ï¸ï¸zbasic multilingualu ð»ð»znon-BMPc C s t |¡}tddddddd}t|}| |¡|ks8tt j|dd}t|d d
|
|D tj t
|
d d t |d g W 5 Q R X d S )Nr$ rG r Z2wTr- Úi4r c S s g | ] }t |qS r$ )Úord)Ú.0Úcr$ r$ r% Ú
|
<listcomp> s z3TestScalarPEP3118.test_str_ucs4.<locals>.<listcomp>r5 r6 r8 ) r Zstr_r/ r r0 r1 Z
|
frombufferr r9 r: r; r )r rL r3 rJ Z code_pointsr$ r$ r% Ú test_str_ucs4z s
|
ÿz TestScalarPEP3118.test_str_ucs4c C sN t d}tt t| W 5 Q R X tjtdd t|dg W 5 Q R X d S )Nr@ r5 r6 r8 )r r Ú TypeErrorr r9 r: r; r )r Úrr$ r$ r% Ú test_user_scalar_fails_buffer s
|
|
z/TestScalarPEP3118.test_user_scalar_fails_bufferN)Ú__name__Ú
|
__module__Ú __qualname__r9 ÚmarkZ parametrizeÚ scalars_onlyÚ
|
codes_onlyr& r, Úscalars_and_codesr4 r< rE r0 rK ÚparamrT rW r$ r$ r$ r% r # s$
|
|
|
|
ý
|
r )#Ú__doc__Únumpyr Znumpy.core._rational_testsr Z numpy.core._multiarray_testsr r9 Z numpy.testingr r r Zbool_ÚbyteÚshortZintcÚint_ZlonglongZubyteZushortZuintcZuintZ ulonglongZhalfZsingleÚdoubleZ
|
longdoubleZcsingleZcdoubleZ clongdoubler^ Úzipr\ r] r r$ r$ r$ r% Ú<module> s4 î
|
