o ?e"@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 rf/home/www/facesmatcher.com/pyenv/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)Z true_mainr )rZbenchmarks_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.Z TEST_TMPDIR)prefixrz.py/c SsHzt|WdStjy#}ztd||WYd}~dSd}~ww)NzError removing %s: %s)rZdelete_recursivelyrZOpErrorloggingerror)dirnameerrrdelete_temp_dirPs z#GetTempDir..delete_temp_dir)_googletest_temp_dirosenvirongettempfilemkdtempr stackpathjoin gettempdirbasenamegetfilerstripreplacesepatexitregister)temp_dirZ first_framerrrr GetTempDirAs    r1cCstjtjdd|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. Z TEST_SRCDIRzorg_tensorflow/tensorflow)r r&r'r!)Z relative_pathrrrtest_src_dir_path]s r2cCsdS)NFrrrrrStatefulSessionAvailablekr3ztest.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. NCleanUpr:rrr__del__s zStubOutForTesting.__del__cCs|Sr7rr:rrr __enter__r4zStubOutForTesting.__enter__cCs |dSr7r=)r;Zunused_exc_typeZunused_exc_valueZ unused_tbrrr__exit__s zStubOutForTesting.__exit__cCs||dS)zDUndoes all SmartSet() & Set() calls, restoring original definitions.N) SmartUnsetAllUnsetAllr:rrrr>s zStubOutForTesting.CleanUpc Cst|\}}t|st|s||jvr|}t||}nrXrBr^rCrrrrr6os ? r6r7) rbr.r rr#ZabslrZabsl.testing.absltestZtensorflow.python.frameworkrZtensorflow.python.lib.iorZtensorflow.python.platformrrrZtensorflow.python.utilrr Z tensorflow.python.util.tf_exportr ZTensorFlowBenchmarkZ Benchmarkrrrrr1r2r3objectr6rrrrs0