Skip to content

Commit

Permalink
Add tests for view extensions
Browse files Browse the repository at this point in the history
  • Loading branch information
gheinrich committed May 20, 2016
1 parent ddb8784 commit 3061efe
Showing 1 changed file with 13 additions and 1 deletion.
14 changes: 13 additions & 1 deletion digits/model/images/generic/test_views.py
Original file line number Diff line number Diff line change
Expand Up @@ -23,6 +23,7 @@
import PIL.Image
from urlparse import urlparse

from digits import extensions
from digits.config import config_value
import digits.dataset.images.generic.test_views
import digits.test_views
Expand Down Expand Up @@ -240,6 +241,11 @@ def test_page_model_new(self):
def test_nonexistent_model(self):
assert not self.model_exists('foo'), "model shouldn't exist"

def test_view_config(self):
extension = extensions.view.get_default_extension()
rv = self.app.get('/models/view-config/%s' % extension.get_id())
assert rv.status_code == 200, 'page load failed with %s' % rv.status_code

def test_visualize_network(self):
rv = self.app.post('/models/visualize-network?framework='+self.FRAMEWORK,
data = {'custom_network': self.network()}
Expand Down Expand Up @@ -595,9 +601,15 @@ def test_infer_many_from_folder(self):
# StringIO wrapping is needed to simulate POST file upload.
file_upload = (StringIO(textfile_images), 'images.txt')

# try selecting the extension explicitly
extension = extensions.view.get_default_extension()
extension_id = extension.get_id()

rv = self.app.post(
'/models/images/generic/infer_many?job_id=%s' % self.model_id,
data = {'image_list': file_upload, 'image_folder': os.path.dirname(self.test_image)}
data = {'image_list': file_upload,
'image_folder': os.path.dirname(self.test_image),
'view_extension_id': extension_id}
)
s = BeautifulSoup(rv.data, 'html.parser')
body = s.select('body')
Expand Down

0 comments on commit 3061efe

Please sign in to comment.