add check for warnings count

hammerlab · Jul 8, 2016 · 0220965 · 0220965
1 parent 3878c9e
commit 0220965
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/test/test_provenance.py b/test/test_provenance.py
@@ -112,11 +112,11 @@ def test_summarize_provenance():
 
         ## should see a warning when loading provenance
         ## & provenance value should be None
-        print(cohort.summarize_provenance_per_cache())
-        ok_(not cohort.summarize_provenance_per_cache()[cache_name])
-
-
-
+        # But, when check_provenance is off, we shouldn't get a warning.
+        with warnings.catch_warnings(record=True) as w:
+            warnings.simplefilter("always")
+            ok_(not cohort.summarize_provenance_per_cache()[cache_name])
+            ok_(len(w)>0)
 
     finally:
         if cohort is not None: