test: support mengxue

SUTLZY · Jan 14, 2020 · 0ec6f76 · 0ec6f76
2 parents d739bd7 + f50ea06
commit 0ec6f76
Showing 1 changed file with 7 additions and 5 deletions.
diff --git a/test_poetry.py b/test_poetry.py
@@ -1,5 +1,4 @@
-#! -*- coding: utf-8 -*-
-# import sqlite3
+# -*- coding: utf-8 -*-
 import os
 import json
 import sys
@@ -15,11 +14,11 @@ def check_json(f, _dir):
     with open(filepath) as file:
         try:
             _ = json.loads(file.read())
-            print(u"%s 校验成功" % _dir)
+            sys.stdout.write(f"{filepath} 校验成功")
             return True
         except:
-            sys.stderr.write(traceback.format_exc())            
-            assert False, u"校验(%s)失败" % f
+            sys.stderr.write(traceback.format_exc())
+            assert False, f"{filepath} 校验失败"
 
 
 def __check_path__(path):
@@ -44,3 +43,6 @@ def __check_path__(path):
 test_sishuwujing = functools.partial(__check_path__, u'./sishuwujing/')
 
 test_yuanqu = functools.partial(__check_path__, u'./yuanqu/')
+
+test_mengxue = functools.partial(__check_path__, u'./mengxue')
+