g~ d dl Z d dlZd dlZd dlZd dlZd dlZd dlmZm Z d dl
m Z d dl m
Z
d dlmZ d dlmZ d Zd Zd Zd
Z G d d e Zed
k r ej dS dS ) N)compile_isolatedFlags)jit)types)TestCase)
numpy_supportc | S N xs Q/granary/edev/quantum/lib/python3.11/site-packages/numba/tests/test_conversion.pyidentityr H c | |z S r
r r
ys r additionr s q5Lr c | |k S r
r r s r equalityr s 6Mr c | S r
r )r
r zs r foobarr r r c l e Zd ZdZd Zd Zd Zd Zd Zd Z d Z
d Z d
Z d Z
d Zd
Zd Zd Zd ZdS )TestConversionz-
Testing Python to Native conversion
c x t }t |t j gt j }g d}|D ]+}| | | | ,t
j | D ]+}| | | | ,t |t j gt j }g d}|D ]+}| | | | ,t
j | D ]+}| | | | ,d S N return_type)y ?y ? ?y y ? )r r r complex64 assertEqual entry_pointnp
complex128)selfpyfunccresxsr
s r test_complex_identityz$TestConversion.test_complex_identity" s] (9,1O= = = -
,
, 5 5A T - -a 0 0! 4 4 4 4 b!! 5 5A T - -a 0 0! 4 4 4 4 )9(:,1,<> > > -
,
, 5 5A T - -a 0 0! 4 4 4 4 r"" 5 5A T - -a 0 0! 4 4 4 4 5 5r c t }t |t j t j gt j }g d}|D ]1}|}| | || ||z 2t
j | D ]1}|}| | || ||z 2t |t j t j gt j }g d}|D ]1}|}| | || ||z 2t
j | D ]1}|}| | || ||z 2d S r )r r r r! r" r# r$ r% )r& r' r( r) r
r s r test_complex_additionz$TestConversion.test_complex_addition6 s %/(J,1O= = = -
,
, < > > -
,
, < .fv s Hr z uintp(uintp)Tnopython) assertRaises
OverflowErrorr r& rB s r test_negative_to_unsignedz(TestConversion.test_negative_to_unsignedu s
}
-
- 6 6 1 .C . . .q 1 1" 5 5 5 6 6 6 6 6 6 6 6 6 6 6 6 6 6 6 6 6 6s $AAAc 4 t }t |t j t j t j gt j }|j }d}| t 5 |D ]\ }}} |||| d d d d S # 1 swxY w Y d S )Nr )r0 )r r1 r2 )r r2 r1 )r r r uint64r# rF rG )r& r' r( r8 test_fail_argsabcs r 'test_multiple_args_negative_to_unsignedz6TestConversion.test_multiple_args_negative_to_unsigned} s u|U\(R,1L: : : =
}
-
- )
1aaA
s (B
BBc R t }t j dt j fdt j fdt j fg }t j | }t
||d d t j t j g|d d }|j }t j
d| }t j |j dz |_
t j |j dz |_ t j |j dz |_ | | 5 |ddf|ddff}|D ]E\ }} }
| t$ 5 ||| |
d d d n # 1 swxY w Y F~~~ ~
t' j d d d d S # 1 swxY w Y d S ) Nprowcolr dtyper2 r1 )r r$ rW r< r
from_dtyper r rK r# recarrayarangesizerR rS rT assertRefCountrF rG gccollect) r& r' mystruct_dtmystructr( r8 st1rL rM rN rO s r test_multiple_args_recordsz)TestConversion.test_multiple_args_records s hbj 1 !2:. !2:. 0 1 1 !+K88 !!!elEL(Q,4QQQK9 9 9 k!;/// #(##a')CH%%))CH%%)
%
% "BlS!RL 9N)
#
#1a&&}55 # #E!QNNN# # # # # # # # # # # # # # # 1a JLLL s6 %-F E, F ,E0 0F 3E0 4F F #F c d }| | t dd | d S )Nc d S r
r r r r rB z1TestConversion.test_with_no_parameters..f Dr z()TrC )r" r rH s r test_with_no_parametersz&TestConversion.test_with_no_parameters sT 9 6c$ 6 6 6q 9 9 ; ;<<<<.f re r c ^ | g}t | t r| D ]}| | z
}|S r
)
isinstancetuple)objobjsv_objectss r ro z7TestConversion.check_argument_cleanup.._objects sD 5D#u%%
( ( (AHHQKK'DDKr r2 Nr1 )r r uint32r\ r# rF rG )r& typrl rB objectsr( ro s @r check_argument_cleanupz%TestConversion.check_argument_cleanup s5 (3-- C #677
T
'
* % % S! $ $ $ % % % % % % % % % % % % % % %
T
'
* * *""=11
*
* b)))
*
*
*
*
*
*
*
*
*
*
*
*
*
*
* * * * * * * * * * * * * * * * EL##677
T
'
* % % Q $ $ $ % % % % % % % % % % % % % % %
T
'
* * *""=11
*
* S)))
*
*
*
*
*
*
*
*
*
*
*
*
*
*
* * * * * * * * * * * * * * * * * * *s A A"%A"3CB1% C1B5 5C8B5 9CC C 9D D #D 1F E/# F/E3 3F6E3 7FF F c t t d }| t j t j dd | d S N xyzr2 C)
memoryview bytearrayrs r Bufferintc)r& mems r test_cleanup_bufferz"TestConversion.test_cleanup_buffer sC 6**++ ##ELQ$D$DcJJJJJr c t j dt j fdt j fg }t j d| }| t j | |d d S )Nr
r r2 rV r )r$ rW r< zerosrs r rX )r& rW recarrs r test_cleanup_recordz"TestConversion.test_cleanup_record sc 3
+c2:->?@@!5))) ##M$.bar s Dr c d d S )Nzliteral stringr )r s r foo2z:TestConversion.test_stringliteral_to_unicode..foo2 s C ! ! ! ! !r )r r void unicode_type)r& r r s @r test_stringliteral_to_unicodez,TestConversion.test_stringliteral_to_unicode s|
UZ*
+
+d ; ; ;
< ;
UZ\\D ) ) ) " " " "
* ) " " "r N)__name__
__module__ __qualname____doc__r* r, r: r= r? rI rP rb rf rs r} r r r r r r r r r s 5 5 5(< < <2 < < < < < < < < < 6 6 6 := = =
* * *<K K KP P P
4 4 4
- - -
"
"
"
"
"r r __main__)arrayr] r5 sysnumpyr$ unittestnumba.core.compilerr r numbar
numba.corer numba.tests.supportr numba.npr r r r r r r mainr r r r s6
7 7 7 7 7 7 7 7 ( ( ( ( ( ( " " " " " "
H" H" H" H" H"X H" H" H"V zHMOOOOO r