diff --git a/pytest.ini b/pytest.ini index 942d738..6f1ef58 100644 --- a/pytest.ini +++ b/pytest.ini @@ -1,3 +1,3 @@ [pytest] flake8-ignore = E501 -addopts = --flake8 --doctest-modules +addopts = --flake8 diff --git a/tests/test_functions.py b/tests/test_functions.py new file mode 100644 index 0000000..6a3c2b4 --- /dev/null +++ b/tests/test_functions.py @@ -0,0 +1,7 @@ +import nbstripout + + +def test_pos_recursive(): + d = {'a': {'b': 1, 'c': 2}} + assert nbstripout.pop_recursive(d, 'a.c') == 2 + assert d == {'a': {'b': 1}}