add common.py in nightly_build

Signed-off-by: hwangdeyu <[email protected]>

Author: hwangdeyu

tests/keras2onnx_applications/nightly_build/common.py

@@ -0,0 +1,11 @@
+# SPDX-License-Identifier: Apache-2.0
+
+import urllib
+
+
+def check_bloburl_access(url):
+    try:
+        response = urllib.request.urlopen(url)
+        return response.getcode() == 200
+    except urllib.error.URLError:
+        return False

tests/keras2onnx_applications/nightly_build/test_transformers.py

@@ -12,6 +12,7 @@
 import numpy as np
 import tensorflow as tf
 from onnxconverter_common.onnx_ex import get_maximum_opset_supported
+from common import check_bloburl_access
 
 sys.path.insert(0, os.path.join(dirname(abspath(__file__)), '../../keras2onnx_tests/'))
 from test_utils import run_onnx_runtime
@@ -21,20 +22,12 @@
 if os.environ.get('ENABLE_FULL_TRANSFORMER_TEST', '0') != '0':
     enable_transformer_test = True
 
-def check_bloburl_access():
-    url = r'https://lotus.blob.core.windows.net/converter-models/transformer_tokenizer/'
-    try:
-        response = urllib.request.urlopen(url)
-        if response.getcode() != 200:
-            return False
-    except urllib.error.URLError:
-        return False
-    return True
+CONVERTER_TRANSFERMER_PATH = r'https://lotus.blob.core.windows.net/converter-models/transformer_tokenizer/'
 
 
 @unittest.skipIf(is_tensorflow_older_than('2.1.0'),
                  "Transformers conversion need tensorflow 2.1.0+")
-@unittest.skipIf(check_bloburl_access(), "Model blob url can't access")
+@unittest.skipIf(check_bloburl_access(CONVERTER_TRANSFERMER_PATH), "Model blob url can't access.")
 class TestTransformers(unittest.TestCase):
 
     text_str = 'The quick brown fox jumps over lazy dog.'

tests/keras2onnx_applications/nightly_build/test_yolov3.py

@@ -9,6 +9,7 @@
 from os.path import dirname, abspath
 sys.path.insert(0, os.path.join(dirname(abspath(__file__)), '../../keras2onnx_tests/'))
 from keras.models import load_model
+from common import check_bloburl_access
 
 import urllib.request
 YOLOV3_WEIGHTS_PATH = r'https://lotus.blob.core.windows.net/converter-models/yolov3.h5'
@@ -25,16 +26,6 @@
 tmp_path = os.path.join(working_path, 'temp')
 
 
-def check_bloburl_access(url):
-    try:
-        response = urllib.request.urlopen(url)
-        if response.getcode() != 200:
-            return False
-    except urllib.error.URLError:
-        return False
-    return True
-
-
 class TestYoloV3(unittest.TestCase):
 
     def setUp(self):
@@ -56,7 +47,7 @@ def post_compute(self, all_boxes, all_scores, indices):
     @unittest.skipIf(StrictVersion(onnx.__version__.split('-')[0]) < StrictVersion("1.5.0"),
                      "NonMaxSuppression op is not supported for onnx < 1.5.0.")
     @unittest.skipIf(check_bloburl_access(YOLOV3_WEIGHTS_PATH) or check_bloburl_access(YOLOV3_TINY_WEIGHTS_PATH),
-                     "Model blob url can't access")
+                     "Model blob url can't access.")
     def test_yolov3(self):
         img_path = os.path.join(os.path.dirname(__file__), '../data', 'street.jpg')
         yolo3_yolo3_dir = os.path.join(os.path.dirname(__file__), '../../../keras-yolo3/yolo3')