o G¾±hóã @s<ddlZddlZddlZddlZddlZzddlZWn ey'ejjZ Ynwddl Z de   ¡fde   ¡fde  ¡fde  ¡fde  ¡fde  ¡fde  ¡fd e  ¡fd e  ¡fd e  ¡fd e  ¡fg Zd d„Zdd„Zej de¡dd„ƒZdd„Zejjdd„ƒZdd„Zdd„Z dd„Z!dd„Z"d d!„Z#d"d#„Z$dS)$éNÚi1Úi2Úi4Úi8Úu1Úu2Úu4Úu8Úf2Úf4Úf8cCstj dd¡}tj |¡}|jdksJ‚|jgksJ‚|jdks"J‚|j |j ks*J‚|j |j ks2J‚|j s7J‚|j s.eqcSs*| |¡rJ‚||kr J‚||ksJ‚dSr\r]r^r+r+r,ÚneŸrbztest_tensor_equals..ner rWrrgð?)é rr)rrrrrrÚascontiguousarrayr!)rarcr'rYrZr+r+r,Útest_tensor_equals™s(  rfcCsJtjtddttj t d¡¡ƒWdƒdS1swYdS)NÚ unhashable)Úmatchr ) ÚpytestÚraisesÚ TypeErrorÚhashrrrrÚaranger+r+r+r,Útest_tensor_hashing®s"ÿrncCs8tj dd¡}tj |¡}tj |¡|jdksJ‚dS)Nr ré) rrrrrrrGÚget_tensor_sizer)r'r(r+r+r,Útest_tensor_size´s rqcCs”tj dd¡}tj |¡}tj |¡}tj   t |ƒd¡}t  ||¡}tj  ||¡tjdkr2| ¡tj|dd}tj |¡ ¡}tj ||¡dS)Nr rzpyarrow-tensor-ipc-read-tensorÚ emscriptenÚr)Úmode)rrrrrrrGrprBrCrDrErFrHr.ÚplatformrRÚ memory_maprJr0ÚtestingÚ assert_equal)rLr'r(Ú data_sizerCÚ write_mmapÚ read_mmapr2r+r+r,Útest_read_tensorºs    r|cCstjdftjdftjdftjdftjdffD]o\}}tjd|d}|j}| ¡}t j   |¡}t |ƒ}|j |ks9J‚|j|jksAJ‚|j|jksIJ‚|jdksPJ‚|j|jksXJ‚|j|jks`J‚|jd |jjkskJ‚t ||¡ ¡|kswJ‚~~t ||¡ ¡|ks…J‚qdS) Nz=bz=qz=QÚeÚdr )r6rro)rÚint8Úint64Úuint64Úfloat16Úfloat64rmr6ÚtolistrrrÚ memoryviewÚformatrrrÚnbytesÚitemsizer;Ú bit_widthÚ frombuffer)r6Úexpected_formatr'Úlstr(Úmr+r+r,Útest_tensor_memoryviewÌs, ü ìrŽ)%rBr.rir7r%ÚnumpyrÚ ImportErrorÚmarkÚ pytestmarkÚpyarrowrrÚint16Úint32r€Úuint8Úuint16Úuint32rr‚Úfloat32rƒÚtensor_type_pairsr-r3Ú parametrizerArNrOrVr[rfrnrqr|rŽr+r+r+r,ÚsH   ÿ           õ!