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

[Tests] Skip if extension not installed
ethantang95 · Nov 14, 2016 · 7fbb9b4 · 7fbb9b4
2 parents 588d2a8 + 0e18837
commit 7fbb9b4
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'