From 14c750ded721ba4e9ae123b6fd2ba8dae46db09a Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?xavier=20dupr=C3=A9?= Date: Tue, 15 Aug 2017 20:29:05 +0200 Subject: [PATCH] fix notebook unit test --- src/ensae_projects/automation/notebook_test_helper.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/src/ensae_projects/automation/notebook_test_helper.py b/src/ensae_projects/automation/notebook_test_helper.py index 8595749..b79d4b2 100644 --- a/src/ensae_projects/automation/notebook_test_helper.py +++ b/src/ensae_projects/automation/notebook_test_helper.py @@ -149,6 +149,9 @@ def _valid_cell(cell): kernel_name = None if "travis" in sys.executable else install_python_kernel_for_unittest( "ensae_projects") 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)