diff options
| author |  Dandelion Man? <dandelion@google.com> | 2017-06-01 17:11:04 -0700 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2017-06-01 17:16:33 -0700 |
| commit | d741d81c5f3412dd97da25a07574c327b2f8b1fc (patch) | |
| tree | 3647b3174ddb5b5df25ea2489b1b24d0c4689552 /tensorflow/python/platform | |
| parent | 40411cd5c68e4f91a1fc0d5861fac88b404329bb (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.py | 4 |
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) |