U md@sddlZddlZddlZddlZddlZddlZddlmZddlm Z ddl Z ddl Z ddl Zzeed\ZZWnek rdZYnXdZeeee[[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 J/home/sam/Atlas/atlas_env/lib/python3.8/site-packages/h5py/tests/common.py setUpClass/szTestCase.setUpClasscCst|jdSN)shutilrmtreer r r r r tearDownClass3szTestCase.tearDownClass.hdf5NcCs|dkr|j}tj|||dS)N)dir)r rmktemp)selfsuffixrrr r r r7szTestCase.mktempcCsH|dkrddlm}|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 YnXdSr)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|dkr d}|dkrd}nd|}t|s6t|r|t|rJt|s\td|||f|||ksxtd||fdS|j|jkstd|j|j|f|j|jkstd|j|j|f|jjdk r|jjD]*}d ||f}|j||||||d qnn|jjd krFtt |d |d |ksrtd ||fn,t|d |d ksrtd|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) npZisscalarr$shapeZdtypenamesassertArrayEqualkindallabsstr)rdsetarrr+r,nr r r r1as<    $ zTestCase.assertArrayEqualFc Csd}z ||}Wn*tk r:}z t|}W5d}~XYnXt|trJ|n|f}|dkr|||||s||j||nD||||W5QRX|s|||j|W5QRXdS)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 srJ.rc csTtjd|dd&}|j}|dk r2|||W5QRX|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 rVcstfdd}|S)z!Runs a test in its own subprocessc std|jj}d|}dD]}||d}q"tj|d}|rV|f||Sdtj|<tj}d||<| t dit V}t |d"} t jtjdd |g| | |d } W5QRXt |d } | } W5QRXW5QRX| d kstd | dS)Nz::ZIN_SUBPROCESS_z/\,:._1 subproc_envrMz-mZpytest)stdoutstderrenvrtr )inspect getsourcefilenoderQreplaceosenvirongetcopyupdategetattrrVopen subprocesscallsys executabler=r$) requestargskwargsZ curr_testZinsubcZdefinedr\rZfhZrtnoutrr r wrappers*    zinsubprocess..wrapperr)rrur rtr insubprocesssrvcsfdd}|S)z9Set environment variables for the @insubprocess decoratorcs |_|Sr)rYrtdr r decoratorszsubproc_env..decoratorr )rxryr rwr rYs rY)rN)rlrcrr_rrj contextlibr functoolsrnumpyr.rZunittestutmkstempchrZtestfiler UnicodeErrorZUNICODE_FILENAMESr"unlinkrr< __version__splitZNUMPY_RELEASE_VERSIONrVrvrYr r r r  s2     |"