1VhXddlZddlZddlZddlZddlZddlZddlmZddlm Z ddl Z ddl m Z ddlZddlZ ej"ed\ZZdZej,eej.e[[Gdd ej2Zee j6j9d dd Dcgc] }e| c}Zedd Zd Z dZ!y#e$rdZYfwxYwcc}w)N)contextmanagerwraps) repack_fieldsiTFcbeZdZdZedZedZd dZd dZdZ dZ d Z dd Z dd Z y)TestCasez( Base class for unit tests. c:tjd|_y)Nz h5py-test_)prefix)tempfilemkdtemptempdirclss A/home/dcms/DCMS/lib/python3.12/site-packages/h5py/tests/common.py setUpClasszTestCase.setUpClass0s&&l; cBtj|jyN)shutilrmtreer rs r tearDownClasszTestCase.tearDownClass4s ckk"rNcN| |j}tj|||S)N)dir)r r mktemp)selfsuffixr rs rrzTestCase.mktemp8s# ;,,Cvv377rc|ddlm}|j}d}|jdk(r|j |||}|j |d}|S)Nr)MPI)mpi4pyr COMM_WORLDGet_rankrbcast)rcommrr rrfnames r mktemp_mpizTestCase.mktemp_mpi=sM < ">>D ==?a KK4E 5!$ rcVtj|jd|_y)Nw)h5pyFilerfrs rsetUpzTestCase.setUpGs4;;=#.rcb |jr|jjyy#YyxYwr)r*closer+s rtearDownzTestCase.tearDownJs* vv   s&*.c|D]#}d}|D] }||k(s d} |rtd|z|D]#}d}|D] }||k(s d} |rtd|zy)NFTz Item '%s' appears in a but not bz Item '%s' appears in b but not a)AssertionError)rabxmatchys rassertSameElementszTestCase.assertSameElementsQs MAE !6 E !$%G!%KLL  M MAE !6 E !$%G!%KLL  Mrc|d}|d}nd|z}tj|stj|rdtj|rtj|sJd|d|d|tj|}tj|}|j|jk(s$Jd|jd |jd ||j|jk7r|r|j}|j}n*t |j}t |j}||k(sJd |d |d ||sZ||jk7rt tj|}||jk7rt tj|}|jj @|jj D]&}d |d |}|j|||||||(y|jjdvrAtjtj|d|dz |ks Jd||fzy|jjdk(r?t|j|jD]\} } |j| | |||ytj|d|dk(s!Jd|jjd|y)z Make sure dset and arr have the same shape, dtype and contents, to within the given precision, optionally ignoring differences in dtype alignment. Note that dset may be a NumPy array or an HDF5 dataset. Ngh㈵>z (%s)zScalar/array mismatch ("z" vs "z")zShape mismatch (z vs )zDtype mismatch (z[FIELD z] )message precisioncheck_alignment)ir*.z!Arrays differ by more than %.3f%sOzArrays are not equal (dtype z) )npisscalarasarrayshapedtypernamesassertArrayEqualkindallabszipflatstr) rdsetarrr;r<r=normalized_dset_dtypenormalized_arr_dtypenv1v2s rrFzTestCase.assertArrayEqualbs  I ?G'G ;;t  C 0;;t$S)9 P<@#wO P9::d#D**S/CzzSYY& M-1ZZG L M& :: "(, %'*yy$(5djj(A%'4SYY'?$(,@@ g1FH\^ef g@#(DJJ6(D)9:D'3994' 38C 99?? &YY__ ~-.8%%d1gs1vwR[m|%} ~YY^^z )66"&&cSX!56BC K3y'6JJ KC YY^^s "dii2 uB%%b"gds%t u66$s)s3x/0 P8; wO P0rcd} ||}t|tr|n|f}|D|j |||s,|j |j j ||yy|j|5||ddd|s6|j|5|j j |dddyy#t$r}t|}Yd}~d}~wwxYw#1swYcxYw#1swYyxYw)a Apply slicing arguments "s" to both dset and arr. Succeeds if the results of the slicing are identical, or the exception raised is of the same type for both. "arr" must be a Numpy array; "dset" may be a NumPy array or dataset. N) Exceptiontype isinstancetuplerF _fast_readerread assertRaises) rrMrNsskip_fast_readerexc arr_resultes_fasts rassertNumpyBehaviorzTestCase.assertNumpyBehaviors QJ!E* ;  ! !$q': 6#%%%%**62$ ""3' Q $&&s+3%%**6233$# q'C   33s/B<5CC)< C CCC&)C2).hdf5r9N)Nrcr9N)NNT)F)__name__ __module__ __qualname____doc__ classmethodrrrr%r,r/r7rFrbrrrr*sU<<##8 /M"2Ph3rr.c#Ktjd|d5}|j}|!|j||j dddt j |dy#1swY%xYww)z Context manager which yields the path to a closed temporary file with the suffix `suffix`. The file will be deleted on exiting the context. An additional argument `text` can be provided to have the file contain `text`. w+tF)rdeleteNT) ignore_errors)r NamedTemporaryFilenamewriteflushrr)rtext test_file file_names rclosed_tempfilerwsn  $ $ fU  NN   OOD ! OO   O MM)40sA;0A/ $A;/A84A;c.tfd}|S)z!Runs a test in its own subprocessctj dz|jjz}d|z}dD]}|j |d}t j j|d}|r  |g|i|Sdt j |<t j j}d||<|jt dit5}t|d5} tjtj dd |g| | | } dddt|d 5} | j#} dddddd d k(s Jd  zy#1swYFxYw#1swY-xYw#1swY1xYw)Nz::IN_SUBPROCESS_z/\,:._1 subproc_envrmz-mpytest)stdoutstderrenvrtr )inspect getsourcefilenoderqreplaceosenvirongetcopyupdategetattrrwopen subprocesscallsys executablerZ) requestargskwargs curr_testinsubcdefinedrrfhrtnoutr*s rwrapperzinsubprocess..wrappersd))!,t3gll6G6GG  9, *AMM!S)E ***..- W.t.v. . #BJJu **//#CCJ JJwq-4 5 " $f&%(IB$//3>>49*U13BCICI&$'$2'')C$  $!8 'TCZ '8 II$$  $ $s< E/&,EE/&E#7E/E E/#E, (E//E8r)r*rs` r insubprocessrs  1X((0 Nrcfd}|S)z9Set environment variables for the @insubprocess decoratorc|_|Sr)r})r*ds r decoratorzsubproc_env..decorators rri)rrs` rr}r}s r)r9N)"rrrrr r contextlibr functoolsrnumpyr@numpy.lib.recfunctionsrr(unittestutmkstempchrtestfiler$UNICODE_FILENAMESr.unlink UnicodeErrorrrX __version__splitintNUMPY_RELEASE_VERSIONrwrr})r>s0rrs %0  &h&&s6{3OHe BHHX BIIe J3r{{J3Xr~~/C/CC/H1/MN!s1vNO11"<QlOsC2C%C"!C"