o gi>$@sJddlZddlZddlZddlZddlZddlZddlZddlmZddl m Z ddl Z ddl mZddlZddlmZddlZz eed\ZZWn eyVdZYnwdZeeee[[Gd d d ejZed d e jd ddDZ edddZ!ddZ"ddZ#e$Z%dde&de&fddZ'de(fddZ)dS)N)contextmanagerwraps) repack_fields)philiFTc@sheZdZdZeddZeddZdd d Zdd d ZddZ ddZ ddZ dddZ dddZ dS)TestCasez( Base class for unit tests. cCstjdd|_dS)Nz h5py-test_)prefix)tempfilemkdtemptempdirclsrO/home/ubuntu/SoloSpeech/.venv/lib/python3.10/site-packages/h5py/tests/common.py setUpClass1szTestCase.setUpClasscCst|jdSN)shutilrmtreer r rrr tearDownClass5szTestCase.tearDownClass.hdf5tmpNcCs"|dur|j}tj|t||dS)N)dir)r r mktemp make_name)selfsuffixrrrrrr9szTestCase.mktempcCsH|dur ddlm}|j}d}|dkr||||}||d}|S)Nr)MPI)mpi4pyr COMM_WORLDGet_rankrbcast)rcommrrrrfnamerrr mktemp_mpi>s   zTestCase.mktemp_mpicCst|d|_dS)Nw)h5pyFilerfrrrrsetUpHszTestCase.setUpcCs*z|jr |jWdSWdSYdSr)r(closer)rrrtearDownKs zTestCase.tearDowncCsl|D]}d}|D]}||krd}q|std|q|D]}d}|D]}||kr*d}q"|s3td|qdS)NFTz Item '%s' appears in a but not bz Item '%s' appears in b but not a)AssertionError)rabxmatchyrrrassertSameElementsRs&  zTestCase.assertSameElementsTc Cs|durd}|dur d}nd|}t|st|r8t|r%t|s.Jd|||ft|}t|}|j|jksIJd|j|j|f|j|jkr|rX|j}|j}n t|j}t|j}||ksoJd|||f|s||jkr}tt|}||jkrtt|}|jjdur|jjD]}d||f}|j|||||||d qdS|jjd vrt t |d |d |ksJd ||fdS|jjd krt |j |j ddD]\} } |j| | |||d qdSt |d |d ksJd|jj |fdS)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㈵>rz (%s)z&Scalar/array mismatch ("%r" vs "%r")%szShape mismatch (%s vs %s)%szDtype mismatch (%s vs %s)%sz [FIELD %s] %s)message precisioncheck_alignment)ir(.z!Arrays differ by more than %.3f%sOT)strictz"Arrays are not equal (dtype %s) %s)npisscalarasarrayshapedtypernamesassertArrayEqualkindallabszipflatstr) rdsetarrr4r5r6normalized_dset_dtypenormalized_arr_dtypenv1v2rrrr@csX              "  zTestCase.assertArrayEqualFc CsZd}z||}Wnty}z t|}WYd}~nd}~wwt|tr&|n|f}|durX|||||sVt||j||WddS1sOwYdSdS|| ||Wdn1slwY|s||)t|j|Wdn1swYWddSWddS1swYdSdS)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 isinstancetupler@r _fast_readerread assertRaises) rrGrHsskip_fast_readerexc arr_resultes_fastrrrassertNumpyBehaviors4  "  PzTestCase.assertNumpyBehavior)rrN)NrrN)NNT)F)__name__ __module__ __qualname____doc__ classmethodrrrr$r*r,r3r@r[rrrrr+s      4rcCsg|]}t|qSr)int).0r7rrr src.rccsjtjd|dd}|j}|dur|||Wdn1s$wY|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)r NamedTemporaryFilenamewriteflushrr)rtext test_file file_namerrrclosed_tempfiles rpcstfdd}|S)z!Runs a test in its own subprocessc s6td|jj}d|}dD]}||d}qtj|d}|r.|g|Ri|Stj}d||<| t dit D}t |d} t jtjdd |g| | |d } Wdn1sbwYt |d  } | } Wdn1s{wYWdn1swY| d ksJd | dS)Nz::IN_SUBPROCESS_z/\,:._1 subproc_envrfz-mpytest)stdoutstderrenvrtr )inspect getsourcefilenoderjreplaceosenvirongetcopyupdategetattrrpopen subprocesscallsys executablerS) requestargskwargs curr_testinsubcdefinedrxrvfhrtnoutr(rrwrappers,    zinsubprocess..wrapperr)r(rrrr insubprocesssrcsfdd}|S)z9Set environment variables for the @insubprocess decoratorcs |_|Sr)rtrdrr decoratorszsubproc_env..decoratorr)rrrrrrts rtfootemplate_or_prefixreturncCs8t}|tkr dnd|}d|vr||S||S)aReturn a static name, to be used e.g. as dataset name. When running in pytest-run-parallel, append a thread ID to the name. This allows running tests on shared resources, e.g. two threads can attempt to write to separate datasets on the same File at the same time (even though the actual writes will be serialized by the `phil` lock). Calling this function twice from the same thread will return the same name. Parameters ---------- template_or_prefix Either a prefix to which potentially append the thread ID, or a template containing exactly one "{}" to be replaced with the thread ID. r-z{}) threading get_identMAIN_THREAD_IDformat)rtidrrrrrs  rcCst}|tkS)zReturn True if the test calling this function is being executed in the main thread; False otherwise. This can be used to detect when a test is running in pytest-run-parallel. that spawns multiple separate threads to run the tests. )rrr)rrrris_main_threadsr)rN)r)*rrrr{r rr contextlibr functoolsrnumpyr:numpy.lib.recfunctionsrr& h5py._objectsrunittestutmkstempchrtestfiler# UnicodeErrorUNICODE_FILENAMESr+unlinkrrQ __version__splitNUMPY_RELEASE_VERSIONrprrtrrrFrboolrrrrrsD        "