From 468686a92fb8bb2dbabc514f57b2769af26d3228 Mon Sep 17 00:00:00 2001
From: Arthur Meyre <arthur.meyre@zama.ai>
Date: Thu, 2 Sep 2021 15:28:34 +0200
Subject: [PATCH] tools: add pytest_nb target to easily test notebooks

---
 .github/workflows/continuous-integration.yaml | 2 +-
 Makefile                                      | 4 ++++
 2 files changed, 5 insertions(+), 1 deletion(-)

diff --git a/.github/workflows/continuous-integration.yaml b/.github/workflows/continuous-integration.yaml
index b2e21364e..fe97021cd 100644
--- a/.github/workflows/continuous-integration.yaml
+++ b/.github/workflows/continuous-integration.yaml
@@ -82,7 +82,7 @@ jobs:
         run: |
           make strip_nb
           make notebook_timeout
-          poetry run pytest --nbmake examples/*.ipynb
+          make pytest_nb
       - name: Test coverage
         id: coverage
         if: ${{ steps.pytest.outcome != 'skipped' && !cancelled() }}
diff --git a/Makefile b/Makefile
index d68796f11..7f300a877 100644
--- a/Makefile
+++ b/Makefile
@@ -161,6 +161,10 @@ notebook_timeout:
 	poetry run python ./script/nbmake_utils/notebook_test_timeout.py examples
 .PHONY: notebook_timeout
 
+pytest_nb:
+	poetry run pytest --nbmake examples/*.ipynb
+.PHONY: pytest_nb
+
 benchmark:
 	poetry run pytest benchmarks/ --benchmark-save=findings
 .PHONY: benchmark