AVh"dZddlZddlZddlZddlZddlmZddlddlm Z ddl m Z ddl m Z ddl mZdd lmZdd lmZdd lmZe j*ZeZd ad ZddZdZdZdZedgGddeZy)z@Imports absltest as a replacement for testing.pybase.googletest.N)app)*)errors)file_io) benchmark) tf_logging) tf_decorator) tf_inspect) tf_exportct|y)zDelegate to absltest.main.argvN) absltest_mainrs U/home/dcms/DCMS/lib/python3.12/site-packages/tensorflow/python/platform/googletest.pyg_mainr0s Tc>fd}tj|y)Ncb}|tj}tjt|S)N)mainr)sysrrrunr)argsrs r main_wrapperzmain..main_wrapper8s' D | XXd 77T **r) true_mainr)rbenchmarks_main)rrs` rrr7s+  l>rcdts$tjjdr(t j tjd}nt jdd}tjjt jtjjt j|}t j |jd}|jdtj}|fd}t!j"||atS)z.Return a temporary directory for tests to use. TEST_TMPDIR)prefixrz.py/c tj|y#tj$r!}t j d||Yd}~yd}~wwxYw)NzError removing %s: %s)rdelete_recursivelyrOpErrorloggingerror)dirnamees rdelete_temp_dirz#GetTempDir..delete_temp_dirPs=;""7+ ^^; -w::;sA AA )_googletest_temp_dirosenvirongettempfilemkdtempr stackpathjoin gettempdirbasenamegetfilerstripreplacesepatexitregister)temp_dir first_framer)s r GetTempDirr=As  zz~~m$!!M)BCh$$&r*1-kh113 gg..z/A/A+/NOQh!!)?@hRVV,H (;  OOO$# rcftjjtjdd|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+r1r2r,) relative_paths rtest_src_dir_pathrA]s+ bjj/1= BBrcy)NFrCrrStatefulSessionAvailablerDks rztest.StubOutForTesting)v1cFeZdZdZdZdZdZdZdZdZ dZ d Z d Z y ) 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. c g|_g|_yN)cachestubsselfs r__init__zStubOutForTesting.__init__sDJDJrc$|jy)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. NCleanUprLs r__del__zStubOutForTesting.__del__s  LLNrc|SrIrCrLs r __enter__zStubOutForTesting.__enter__s Krc$|jyrIrP)rMunused_exc_typeunused_exc_value unused_tbs r__exit__zStubOutForTesting.__exit__s LLNrcD|j|jy)zDUndoes all SmartSet() & Set() calls, restoring original definitions.N) SmartUnsetAllUnsetAllrLs rrQzStubOutForTesting.CleanUpsMMOrctj|\}}tj|s#tj|s||j vr|}t ||}ntj|s)ttj|j}nttj|}|jd}d}|D]} | }t ||}d}|s td|j j|} | t| tr t|}|jj!||ft#|||y#t$rYwxYw)a"Replace obj.attr_name with new_attr. This method is smart and works at the module, class, and instance level while preserving proper inheritance. It will not stub out C types however unless that has been explicitly allowed by the type. This method supports the case where attr_name is a staticmethod or a classmethod of obj. Notes: - If obj is an instance, then it is its class that will actually be stubbed. Note that the method Set() does not do that: if obj is an instance, it (and not its class) will be stubbed. - The stubbing is using the builtin getattr and setattr. So, the __get__ and __set__ will be called when stubbing (TODO: A better idea would probably be to manipulate obj.__dict__ instead of getattr() and setattr()). Args: obj: The object whose attributes we want to modify. attr_name: The name of the attribute to modify. new_attr: The new value for the attribute. Raises: AttributeError: If the attribute cannot be found. NFTzAttribute not found.)r unwrapr ismoduleisclass__dict__getattrlistgetmro __class__reverseAttributeErrorr- isinstance staticmethodrKappendsetattr) rMobj attr_namenew_attr_orig_obj orig_attrmro found_attrcls old_attributes rSmartSetzStubOutForTesting.SmartSetsB6  %FAsC    $cll)Bh#y)i    $:$$S]]34:$$S)* kkmij# (c9-)* 344LL$$Y/M Z |%Ly)iJJxI67 Hi*   sE E%$E%cVt|jD] }t| g|_y)aKReverses SmartSet() calls, restoring things to original definitions. This method is automatically called when the StubOutForTesting() object is deleted; there is no need to call it explicitly. It is okay to call SmartUnsetAll() repeatedly, as later calls have no effect if no SmartSet() calls have been made. N)reversedrKrk)rMrs rr[zStubOutForTesting.SmartUnsetAlls,$ tnDJrct||}|jj|}|t|tr t |}|j j |||ft|||y)aIn parent, replace child_name's old definition with new_child. The parent could be a module when the child is a function at module scope. Or the parent could be a class when a class' method is being replaced. The named child is set to new_child, while the prior definition is saved away for later, when UnsetAll() is called. This method supports the case where child_name is a staticmethod or a classmethod of parent. Args: parent: The context in which the attribute child_name is to be changed. child_name: The name of the attribute to change. new_child: The new value of the attribute. N)rbrar-rhrirJrjrk)rMparent child_name new_child old_childrus rSetzStubOutForTesting.Setsb" +IOO'' 3M Z |%Ly)iJJvy*56 FJ *rcht|jD]\}}}t|||g|_y)aBReverses Set() calls, restoring things to their original definitions. This method is automatically called when the StubOutForTesting() object is deleted; there is no need to call it explicitly. It is okay to call UnsetAll() repeatedly, as later calls have no effect if no Set() calls have been made. N)rxrJrk)rMrzr}r{s rr\zStubOutForTesting.UnsetAlls6,4DJJ+?-'J fj),-DJrN) __name__ __module__ __qualname____doc__rNrRrTrYrQrvr[r~r\rCrrrGrGos5  =+~ +4 rrGrI) rr9r+rr.abslrabsl.testing.absltesttensorflow.python.frameworkrtensorflow.python.lib.iortensorflow.python.platformrrr%tensorflow.python.utilr r tensorflow.python.util.tf_exportr TensorFlowBenchmark Benchmarkrrr*rr=rArDobjectrGrCrrrs G #/,0</-6  ) )   ?8 B '()]]*]r