Merge pull request #1263 from lukeyeager/tests-without-plugins

[Tests] Skip if extension not installed
NVIDIA · Nov 14, 2016 · 7e4afa9 · 7e4afa9
2 parents ade0287 + 0812313
commit 7e4afa9
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/digits/dataset/generic/test_views.py b/digits/dataset/generic/test_views.py
@@ -4,6 +4,7 @@
 import json
 import os
 import tempfile
+import unittest
 
 from bs4 import BeautifulSoup
 import numpy as np
@@ -179,6 +180,8 @@ def create_variable_size_random_imageset(cls, **kwargs):
 
     @classmethod
     def setUpClass(cls, **kwargs):
+        if extensions.data.get_extension(cls.EXTENSION_ID) is None:
+            raise unittest.SkipTest('Extension "%s" is not installed' % cls.EXTENSION_ID)
         super(BaseViewsTestWithDataset, cls).setUpClass()
         cls.dataset_id = cls.create_dataset(json=True, **kwargs)
         assert cls.dataset_wait_completion(cls.dataset_id) == 'Done', 'create failed'