aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/python/platform
diff options
context:
space:
mode:
authorGravatar Dandelion Man? <dandelion@google.com>2017-06-01 17:11:04 -0700
committerGravatar TensorFlower Gardener <gardener@tensorflow.org>2017-06-01 17:16:33 -0700
commitd741d81c5f3412dd97da25a07574c327b2f8b1fc (patch)
tree3647b3174ddb5b5df25ea2489b1b24d0c4689552 /tensorflow/python/platform
parent40411cd5c68e4f91a1fc0d5861fac88b404329bb (diff)
Expose tf.test.StubOutForTesting in the tf testing api
Also redirect TensorBoard usage to use that endpoint. This is part of my ongoing effort to have TensorBoard only depend on TensorFlow via its public api, so that it can be split into a project with a fast external build. PiperOrigin-RevId: 157784552
Diffstat (limited to 'tensorflow/python/platform')
-rw-r--r--tensorflow/python/platform/test.py4
1 files changed, 4 insertions, 0 deletions
diff --git a/tensorflow/python/platform/test.py b/tensorflow/python/platform/test.py
index 5cb2c152b0..a307347f60 100644
--- a/tensorflow/python/platform/test.py
+++ b/tensorflow/python/platform/test.py
@@ -61,6 +61,9 @@ else:
# Import Benchmark class
Benchmark = _googletest.Benchmark # pylint: disable=invalid-name
+# Import StubOutForTesting class
+StubOutForTesting = _googletest.StubOutForTesting # pylint: disable=invalid-name
+
def main(argv=None):
"""Runs all unit tests."""
@@ -117,6 +120,7 @@ _allowed_symbols = [
# We piggy-back googletest documentation.
'Benchmark',
'mock',
+ 'StubOutForTesting',
]
remove_undocumented(__name__, _allowed_symbols)