U
|
O ±d ã @ sl d dl mZ d dlmZ ddlmZmZ ddlmZmZ ddl m
|
Z
|
m Z m Z m Z dd Zd d
|
Zd d Zd S )é )Úgetfullargspec)Ú assert_raisesé )ÚasarrayÚ_elementwise_functions)Úbitwise_left_shiftÚbitwise_right_shift)Ú_dtype_categoriesÚ_boolean_dtypesÚ_floating_dtypesÚ_integer_dtypesc C s t t| jS ©N)Úlenr Úargs)Úfunc© r úgd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\numpy/array_api/tests/test_elementwise_functions.pyÚnargs s r c 9 s ddddddddddddddddddddddddddddddddddddddddddddddddddddddddd8} dd }| D ]| ¡ D ]|\}}t| }tt| t d
|
krî| D ].j|ksÔj|kr¼tt fd d q¼qj|krtt fd d qqd S )NÚnumericzfloating-pointzinteger or booleanÚintegerÚallÚboolean)8ÚabsÚacosÚacoshÚaddÚasinÚasinhÚatanÚatan2ÚatanhZ bitwise_andZbitwise_invertr Z
|
bitwise_orr Z bitwise_xorÚceilÚcosÚcoshÚdivideÚequalÚexpÚexpm1ÚfloorZ floor_divideZgreaterZ greater_equalÚisfiniteÚisinfÚisnanZlessZ
|
less_equalÚlogZ logaddexpÚlog10Úlog1pÚlog2Ú logical_andZ logical_notÚ
|
logical_orÚ logical_xorÚmultiplyÚnegativeÚ not_equalZpositiveÚpowÚ remainderÚroundÚsignÚsinÚsinhÚsqrtZsquareÚsubtractÚtanÚtanhÚtruncc s sL t D ]} td| dV qtD ]} td| dV q tD ]} td| dV q4d S )Né )ÚdtypeFg ð?)r r r
|
r )Údr r r Ú _array_valsS s z(test_function_types.<locals>._array_valsr c s
|
S r r r ©r ÚxÚyr r Ú<lambda>b ó z%test_function_types.<locals>.<lambda>c s S r r r )r rF r r rH e rI )Úitemsr Úgetattrr r rB r Ú TypeError)Z elementwise_function_input_typesrD Ú func_nameÚtypesZdtypesr rE r Útest_function_types s È;
|
|
|
|
rO c C s t tdd t tdd d S )Nc S s t tddgtddgS ©NrA éÿÿÿÿ)r r r r r r rH k rI z*test_bitwise_shift_error.<locals>.<lambda>c S s t tddgtddgS rP )r r r r r r rH n rI )r Ú
|
ValueErrorr r r r Útest_bitwise_shift_errorh s ÿ ÿrS N)Úinspectr Z numpy.testingr Ú r r r r Z_dtypesr r
|
r r r rO rS r r r r Ú<module> s U
|
