1 files changed, 25 insertions, 0 deletions

diff --git a/lib/oeqa/runtime/miutils/tests/squeezenet_model_download_test.py b/lib/oeqa/runtime/miutils/tests/squeezenet_model_download_test.py
new file mode 100644
index 00000000..a3e46a0a
--- /dev/null
+++ b/lib/oeqa/runtime/miutils/tests/squeezenet_model_download_test.py
@@ -0,0 +1,25 @@
+class SqueezenetModelDownloadTest(object):
+    download_files = {'squeezenet1.1.prototxt': 'https://raw.githubusercontent.com/DeepScale/SqueezeNet/a47b6f13d30985279789d08053d37013d67d131b/SqueezeNet_v1.1/deploy.prototxt',
+                      'squeezenet1.1.caffemodel': 'https://github.com/DeepScale/SqueezeNet/raw/a47b6f13d30985279789d08053d37013d67d131b/SqueezeNet_v1.1/squeezenet_v1.1.caffemodel'}
+
+    def __init__(self, target, work_dir):
+        self.target = target
+        self.work_dir = work_dir
+
+    def setup(self):
+        self.target.run('mkdir -p %s' % self.work_dir)
+
+    def tear_down(self):
+        self.target.run('rm -rf %s' % self.work_dir)
+
+    def test_can_download_squeezenet_model(self, proxy_port):
+        return self.target.run('cd %s; wget %s -e https_proxy=%s' %
+                               (self.work_dir,
+                                self.download_files['squeezenet1.1.caffemodel'],
+                                proxy_port))
+
+    def test_can_download_squeezenet_prototxt(self, proxy_port):
+        return self.target.run('cd %s; wget %s -e https_proxy=%s' %
+                               (self.work_dir,
+                                self.download_files['squeezenet1.1.prototxt'],
+                                proxy_port))