U id_@sBddlZddlZddlZddlZddlmZGdddejZdS)N)Pathc@seZdZddZdS)TestUnpickleDeletedModulec Csdddddg}d}tN}tj|t|d|d}t|d }||| d d l m }W5QRXtj |=t |}t |}||d d dS)aCreate a module that uses Numba, import a function from it. Then delete the module and pickle the function. The function should load from the pickle without a problem. Note - This is a simplified version of how Numba might be used on a distributed system using e.g. dask distributed. With the pickle being sent to the worker but not the original module.  zfrom numba import vectorizez @vectorize(['float64(float64)'])z def inc1(x):z return x + 1 tmp_module/z.pyar)inc1N)jointempfileTemporaryDirectorysyspathappendropenwritecloserrmodulespickledumpsloads assertEqual)selfsourcemodnameZtmp_dirfilenamefrZpklra/home/sam/Atlas/atlas_env/lib/python3.8/site-packages/numba/tests/test_unpickle_without_module.py"test_loading_pickle_with_no_module s&        zs