o à8Vaã@s.dZddlZddlZGdd„dƒZdd„ZdS)zN This module adds context manager for temporary files generated by the tests. éNc@sHeZdZdZedƒZedƒZed dd„ƒZed dd„ƒZ edd„ƒZ d S) ÚTmpFileManagerzP A class to track record of every temporary files created by the tests. ÚcCó|j |¡|S©N)Ú tmp_filesÚadd©ÚclsÚname©r ú8/usr/lib/python3/dist-packages/sympy/testing/tmpfiles.pyÚtmp_fileó zTmpFileManager.tmp_filecCrr)Ú tmp_foldersrrr r r Ú tmp_folderrzTmpFileManager.tmp_foldercCsT|jr|j ¡}tj |¡rt |¡|js|jr(|j ¡}t |¡|jsdSdSr) rÚpopÚosÚpathÚisfileÚremoverÚshutilÚrmtree)r ÚfileZfolderr r r Úcleanups   ý  þzTmpFileManager.cleanupN)r) Ú__name__Ú __module__Ú __qualname__Ú__doc__ÚsetrrÚ classmethodr rrr r r r r s  rcs‡fdd„}|S)zP A decorator to help test codes remove temporary files after the tests. cs z ˆƒWt ¡dSt ¡wr)rrr ©Ú test_funcr r Úwrapper_function(sz+cleanup_tmp_files..wrapper_functionr )r!r"r r r Úcleanup_tmp_files$s r#)rrrrr#r r r r Ús