aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/tools/docs/build_docs_test.py
diff options
context:
space:
mode:
Diffstat (limited to 'tensorflow/tools/docs/build_docs_test.py')
-rw-r--r--tensorflow/tools/docs/build_docs_test.py6
1 files changed, 4 insertions, 2 deletions
diff --git a/tensorflow/tools/docs/build_docs_test.py b/tensorflow/tools/docs/build_docs_test.py
index 0cbf8b478f..4d3bedda2d 100644
--- a/tensorflow/tools/docs/build_docs_test.py
+++ b/tensorflow/tools/docs/build_docs_test.py
@@ -30,9 +30,11 @@ from tensorflow.tools.docs import generate_lib
class Flags(object):
resource_root = resource_loader.get_root_dir_with_all_resources()
- src_dir = os.path.join(resource_root, 'tensorflow/docs_src')
+ src_dir = os.path.join(googletest.GetTempDir(), 'input')
+ os.mkdir(src_dir)
base_dir = os.path.join(resource_root, 'tensorflow/')
- output_dir = googletest.GetTempDir()
+ output_dir = os.path.join(googletest.GetTempDir(), 'output')
+ os.mkdir(output_dir)
class BuildDocsTest(googletest.TestCase):