more tests, fix print twice

Author: samuelcolvin

pytest_examples/eval_example.py

@@ -84,7 +84,6 @@ def run_print_check(
         Run the example and check print statements.
 
         :param example: The example to run.
-        :param line_length: The line length to use when wrapping print statements.
         :param rewrite_assertions: If True, rewrite assertions in the example using pytest's assertion rewriting.
         """
         __tracebackhide__ = True
@@ -101,7 +100,6 @@ def run_print_update(
         Run the example and update print statements, requires `--update-examples`.
 
         :param example: The example to run.
-        :param line_length: The line length to use when wrapping print statements.
         :param rewrite_assertions: If True, rewrite assertions in the example using pytest's assertion rewriting.
         """
         __tracebackhide__ = True

pytest_examples/insert_print.py

@@ -111,14 +111,18 @@ def _insert_print_statements(self, example: CodeExample) -> str:
 
         lines = example.source.splitlines()
 
+        old_line_no = -1
+
         for s in reversed(self.print_func.statements):
             line_no, col = find_print_location(example, s.line_no)
 
             # switch from 1-indexed line number to 0-indexed indexes into lines
             line_index = line_no - 1
 
-            remove_old_print(lines, line_index)
+            if s.line_no != old_line_no:
+                remove_old_print(lines, line_index)
             self._insert_print_args(lines, s, example.in_py_file(), line_index, col)
+            old_line_no = s.line_no
 
         return '\n'.join(lines) + '\n'
 
@@ -211,7 +215,18 @@ def find_print_location(example: CodeExample, line_no: int) -> tuple[int, int]:
 
 
 # ast nodes that have a body
-with_body = ast.Module, ast.FunctionDef, ast.If, ast.Try, ast.ExceptHandler, ast.With, ast.For, ast.AsyncFor, ast.While
+with_body = (
+    ast.Module,
+    ast.FunctionDef,
+    ast.If,
+    ast.Try,
+    ast.ExceptHandler,
+    ast.With,
+    ast.For,
+    ast.AsyncFor,
+    ast.While,
+    ast.ClassDef,
+)
 
 
 def find_print(node: Any, line: int) -> tuple[int, int] | None:

tests/cases_update/call_twice.md

@@ -0,0 +1,30 @@
+```py
+def foobar(x):
+    print(f'x={x}')
+
+foobar(1)
+foobar(2)
+```
+################### OUTPUT ###################
+```py
+def foobar(x):
+    print(f'x={x}')
+    #> x=1
+    #> x=2
+
+
+foobar(1)
+foobar(2)
+```
+################### TEST #####################
+```py
+from pytest_examples import find_examples, CodeExample, EvalExample
+import pytest
+
+
+@pytest.mark.parametrize('example', find_examples('.'), ids=str)
+def test_find_examples(example: CodeExample, eval_example: EvalExample):
+    if eval_example.update_examples:
+        eval_example.format(example)
+        eval_example.run_print_update(example)
+```

tests/cases_update/dataclass_indent.md

@@ -0,0 +1,43 @@
+```py
+from dataclasses import dataclass
+
+
+@dataclass
+class User:
+    foobar: int
+
+    def __post_init__(self):
+        print(self.foobar)
+
+
+User(foobar=123)
+```
+################### OUTPUT ###################
+```py
+from dataclasses import dataclass
+
+
+@dataclass
+class User:
+    foobar: int
+
+    def __post_init__(self):
+        print(self.foobar)
+        #> 123
+
+
+User(foobar=123)
+```
+################### TEST #####################
+```py
+from pytest_examples import find_examples, CodeExample, EvalExample
+import pytest
+
+
+@pytest.mark.parametrize('example', find_examples('.'), ids=str)
+def test_find_examples(example: CodeExample, eval_example: EvalExample):
+    eval_example.set_config(quotes='double')
+    if eval_example.update_examples:
+        eval_example.format(example)
+        eval_example.run_print_update(example)
+```

tests/test_insert_print.py

@@ -106,6 +106,20 @@ def foobar():
         (2, 0),
         id='call-attribute',
     ),
+    pytest.param(
+        """\
+@dataclass
+class User:
+    foobar: int
+
+    def __post_init__(self):
+        print(self.foobar)
+        #> 123
+""",
+        6,
+        (6, 8),
+        id='class-method',
+    ),
 ]