fix test cov misses

aocd/__init__.py

@@ -15,6 +15,7 @@
 from . import utils
 from .exceptions import AocdError
 from .get import get_data
+from .get import get_puzzle
 from .get import get_day_and_year
 from .post import submit as _impartial_submit
 
@@ -48,7 +49,7 @@ def __getattr__(name: str) -> t.Any:
         return get_data(day=day, year=year)
     if name == "puzzle":
         day, year = get_day_and_year()
-        return get.get_puzzle(day=day, year=year)
+        return get_puzzle(day=day, year=year)
     if name == "submit":
         try:
             day, year = get_day_and_year()

tests/test_aocd.py

@@ -52,3 +52,14 @@ def test_data_in_interactive_mode(monkeypatch, mocker, freezer):
 def test_attribute_errors_have_context():
     with pytest.raises(AttributeError("module 'aocd' has no attribute 'nope'")):
         aocd.nope
+
+
+def test_import_puzzle(mocker):
+    fake_stack = [("~/2023/q21.py", 1, "<test>", "from aocd import data")]
+    mocker.patch("aocd.get.traceback.extract_stack", return_value=fake_stack)
+    mocker.patch("aocd.models.Puzzle.input_data", "test puzzle")
+    from aocd import puzzle
+
+    assert puzzle.year == 2023
+    assert puzzle.day == 21
+    assert puzzle.input_data == "test puzzle"