a v=ic@sddlZddlZddlZddlZddlZddlZddlmZddlm Z ddl Z ddl Z ddl Zzeed\ZZWneydZYn0dZeeee[[GdddejZed d e jd dd DZedddZddZddZdS)N)contextmanagerwrapsiFTc@sheZdZdZeddZeddZdd d Zdd d Zd dZ ddZ ddZ dddZ dddZ dS)TestCasez( Base class for unit tests. cCstjdd|_dS)Nz h5py-test_)prefix)tempfilemkdtemptempdirclsr b/home/droni/.local/share/virtualenvs/DPS-5Je3_V2c/lib/python3.9/site-packages/h5py/tests/common.py setUpClass/szTestCase.setUpClasscCst|jdSN)shutilrmtreer r r r r tearDownClass3szTestCase.tearDownClass.hdf5NcCs|dur|j}tj|||dS)N)dir)r rmktemp)selfsuffixrrr r r r7szTestCase.mktempcCsH|durddlm}|j}d}|dkr8||||}||d}|S)Nr)MPI)Zmpi4pyrZ COMM_WORLDZGet_rankrZbcast)rZcommrrrrfnamer r r mktemp_mpi<s   zTestCase.mktemp_mpicCst|d|_dS)Nw)h5pyFilerfrr r r setUpFszTestCase.setUpcCs&z|jr|jWn Yn0dSr)rcloser r r r tearDownIs zTestCase.tearDowncCsl|D].}d}|D]}||krd}q|std|q|D].}d}|D]}||krDd}qD|s8td|q8dS)NFTz Item '%s' appears in a but not bz Item '%s' appears in b but not a)AssertionError)rabxmatchyr r r assertSameElementsPszTestCase.assertSameElementscCsv|dur d}|durd}nd|}t|s6t|r|t|rJt|s\Jd|||f|||ksxJd||fdS|j|jksJd|j|j|f|j|jksJd|j|j|f|jjdur|jjD]*}d ||f}|j||||||d qnn|jjd vrFtt|d |d |ksrJd ||fn,t|d |d ksrJd|jj |fdS)z Make sure dset and arr have the same shape, dtype and contents, to within the given precision. Note that dset may be a NumPy array or an HDF5 dataset. Ngh㈵>rz (%s)z&Scalar/array mismatch ("%r" vs "%r")%sz"Scalars differ by more than %.3f%szShape mismatch (%s vs %s)%szDtype mismatch (%s vs %s)%sz [FIELD %s] %s)message precision)ir.z!Arrays differ by more than %.3f%sz"Arrays are not equal (dtype %s) %s) npZisscalarshapeZdtypenamesassertArrayEqualkindallabsstr)rdsetarrr+r,nr r r r1as<    $ zTestCase.assertArrayEqualFc Csd}z ||}Wn,ty<}zt|}WYd}~n d}~00t|trL|n|f}|dur|||||s||j||nl||||Wdn1s0Y|s|||j|Wdn1s0YdS)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 isinstancetupler1Z _fast_readerread assertRaises) rr6r7sZskip_fast_readerexcZ arr_resulteZs_fastr r r assertNumpyBehaviors$   & zTestCase.assertNumpyBehavior)rrN)NrrN)NN)F)__name__ __module__ __qualname____doc__ classmethodrrrrr!r#r*r1rBr r r r r)s     $rcCsg|] }t|qSr )int).0r-r r r rJ.rccshtjd|dd0}|j}|dur2|||Wdn1sF0Y|Vtj|dddS)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)rNamedTemporaryFilenamewriteflushrr)rtextZ test_file file_namer r r closed_tempfiles &rWcstfdd}|S)z!Runs a test in its own subprocessc sFtd|jj}d|}dD]}||d}q"tj|d}|r\|g|Ri|Sdtj|<tj}d||<| t dit }t |d,} t jtjdd |g| | |d } Wdn1s0Yt |d } | } Wdn1s0YWdn1s"0Y| d ksBJd | dS)Nz::ZIN_SUBPROCESS_z/\,:._1 subproc_envrNz-mZpytest)stdoutstderrenvrtr )inspect getsourcefilenoderRreplaceosenvirongetcopyupdategetattrrWopen subprocesscallsys executabler=) requestargskwargsZ curr_testZinsubcZdefinedr]r[fhZrtnoutrr r wrappers&   $ Hzinsubprocess..wrapperr)rrvr rur insubprocesssrwcsfdd}|S)z9Set environment variables for the @insubprocess decoratorcs |_|Sr)rZrudr r decoratorszsubproc_env..decoratorr )ryrzr rxr rZs rZ)rN)rmrdrr`rrk contextlibr functoolsrnumpyr.rZunittestutmkstempchrZtestfiler UnicodeErrorZUNICODE_FILENAMESr"unlinkrr< __version__splitZNUMPY_RELEASE_VERSIONrWrwrZr r r r  s2      |"