o 2h"@sdZddlZddlZddlZddlZddlmZddlTddlm Z ddl m Z ddl m Z ddl mZdd lmZdd lmZdd lmZe jZeZd ad dZdddZddZddZddZedgdGdddeZdS)z@Imports absltest as a replacement for testing.pybase.googletest.N)app)*)errors)file_io) benchmark) tf_logging) tf_decorator) tf_inspect) tf_exportcCst|ddS)zDelegate to absltest.main.argvN) absltest_mainr r`/var/www/html/chatgem/venv/lib/python3.10/site-packages/tensorflow/python/platform/googletest.pyg_main0srcsfdd}tj|ddS)Ncs }|dur tj}tjt|dS)N)mainr )sysr rrunr)argsr rr main_wrapper8szmain..main_wrapper) true_mainr )rbenchmarks_main)r rrr rr7s rcCstsGtjdrtjtjdd}n!tdd}tj t tj t |}tj| dd}|dtj}|fdd}t||atS) z.Return a temporary directory for tests to use. TEST_TMPDIR)prefixrz.py/c SsHzt|WdStjy#}ztd||WYd}~dSd}~ww)NzError removing %s: %s)rdelete_recursivelyrOpErrorloggingerror)dirnameerrrdelete_temp_dirPs z#GetTempDir..delete_temp_dir)_googletest_temp_dirosenvirongettempfilemkdtempr stackpathjoin gettempdirbasenamegetfilerstripreplacesepatexitregister)temp_dir first_framer#rrr GetTempDirAs    r7cCstjtjdd|S)zCreates an absolute test srcdir path given a relative path. Args: relative_path: a path relative to tensorflow root. e.g. "contrib/session_bundle/example". Returns: An absolute path to the linked in runfiles. TEST_SRCDIRzorg_tensorflow/tensorflow)r%r+r,r&) relative_pathrrrtest_src_dir_path]s r:cCsdS)NFrrrrrStatefulSessionAvailablekr;ztest.StubOutForTesting)v1c@sXeZdZdZddZddZddZdd Zd d Zd d Z ddZ ddZ ddZ dS)StubOutForTestingaSupport class for stubbing methods out for unit testing. Sample Usage: You want os.path.exists() to always return true during testing. stubs = StubOutForTesting() stubs.Set(os.path, 'exists', lambda x: 1) ... stubs.CleanUp() The above changes os.path.exists into a lambda that returns 1. Once the ... part of the code finishes, the CleanUp() looks up the old value of os.path.exists and restores it. cCsg|_g|_dSN)cachestubsselfrrr__init__s zStubOutForTesting.__init__cCs |dS)zDo not rely on the destructor to undo your stubs. You cannot guarantee exactly when the destructor will get called without relying on implementation details of a Python VM that may change. NCleanUprBrrr__del__s zStubOutForTesting.__del__cCs|Sr?rrBrrr __enter__r<zStubOutForTesting.__enter__cCs |dSr?rE)rCunused_exc_typeunused_exc_value unused_tbrrr__exit__s zStubOutForTesting.__exit__cCs||dS)zDUndoes all SmartSet() & Set() calls, restoring original definitions.N) SmartUnsetAllUnsetAllrBrrrrFs zStubOutForTesting.CleanUpc Cst|\}}t|st|s||jvr|}t||}nos ? r>r?) rqr3r%rr(abslrabsl.testing.absltesttensorflow.python.frameworkrtensorflow.python.lib.iortensorflow.python.platformrrrtensorflow.python.utilrr tensorflow.python.util.tf_exportr TensorFlowBenchmark Benchmarkrrr$rr7r:r;objectr>rrrrs0