o h @s<ddlZddlZddlZddlZddlZzddlZWn ey'ejjZ Ynwddl Z de fde fde fde fde fde fde fd e fd e fd e fd e fg Zd dZddZejdeddZddZejjddZddZddZ ddZ!ddZ"d d!Z#d"d#Z$dS)$Ni1i2i4i8u1u2u4u8f2f4f8cCstjdd}tj|}|jdksJ|jgksJ|jdks"J|j |j ks*J|j |j ks2J|j s7J|j s.eqcSs*||rJ||kr J||ksJdSr\r]r^r+r+r,nerbztest_tensor_equals..ner rWrrg?) rr)rrrrrrascontiguousarrayr!)rarcr'rYrZr+r+r,test_tensor_equalss(  rfcCsJtjtddttjtdWddS1swYdS)N unhashable)matchr ) pytestraises TypeErrorhashrrrraranger+r+r+r,test_tensor_hashings"rncCs8tjdd}tj|}tj||jdksJdS)Nr r) rrrrrrrGget_tensor_sizer)r'r(r+r+r,test_tensor_sizes rqcCstjdd}tj|}tj|}tj t |d}t ||}tj ||tjdkr2|tj|dd}tj|}tj||dS)Nr rzpyarrow-tensor-ipc-read-tensor emscriptenr)mode)rrrrrrrGrprBrCrDrErFrHr.platformrR memory_maprJr0testing assert_equal)rLr'r( data_sizerC write_mmap read_mmapr2r+r+r,test_read_tensors    r|cCstjdftjdftjdftjdftjdffD]o\}}tjd|d}|j}|}t j |}t |}|j |ks9J|j|jksAJ|j|jksIJ|jdksPJ|j|jksXJ|j|jks`J|jd |jjkskJt|||kswJ~~t|||ksJqdS) Nz=bz=qz=Qedr )r6rro)rint8int64uint64float16float64rmr6tolistrrr memoryviewformatrrrnbytesitemsizer; bit_width frombuffer)r6expected_formatr'lstr(mr+r+r,test_tensor_memoryviews,  r)%rBr.rir7r%numpyr ImportErrormark pytestmarkpyarrowrrint16int32ruint8uint16uint32rrfloat32rtensor_type_pairsr-r3 parametrizerArNrOrVr[rfrnrqr|rr+r+r+r,sH              !