diff --git a/pytest_examples/run_code.py b/pytest_examples/run_code.py index 737cd28..e0af79e 100644 --- a/pytest_examples/run_code.py +++ b/pytest_examples/run_code.py @@ -212,9 +212,8 @@ def _insert_print_statements(self, example: CodeExample) -> str: def _insert_print_args( self, lines: list[str], statement: PrintStatement, in_python: bool, line_index: int, col: int ) -> None: - single_line = statement.sep.join(map(str, statement.args)) - if self.print_callback: - single_line = self.print_callback(single_line) + formatted_args = [self.print_callback(str(arg)) if self.print_callback else str(arg) for arg in statement.args] + single_line = statement.sep.join(formatted_args) indent_str = ' ' * col max_single_length = self.config.line_length - len(indent_str) if '\n' not in single_line and len(single_line) + len(comment_prefix) < max_single_length: