add doc test first example

PaddlePaddle · Mar 31, 2020 · 0eedece · 0eedece
1 parent c29d0df
commit 0eedece
Show file tree

Hide file tree

Showing 3 changed files with 14 additions and 2 deletions.
diff --git a/doc/BERT_10_MINS.md b/doc/BERT_10_MINS.md
@@ -95,7 +95,8 @@ sh pip_app.sh
 python bert_10.py
 sh server.sh &
 wget https://paddle-serving.bj.bcebos.com/bert_example/data-c.txt --no-check-certificate
-cat data-c.txt | python bert_client.py
+head -n 500 data-c.txt > data.txt
+cat data.txt | python bert_client.py
 if [[ $? -eq 0 ]]; then
     echo "test success"
 else

diff --git a/tools/doc_test.sh b/tools/doc_test.sh
@@ -0,0 +1,10 @@
+#!/usr/bin/env bash
+
+
+function main() {
+   cat Serving/doc/doc_test_list | xargs python Serving/tools/doc_tester_reader.py Serving/doc/ 
+
+}
+
+
+main $@
diff --git a/tools/doc_tester_reader.py b/tools/doc_tester_reader.py
@@ -58,7 +58,8 @@ def GetTestFile(lines, i, filename):
 def RunTest():
     folder = 'test'
     os.system('cd ' + folder + ' && sh start.sh')
+    os.system('cd .. && rm -rf ' + folder)     
 
 if __name__ == '__main__':
-    ReadMarkDown(sys.argv[1])
+    ReadMarkDown(os.path.join(sys.argv[1], sys.argv[2]))
     RunTest()