pdit 0.3.2__tar.gz → 0.4.0__tar.gz

{pdit-0.3.2/pdit.egg-info → pdit-0.4.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: pdit
-Version: 0.3.2
+Version: 0.4.0
 Summary: Interactive Python code editor with inline execution results
 Author: Harry Vangberg
 License: MIT

{pdit-0.3.2 → pdit-0.4.0}/pdit/ipython_executor.py

@@ -167,6 +167,9 @@ del _register_pdit_runtime_hooks
 
         for node in tree.body:
             line_start = node.lineno
+            if isinstance(node, (ast.FunctionDef, ast.AsyncFunctionDef, ast.ClassDef)):
+                if node.decorator_list:
+                    line_start = min(dec.lineno for dec in node.decorator_list)
             line_end = node.end_lineno or node.lineno
 
             # Extract source

{pdit-0.3.2 → pdit-0.4.0/pdit.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: pdit
-Version: 0.3.2
+Version: 0.4.0
 Summary: Interactive Python code editor with inline execution results
 Author: Harry Vangberg
 License: MIT

{pdit-0.3.2 → pdit-0.4.0}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "pdit"
-version = "0.3.2"
+version = "0.4.0"
 description = "Interactive Python code editor with inline execution results"
 readme = "README.md"
 requires-python = ">=3.9,<3.14"

{pdit-0.3.2 → pdit-0.4.0}/tests/test_ipython_executor.py

@@ -448,6 +448,26 @@ result"""
         assert results[2]["isInvisible"] is True  # assignment
         assert "7" in results[3]["output"][0]["content"]
 
+    async def test_function_decorator_applied(self, executor):
+        """Test that function decorators are applied."""
+        script = """from functools import wraps
+
+def add_one(fn):
+    @wraps(fn)
+    def wrapper(*args, **kwargs):
+        return fn(*args, **kwargs) + 1
+    return wrapper
+
+@add_one
+def add(a, b):
+    return a + b
+
+add(2, 3)"""
+        results = await collect_results(executor.execute_script(script))
+
+        final_result = results[-1]
+        assert "6" in final_result["output"][0]["content"]
+
     async def test_list_comprehension(self, executor):
         """Test list comprehension execution."""
         script = "[x**2 for x in range(5)]"