U
|
O ±do ã @ sÖ d Z ddlZddlZddlZddlZddlm mZ ddl
|
m Z m Z G dd dZ G dd de ZG dd d e ZG d
|
d d e ZG d d d e ZG dd de ZG dd de ZG dd de ZG dd dZdS )z8
|
Tests for numpy/core/src/multiarray/conversion_utils.c
|
é N)Ú assert_warnsÚIS_PYPYc @ sD e Zd ZdZdZdZdZdd Zdd Zdd Z d d
|
Z
|
d d Z d S )ÚStringConverterTestCaseTFc C s< d t t|¡¡}tjt|d}| |¡ W 5 Q R X d S )Nz
|
\(got {}\))Úmatch)ÚformatÚreÚescapeÚreprÚpytestÚraisesÚ
|
ValueErrorÚconv)ÚselfÚvalÚpatternÚexc© r ú]d:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\numpy/core/tests/test_conversion_utils.pyÚ_check_value_error s z*StringConverterTestCase._check_value_errorc C sD | j r.tt}| |¡|ks"tW 5 Q R X n| |¡|ks@td S ©N)Úwarnr ÚDeprecationWarningr ÚAssertionError)r r Úexpectedr r r r Ú_check_conv_assert_warn s
|
z/StringConverterTestCase._check_conv_assert_warnc C s | |¡|kst| jr2| | d¡¡|ksXtn&t t¡ | | d¡¡ W 5 Q R X t|dkr | jr| |dd
¡ | |d ¡ n|
|
|dd
|¡ | j rà|| ¡ krÂ|
|
| ¡ |¡ || ¡ krÞ|
|
| ¡ |¡ n6|| ¡ krú| | ¡ ¡ || ¡ kr| | ¡ ¡ dS )zTakes valid non-deprecated inputs for converters,
|
runs converters on inputs, checks correctness of outputs,
|
warnings and errorsÚasciié Nú )r r Ú allow_bytesÚencoder
|
r Ú TypeErrorÚlenÚ exact_matchr r Úcase_insensitiveÚlowerÚupper)r r r r r r Ú_check s& z StringConverterTestCase._checkc C sD t t¡ | i ¡ W 5 Q R X t t¡ | g ¡ W 5 Q R X d S r ©r
|
r r r ©r r r r Útest_wrong_type> s z'StringConverterTestCase.test_wrong_typec C sB | d¡ | d¡ | jr.| d¡ | d¡ | jr>| d¡ d S )NÚ u Ïó ó ÿz there's no way this is supported)r r r" r( r r r Útest_wrong_valueE s
|
|
|
|
z(StringConverterTestCase.test_wrong_valueN) Ú__name__Ú
|
__module__Ú __qualname__r r# r" r r r r& r) r- r r r r r s r c @ s" e Zd ZdZejZdZdd ZdS )ÚTestByteorderConverterz% Tests of PyArray_ByteorderConverter Fc C sr dD ]}| |d¡ qdD ]}| |d¡ qdD ]}| |d¡ q0dD ]}| |d¡ qFd D ]}| |d
|
¡ q\d S ) N)Úbigú>ZNPY_BIG)Úlittleú<Z
|
NPY_LITTLE)Znativeú=Z
|
NPY_NATIVE)Úignoreú|Z
|
NPY_IGNORE)ZswapZNPY_SWAP©r& )r Úsr r r Ú
|
test_validW s z!TestByteorderConverter.test_validN) r. r/ r0 Ú__doc__ÚmtZrun_byteorder_converterr r r; r r r r r1 R s r1 c @ s" e Zd ZdZejZdZdd ZdS )ÚTestSortkindConverterz$ Tests of PyArray_SortkindConverter Fc C s4 | dd¡ | dd¡ | dd¡ | dd¡ d S )NZ quicksortZ NPY_QUICKSORTZheapsortZ NPY_HEAPSORTZ mergesortZNPY_STABLESORTZstabler9 r( r r r r; i s z TestSortkindConverter.test_validN) r. r/ r0 r< r= Zrun_sortkind_converterr r r; r r r r r> d s r> c @ s& e Zd ZdZejZdZdZdd Z dS )ÚTestSelectkindConverterz& Tests of PyArray_SelectkindConverter FTc C s | dd¡ d S )NZ introselectZNPY_INTROSELECTr9 r( r r r r; v s z"TestSelectkindConverter.test_validN)
|
r. r/ r0 r< r= Zrun_selectkind_converterr r# r" r; r r r r r? p s
|
r? c @ s e Zd ZdZejZdd ZdS )ÚTestSearchsideConverterz& Tests of PyArray_SearchsideConverter c C s | dd¡ | dd¡ d S )NÚleftZNPY_SEARCHLEFTÚrightZNPY_SEARCHRIGHTr9 r( r r r r; } s z"TestSearchsideConverter.test_validN)r. r/ r0 r< r= Zrun_searchside_converterr r; r r r r r@ z s r@ c @ s* e Zd ZdZejZdZdd Zdd Z dS )ÚTestOrderConverterz! Tests of PyArray_OrderConverter Fc C s4 | dd¡ | dd¡ | dd¡ | dd¡ d S ) NÚcZ
|
NPY_CORDERÚfZNPY_FORTRANORDERÚaZ NPY_ANYORDERÚkZ NPY_KEEPORDERr9 r( r r r r; s z TestOrderConverter.test_validc
|
C sN t t¡ | d¡ W 5 Q R X dD ]$}t t¡ | |¡ W 5 Q R X q$d S )NÚZ)FTr é )r
|
r r r r )r Úorderr r r Útest_flatten_invalid_order s
|
z-TestOrderConverter.test_flatten_invalid_orderN)
|
r. r/ r0 r< r= Zrun_order_converterr r r; rK r r r r rC s
|
rC c @ s e Zd ZdZejZdd ZdS )ÚTestClipmodeConverterz$ Tests of PyArray_ClipmodeConverter c C sd | dd¡ | dd¡ | dd¡ | tj¡dks8t| tj¡dksLt| tj¡dks`td S )NZclipZNPY_CLIPÚwrapZNPY_WRAPÚraiseZ NPY_RAISE)r& r ÚnpZCLIPr ZWRAPZRAISEr( r r r r; s z TestClipmodeConverter.test_validN)r. r/ r0 r< r= Zrun_clipmode_converterr r; r r r r rL s rL c @ s& e Zd ZdZejZdZdZdd Z dS )ÚTestCastingConverterz# Tests of PyArray_CastingConverter FTc C s@ | dd¡ | dd¡ | dd¡ | dd¡ | d d
|
¡ d S ) NÚnoZNPY_NO_CASTINGÚequivZNPY_EQUIV_CASTINGÚsafeZNPY_SAFE_CASTINGZ same_kindZNPY_SAME_KIND_CASTINGÚunsafeZNPY_UNSAFE_CASTINGr9 r( r r r r; ª s
|
z TestCastingConverter.test_validN)
|
r. r/ r0 r< r= Zrun_casting_converterr r# r" r; r r r r rP ¤ s
|
rP c @ sZ e Zd ZdZejZdd Zdd Ze j
|
j e o6e jjdkddd d
|
Zd d Zd d ZdS )ÚTestIntpConverterz Tests of PyArray_IntpConverter c C sP | d¡dkst| d¡dks$t| ddg¡dks:t| d¡dksLtd S )Nr )r )r é rV r )r r r( r r r Ú
|
test_basic¶ s z TestIntpConverter.test_basicc C s, t t¡ | d ¡dks tW 5 Q R X d S )Nr )r
|
Zwarnsr r r r( r r r Ú test_none¼ s zTestIntpConverter.test_none)é é rI z PyPy bug in error formatting)Úreasonc C sH t t¡ | d¡ W 5 Q R X t t¡ | ddg¡ W 5 Q R X d S )Ng ð?r r' r( r r r Ú
|
test_floatÁ s z TestIntpConverter.test_floatc C s$ t t¡ | d¡ W 5 Q R X d S )Nl )r
|
r r r r( r r r Útest_too_largeÉ s z TestIntpConverter.test_too_largec C sB | dgd ¡dkstt t¡ | dgd ¡ W 5 Q R X d S )Nr é ) 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 r r( r r r Útest_too_many_dimsÍ s z$TestIntpConverter.test_too_many_dimsN)r. r/ r0 r< r= Zrun_intp_converterr rW rX r
|
ÚmarkZskipifr ÚsysÚimplementationÚversionr\ r] r` r r r r rU ² s ÿ
|
rU )r< r rb r
|
ÚnumpyrO Z numpy.core._multiarray_testsÚcoreZ_multiarray_testsr= Z numpy.testingr r r r1 r> r? r@ rC rL rP rU r r r r Ú<module> s D
|
|
