U L?h@s.dZddlZddlZGdddZddZdS)zN This module adds context manager for temporary files generated by the tests. Nc@sHeZdZdZedZedZed ddZed ddZ eddZ d S) TmpFileManagerzP A class to track record of every temporary files created by the tests. cCs|j||SN) tmp_filesaddclsnamer H/var/www/html/venv/lib/python3.8/site-packages/sympy/testing/tmpfiles.pytmp_files zTmpFileManager.tmp_filecCs|j||Sr) tmp_foldersrrr r r tmp_folders zTmpFileManager.tmp_foldercCsH|jr(|j}tj|rt|q|jrD|j}t|q(dSr) rpopospathisfileremover shutilrmtree)rfilefolderr r r cleanups    zTmpFileManager.cleanupN)r)r) __name__ __module__ __qualname____doc__setrr classmethodr rrr r r r r s  rcsfdd}|S)zP A decorator to help test codes remove temporary files after the tests. csz W5tXdSr)rrr  test_funcr r wrapper_function(s z+cleanup_tmp_files..wrapper_functionr )r r!r rr cleanup_tmp_files$s r")rrrrr"r r r r s