fix notebook unit test

sdpython · Aug 15, 2017 · e9de8e1 · e9de8e1
1 parent 5f718b8
commit e9de8e1
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/src/code_beatrix/automation/notebook_test_helper.py b/src/code_beatrix/automation/notebook_test_helper.py
@@ -142,5 +142,9 @@ def valid_cell(cell):
     kernel_name = None if "travis" in sys.executable else install_python_kernel_for_unittest(
         "code_beatrix")
     addpaths = get_additional_paths()
+    if filter:
+        notebooks = [_ for i, _ in enumerate(notebooks) if filter(i, _)]
+    if len(notebooks) == 0:
+        raise ValueError("Empty list of notebooks.")
     return execute_notebook_list(
         folder, notebooks, fLOG=fLOG, valid=valid_cell, additional_path=addpaths, kernel_name=kernel_name)