Tests: Make lfwSubset assertions more informative.

cmusatyalab · Feb 21, 2016 · a1a644c · a1a644c
1 parent d895487
commit a1a644c
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 6 deletions.
diff --git a/tests/openface_batch_represent_tests.py b/tests/openface_batch_represent_tests.py
@@ -35,8 +35,7 @@
 
 
 def test_batch_represent():
-    # Get lfw-subset by running ./data/download-lfw-subset.sh
-    assert os.path.isdir(lfwSubset)
+    assert os.path.isdir(lfwSubset), "Get lfw-subset by running ./data/download-lfw-subset.sh"
 
     workDir = tempfile.mkdtemp(prefix='OpenFaceBatchRep-')
 

diff --git a/tests/openface_demo_tests.py b/tests/openface_demo_tests.py
@@ -52,8 +52,7 @@ def test_classification_demo_pretrained():
 
 
 def test_classification_demo_training():
-    # Get lfw-subset by running ./data/download-lfw-subset.sh
-    assert os.path.isdir(lfwSubset)
+    assert os.path.isdir(lfwSubset), "Get lfw-subset by running ./data/download-lfw-subset.sh"
 
     workDir = tempfile.mkdtemp(prefix='OpenFaceCls-')
 

diff --git a/tests/openface_neural_net_training_tests.py b/tests/openface_neural_net_training_tests.py
@@ -33,8 +33,7 @@
 
 
 def test_dnn_training():
-    # Get lfw-subset by running ./data/download-lfw-subset.sh
-    assert os.path.isdir(lfwSubset)
+    assert os.path.isdir(lfwSubset), "Get lfw-subset by running ./data/download-lfw-subset.sh"
 
     imgWorkDir = tempfile.mkdtemp(prefix='OpenFaceTrainingTest-Img-')
     cmd = ['python2', os.path.join(openfaceDir, 'util', 'align-dlib.py'),