o ?eH@sXdZddlmZddlmZddlmZddlm Z ddlm Z ddlm Z ddlm Z dd lmZdd lmZdd lZdd lZdd lmZejjd krQdd lZnddlmZedgdeejZejZedd)ddZedgdddZedgdddZedddZedddZed d!d"Z ed#d$d%Z!ed&d'd(Z"d S)*zTesting.) test_util) googletest)assert_equal_graph_def)create_local_cluster)TensorFlowTestCase)gpu_device_name)is_gpu_available)compute_gradient_error)compute_gradientN) tf_export)mockz test.mock)v1z test.maincCstt|S)zRuns all unit tests.) _test_utilZInstallStackTraceHandler _googletestmain)argvr`/home/www/facesmatcher.com/pyenv/lib/python3.10/site-packages/tensorflow/python/platform/test.pyr4s rztest.get_temp_dircCtS)aReturns a temporary directory for use during tests. There is no need to delete the directory after the test. @compatibility(TF2) This function is removed in TF2. Please use `TestCase.get_temp_dir` instead in a test case. Outside of a unit test, obtain a temporary directory through Python's `tempfile` module. @end_compatibility Returns: The temporary directory. )rZ GetTempDirrrrr get_temp_dir;srztest.test_src_dir_pathcCs t|S)zCreates an absolute test srcdir path given a relative path. Args: relative_path: a path relative to tensorflow root. e.g. "core/platform". Returns: An absolute path to the linked in runfiles. )rtest_src_dir_path)Z relative_pathrrrrNs rztest.is_built_with_cudacCr)a%Returns whether TensorFlow was built with CUDA (GPU) support. This method should only be used in tests written with `tf.test.TestCase`. A typical usage is to skip tests that should only run with CUDA (GPU). >>> class MyTest(tf.test.TestCase): ... ... def test_add_on_gpu(self): ... if not tf.test.is_built_with_cuda(): ... self.skipTest("test is only applicable on GPU") ... ... with tf.device("GPU:0"): ... self.assertEqual(tf.math.add(1.0, 2.0), 3.0) TensorFlow official binary is built with CUDA. )rZIsGoogleCudaEnabledrrrris_built_with_cuda\rztest.is_built_with_rocmcCr)a)Returns whether TensorFlow was built with ROCm (GPU) support. This method should only be used in tests written with `tf.test.TestCase`. A typical usage is to skip tests that should only run with ROCm (GPU). >>> class MyTest(tf.test.TestCase): ... ... def test_add_on_gpu(self): ... if not tf.test.is_built_with_rocm(): ... self.skipTest("test is only applicable on GPU") ... ... with tf.device("GPU:0"): ... self.assertEqual(tf.math.add(1.0, 2.0), 3.0) TensorFlow official binary is NOT built with ROCm. )rZIsBuiltWithROCmrrrris_built_with_rocmqrrztest.disable_with_predicatecsfdd}|S)z"Disables the test if pred is true.cstfdd}|S)Ncs*r |dS|g|Ri|SN)ZskipTest)selfargskwargs)funcpred skip_messagerrwrapper_disable_with_predicateszhdisable_with_predicate..decorator_disable_with_predicate..wrapper_disable_with_predicate) functoolswraps)rr"r r!)rr decorator_disable_with_predicatesz@disable_with_predicate..decorator_disable_with_predicater)r r!r&rr%rdisable_with_predicates r'ztest.is_built_with_gpu_supportcCs tptS)a9Returns whether TensorFlow was built with GPU (CUDA or ROCm) support. This method should only be used in tests written with `tf.test.TestCase`. A typical usage is to skip tests that should only run with GPU. >>> class MyTest(tf.test.TestCase): ... ... def test_add_on_gpu(self): ... if not tf.test.is_built_with_gpu_support(): ... self.skipTest("test is only applicable on GPU") ... ... with tf.device("GPU:0"): ... self.assertEqual(tf.math.add(1.0, 2.0), 3.0) TensorFlow official binary is built with CUDA GPU support. )rrrrrris_built_with_gpu_supports r(ztest.is_built_with_xlacCr)a`Returns whether TensorFlow was built with XLA support. This method should only be used in tests written with `tf.test.TestCase`. A typical usage is to skip tests that should only run with XLA. >>> class MyTest(tf.test.TestCase): ... ... def test_add_on_xla(self): ... if not tf.test.is_built_with_xla(): ... self.skipTest("test is only applicable on XLA") ... @tf.function(jit_compile=True) ... def add(x, y): ... return tf.math.add(x, y) ... ... self.assertEqual(add(tf.ones(()), tf.ones(())), 2.0) TensorFlow official binary is built with XLA. )rZIsBuiltWithXLArrrris_built_with_xlasr)r)#__doc__Ztensorflow.python.frameworkrrZtensorflow.python.platformrrZ%tensorflow.python.framework.test_utilrrrZTestCaserrZ&tensorflow.python.ops.gradient_checkerr r r#sysZ tensorflow.python.util.tf_exportr version_infomajorr ZunittestZ BenchmarkZStubOutForTestingrrrrrr'r(r)rrrrsF