switched to max-base and added tests (#10)

* switched to max-base and added tests

* added sleep

* this model is very slow to start

Author: bdwyer2

.dockerignore

@@ -0,0 +1,8 @@
+Dockerfile
+README.*
+.idea/
+.git/
+.gitignore
+tests/
+.pytest_cache
+tests/

.gitignore

@@ -0,0 +1,3 @@
+.idea/
+.pytest_cache/
+__pycache__/

.travis.yml

@@ -0,0 +1,17 @@
+language: python
+python:
+- 3.6
+
+services:
+- docker
+
+install:
+- docker build -t max-inception-resnet-v2 .
+- docker run -it -d -p 5000:5000 max-inception-resnet-v2
+- sleep 30
+
+before_script:
+- pip install pytest requests
+
+script:
+- pytest tests/test.py

Dockerfile

@@ -1,25 +1,22 @@
-FROM continuumio/miniconda3
+FROM codait/max-base
 
 ARG model_bucket=http://max-assets.s3-api.us-geo.objectstorage.softlayer.net/keras
 ARG model_file=inception_resnet_v2.h5
 
 WORKDIR /workspace
-RUN mkdir assets
-RUN wget -nv ${model_bucket}/${model_file} --output-document=/workspace/assets/${model_file}
+
+RUN wget -nv --show-progress --progress=bar:force:noscroll ${model_bucket}/${model_file} --output-document=/workspace/assets/${model_file}
 
 # Python package versions
 ARG numpy_version=1.14.1
 ARG tf_version=1.5.0
 ARG keras_version=2.1.4
 
-RUN pip install --upgrade pip && \
-    pip install numpy==${numpy_version} && \
+RUN pip install numpy==${numpy_version} && \
     pip install tensorflow==${tf_version} && \
     pip install Pillow && \
     pip install h5py && \
-    pip install keras==${keras_version} && \
-    pip install flask-restplus && \
-    pip install ipython
+    pip install keras==${keras_version}
 
 COPY . /workspace
 

tests/test.py

@@ -0,0 +1,22 @@
+import pytest
+import requests
+
+
+def test_response():
+    model_endpoint = 'http://localhost:5000/model/predict'
+    file_path = 'assets/cat.jpg'
+
+    with open(file_path, 'rb') as file:
+        file_form = {'image': (file_path, file, 'image/jpeg')}
+        r = requests.post(url=model_endpoint, files=file_form)
+
+    assert r.status_code == 200
+    response = r.json()
+    assert response['status'] == 'ok'
+    assert response['predictions'][0]['label_id'] == 'n02123045'
+    assert response['predictions'][0]['label'] == 'tabby'
+    assert response['predictions'][0]['probability'] > 0.6
+
+
+if __name__ == '__main__':
+    pytest.main([__file__])