Skip to content
Snippets Groups Projects
Commit d4ac8672 authored by Martin Bauer's avatar Martin Bauer
Browse files

tests: suppress warnings from jupyter notebook runner

parent ab9e5e47
Branches function-call-printing
Tags
No related merge requests found
Pipeline #20268 passed
...@@ -3,6 +3,7 @@ import pytest ...@@ -3,6 +3,7 @@ import pytest
import tempfile import tempfile
import runpy import runpy
import sys import sys
import warnings
# Trigger config file reading / creation once - to avoid race conditions when multiple instances are creating it # Trigger config file reading / creation once - to avoid race conditions when multiple instances are creating it
# at the same time # at the same time
from pystencils.cpu import cpujit from pystencils.cpu import cpujit
...@@ -89,7 +90,7 @@ class IPyNbTest(pytest.Item): ...@@ -89,7 +90,7 @@ class IPyNbTest(pytest.Item):
# in notebooks there is an implicit plt.show() - if this is not called a warning is shown when the next # in notebooks there is an implicit plt.show() - if this is not called a warning is shown when the next
# plot is created. This warning is suppressed here # plot is created. This warning is suppressed here
exec("import warnings;" exec("import warnings;"
"warnings.filterwarnings('ignore', 'Adding an axes using the same arguments as a previous.*')", "warnings.filterwarnings('ignore', 'Adding an axes using the same arguments as a previous.*');",
global_dict) global_dict)
with tempfile.NamedTemporaryFile() as f: with tempfile.NamedTemporaryFile() as f:
f.write(self.code.encode()) f.write(self.code.encode())
...@@ -104,9 +105,12 @@ class IPyNbFile(pytest.File): ...@@ -104,9 +105,12 @@ class IPyNbFile(pytest.File):
exporter.exclude_input_prompt = True exporter.exclude_input_prompt = True
notebook_contents = self.fspath.open() notebook_contents = self.fspath.open()
notebook = nbformat.read(notebook_contents, 4)
code, _ = exporter.from_notebook_node(notebook) with warnings.catch_warnings():
yield IPyNbTest(self.name, self, code) warnings.filterwarnings("ignore", "IPython.core.inputsplitter is deprecated")
notebook = nbformat.read(notebook_contents, 4)
code, _ = exporter.from_notebook_node(notebook)
yield IPyNbTest(self.name, self, code)
def teardown(self): def teardown(self):
pass pass
......
...@@ -2,6 +2,10 @@ ...@@ -2,6 +2,10 @@
python_files = test_*.py *_test.py scenario_*.py python_files = test_*.py *_test.py scenario_*.py
norecursedirs = *.egg-info .git .cache .ipynb_checkpoints htmlcov norecursedirs = *.egg-info .git .cache .ipynb_checkpoints htmlcov
addopts = --doctest-modules --durations=20 --cov-config pytest.ini addopts = --doctest-modules --durations=20 --cov-config pytest.ini
markers =
longrun: tests only run at night since they have large execution time
notebook: jupyter notebooks
[run] [run]
branch = True branch = True
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment