o Z۷iK@s<dZddlZddlZddlZddlZddlmZddlmZddl Z ddl m Z m Z mZmZddlmZmZddlZddlmZmZmZddlmZdd lmZdd lmZd d ZGd ddeZGdddeZGdddeZ Gddde Z!Gddde Z"Gddde Z#GdddeZ$e%dkre&dSdS)z* Test hashing of various supported types. N) defaultdict)dedent)jitconfigtypedtypeof)typesutils)TestCaseskip_unless_py10_or_laterrun_in_subprocess)compile_time_get_string_data)hashing) numpy_versioncCst|SNhashxrN/home/ubuntu/vllm_env/lib/python3.10/site-packages/numba/tests/test_hashing.py hash_usecaserc@seZdZddZdS)TestHashingSetupcCsd}ttjdt|gdS)Na import sys import warnings from collections import namedtuple # hash_info is a StructSequence, mock as a named tuple fields = ["width", "modulus", "inf", "nan", "imag", "algorithm", "hash_bits", "seed_bits", "cutoff"] hinfo = sys.hash_info FAKE_HASHINFO = namedtuple('FAKE_HASHINFO', fields) fd = dict() for f in fields: fd[f] = getattr(hinfo, f) fd['algorithm'] = 'fnv' fake_hashinfo = FAKE_HASHINFO(**fd) # replace the hashinfo with the fnv version sys.hash_info = fake_hashinfo with warnings.catch_warnings(record=True) as warns: # Cause all warnings to always be triggered. warnings.simplefilter("always") from numba import njit @njit def foo(): hash(1) foo() assert len(warns) > 0 expect = "FNV hashing is not implemented in Numba. See PEP 456" for w in warns: if expect in str(w.message): break else: raise RuntimeError("Expected warning not found") z-c) subprocess check_callsys executabler)selfworkrrrtest_warn_on_fnv!s&z!TestHashingSetup.test_warn_on_fnvN)__name__ __module__ __qualname__r rrrrrs rc@seZdZgdgdggdgdgdgdgdggdgd gd gd gd gd ZddZddZdddZddZdS) TestHashAlgs)\ r%r%r%)iL̇lLL//+?iTSlqU:~@[)0&Owq!~r&r')uu C -r(r))|#||#8 ^r*r+)i=վlBT='8i\el\e+=")i[l$E,;iwmHloa_gv)pGnYD~r,r-)SjxY_oWsr.r/)  /lr0r1)iJklJ:8`~il)wJ)iQ l .GUJGiO{l]D])djba33x siphash13 siphash24cCsf|tjjkr d}ntjj}tj }tjdkr|rdnd}n tjdks$J|r(dnd}|j|||S)Nr2littlerbig)r hash_infocutoff algorithmr IS_32BITS byteorder known_hashes)rpositionlengthr<IS_64BITplatformrrrget_expected_hashxs   zTestHashAlgs.get_expected_hashcCsd|S)Nzprint(hash(eval(%a)))r)rrepr_rrrget_hash_commandrzTestHashAlgs.get_hash_commandNcCsVtj}|durt||d<n|ddt|||d\}}|}t |S)NPYTHONHASHSEED)codeenv) osenvironcopystrpopr rFdecodestripint)rrEseedrIout_stdoutrrrget_hashs    zTestHashAlgs.get_hashc Cstd}|D]3\}}}|j||d|jt||d}||t|}|||Wdn1s2wYqdS)N))abcrr)rW*r6) abcdefghijkrXr9) äú∑ℇrr8)rZrX) input_strrR)rR)subTestrVreprrDlen assertEqual)rargsr\rRr@gotexpectedrrrtest_against_cpython_goldsz&TestHashAlgs.test_against_cpython_goldr)r!r"r#r?rDrFrVrdrrrrr$Ls*  #  r$c@sBeZdZddZddZejfddZddZd d Z d d Z d S)BaseTestcCstddt|_dS)NTnopython)rrcfuncrrrrsetUpszBaseTest.setUpc Cs|j}t|D]>}||}||tz ||t|WqtyE}ztdt||t|tdtt|t j d|d}~wwdS)Nzval, nb_hash, hash(val)z%abs(val), hashing._PyHASH_MODULUS - 1r6) rhlistassertIsInstancerQr`rAssertionErrorprintabsr_PyHASH_MODULUS)rvaluesrhvalnb_hasherrrcheck_hash_valuess  zBaseTest.check_hash_valuesccs~dD]9}t|}|j|kr|jksnqd}t|||Vt||d|dVt||d|dVdgVqdS)N)rii`ld)npiinfominmaxrange)rtypstartinfonrrr int_sampless  zBaseTest.int_samplescCs$ttdtt|j||S)NrQ)getattrryrMrzbitsview)rr~valuerrrsafe_constructs$zBaseTest.safe_constructccst|}dd|jd|jdfD]:}d}t|j||j}d|d|fD]%}||kr-q&t|||||}||}|V| V||Vq&qdddd td td  g}||VdS) Nr ?g@@rvg333333?ggginf) ryfinfor|tiny resolutionlinspaceastypemeanfloat)rr~rrrmin_stepsteparrr float_sampless    zBaseTest.float_samplesccsl||D]-}||D]%}|dt|}|dt|}||d|}tt|s2|Vq qdS)Ny?)rr_ryanyisnan)rr~float_tyrealimagrrrrcomplex_samplesszBaseTest.complex_samplesN) r!r"r#rjruryint64rrrrrrrrres  recsjeZdZdZfddZdfdd Zdd Zd d Zd d ZddZ ddZ ddZ e ddZ ZS)TestNumberHashingz' Test hashing of number types. c.tdkrtdkrt|_tdtSNr9rr9r6legacyrry_get_promotion_state initial_state_set_promotion_statesuperrjri __class__rrrj   zTestNumberHashing.setUpreturnNc&tdkrtdkrt|jtSNrrrryrrrtearDownrirrrr  zTestNumberHashing.tearDowncCs2||D]}||jt|||qdSr)rr`dtyperyru)rr~rrrr check_floatss zTestNumberHashing.check_floatscCs4|||D]}||jt|||qdSr)rr`rryru)rr~rrrrr check_complexs zTestNumberHashing.check_complexcCs|tj|tjdSr)rryfloat32float64rirrr test_floatss zTestNumberHashing.test_floatscCs$|tjtj|tjtjdSr)rry complex64r complex128rrirrr test_complexszTestNumberHashing.test_complexcCs|ddgdS)NFTrurirrr test_bool szTestNumberHashing.test_boolcCsg}tjtjtjtjtjtjtjtjfD]}| |D]}| |qt |}| | |dg| |dgdt |v}|j}|j|jfD]\}d|}|} t|D]O} d} d} | g} | | fD]}| | @}|tjkrs| |qc| D] }| ||gqv|r| D]}||jkr| || gq|dkr| dBd>} qV| d?} qVqJq| tdg| td g| tdg| td g| td g| td g| td g| td g| tdg| tdgdS)Nrxruint*UU*UU UU*UU*r6lllllii id)ryint8uint8int16uint16int32uint32ruint64rrurzrrMitemsizer{r|r}rmaxsizeappend)rminmaxtyrrsignedszrshiftsyitwiddle1twiddle2valstwrrvrrr test_intssZ         zTestNumberHashing.test_intscs<ddtdD}tfdd|D}t|ddS)NcSsg|]}tdqS)nan)r).0rrrr Oz9TestNumberHashing.test_py310_nan_hash..rcsg|]}|qSr)rh)rzrirrrPsr6)r}set assertGreaterr_)rrrSrrirtest_py310_nan_hashFs z%TestNumberHashing.test_py310_nan_hashrN)r!r"r#__doc__rjrrrrrrrr r __classcell__rrrrrs  7rcsFeZdZdZfddZdfdd Zdd Zd d Zd d ZZ S)TestTupleHashingz! Test hashing of tuples. crrrrirrrrjYrzTestTupleHashing.setUprNcrrrrirrrrbrzTestTupleHashing.tearDowncs*|D]}fdd|D}||qdS)Ncsg|]}|qSrr)rrsplitrrrlrz1TestTupleHashing.check_tuples..r)rvalue_generatorrrqtuplesrrr check_tuplesjs zTestTupleHashing.check_tuplescsTtjfdd}fdd}||gddS)Ncs$|}|d@|d@fS)z9 Split i's bits into 2 integers. rrrrrr~rrsplit2rs   z8TestTupleHashing.test_homogeneous_tuples..split2cs.|}|d@|d@|d@fS)z9 Split i's bits into 3 integers. l$$$$l$I$I$I$IlIIII rrrrrsplit3{s    z8TestTupleHashing.test_homogeneous_tuples..split3))r)rr)r)rr)r8)r[)ryrrrru)rrrrrrtest_homogeneous_tuplesos   z(TestTupleHashing.test_homogeneous_tuplescCs d}dd}|||dS)NlcSs4|d@}|d@|d?d@A}t|t|dfS)Nrl*UU g-C6?)ryrr)rrbrrrrsz9TestTupleHashing.test_heterogeneous_tuples..split)rr)rmodulorrrrtest_heterogeneous_tuplessz*TestTupleHashing.test_heterogeneous_tuplesr) r!r"r#rrjrrrrrrrrrrTs  rc@s>eZdZddZddZddZeddd Zd d Z d S) TestUnicodeHashingcCsd}tt|D] }||d|gqd}|t|}tt|D] }||d|gq$d}|t|}tt|D] }||d|gq@d}||dS)Nabcdefghijklmnopqrstuvwxyzu眼u🐍⚡)r}r_rujoinrk)r kind1_stringrsep kind2_string kind4_string empty_stringrrrtest_basic_unicodesz%TestUnicodeHashing.test_basic_unicodecCsDd}tdddd}t|d}||dk||||dS)NrTrfcSs|jSr)_hashrrrrfnsz4TestUnicodeHashing.test_hash_passthrough..fnrxrr assertTruer`rrr hash_valuerrrtest_hash_passthroughs   z(TestUnicodeHashing.test_hash_passthroughcCsHd}tdddd}t|d}||dk|||||fdS)NrTrfcSs|jt|fSr)rrrrrrrz9TestUnicodeHashing.test_hash_passthrough_call..fnrxrrrrrtest_hash_passthrough_calls   z-TestUnicodeHashing.test_hash_passthrough_callz/Needs hash computation at const unpickling timecCs8tdddd}|}td}||t|ddS)NTrfcSsd}|S)Nrrrrrrrsz0TestUnicodeHashing.test_hash_literal..fnrrx)rrr`r )rrrrtmprrrtest_hash_literals  z$TestUnicodeHashing.test_hash_literalcCsdd}tdd|}d}||}||}t|}t|}||dd|dd||d|dkd}||}||}t|}t|}|||dS)NcSs d}d}||}|rt||S)Naaaau 眼眼眼眼r)do_hashconst1const2newrrrimpls z:TestUnicodeHashing.test_hash_on_str_creation..implTrfFrx)rr r`r)rrjitted compute_hashrcrbrrrrrtest_hash_on_str_creationsz,TestUnicodeHashing.test_hash_on_str_creationN) r!r"r#rrrunittestskipr rrrrrrs   rc@seZdZddZddZdS)TestUnhashablec Csttjtjttjtdf}t ddt }|D].}| t  }||Wdn1s5wYdt t|d}||t |jqdS)Nr[Trfzunhashable type: '')rDictemptyrrList empty_listryonesrr assertRaises TypeErrorrMrassertIn exception)r unhashablesrhrraisesrcrrrtest_hash_unhashables  z#TestUnhashable.test_hash_unhashablecCs\tdddd}|t }|Wdn1swYd}||t|jdS)NTrfcSsttjdSr)rrycosrrrrfoorz0TestUnhashable.test_no_generic_hash..fooz"No __hash__ is defined for object )rrrrrMr )rr%r"rcrrrtest_no_generic_hashs  z#TestUnhashable.test_no_generic_hashN)r!r"r#r#r&rrrrrs r__main__)'rrrJrr collectionsrtextwraprnumpyrynumbarrrr numba.corerr numba.tests.supportr r r numba.cpython.unicoder numba.cpythonrnumba.np.numpy_supportrrrr$rerrrrr!mainrrrrs4     -SEpEZ