add tests and optional dep for jupyter users

Author: wimglenn

aocd/version.py

@@ -1 +1 @@
-__version__ = "1.2.4"
+__version__ = "1.3.0"

setup.py

@@ -45,4 +45,5 @@
         "tzlocal",
     ],
     options={"bdist_wheel": {"universal": "1"}},
+    extras_require={"nb": ["IPython", "jupyter-server"]}
 )

tests/test_date_introspection.py

@@ -36,3 +36,28 @@ def test_day_is_invalid(mocker):
     mocker.patch("aocd.get.traceback.extract_stack", return_value=fake_stack)
     with pytest.raises(AocdError("Failed introspection of day")):
         get_day_and_year()
+
+
+def test_ipynb_ok(mocker):
+    fake_stack = [("ipykernel/123456789.py", 1, "<test>", "from aocd import data")]
+    mocker.patch("aocd.get.traceback.extract_stack", return_value=fake_stack)
+    mocker.patch("aocd.get.get_ipynb_path", return_value="puzzle-2020-03.py")
+    day, year = get_day_and_year()
+    assert day == 3
+    assert year == 2020
+
+
+def test_ipynb_fail(mocker):
+    fake_stack = [("ipykernel/123456789.py", 1, "<test>", "from aocd import data")]
+    mocker.patch("aocd.get.traceback.extract_stack", return_value=fake_stack)
+    mocker.patch("aocd.get.get_ipynb_path", side_effect=ImportError)
+    with pytest.raises(AocdError("Failed introspection of filename")):
+        get_day_and_year()
+
+
+def test_ipynb_fail_no_numbers_in_ipynb_filename(mocker):
+    fake_stack = [("ipykernel/123456789.py", 1, "<test>", "from aocd import data")]
+    mocker.patch("aocd.get.traceback.extract_stack", return_value=fake_stack)
+    mocker.patch("aocd.get.get_ipynb_path", "puzzle.py")
+    with pytest.raises(AocdError("Failed introspection of filename")):
+        get_day_and_year()