Merge pull request #98 from nschloe/fix-ws

Fix whitespace
nschloe · Sep 17, 2023 · d0ca7e2 · d0ca7e2
2 parents a8ae092 + f69bdcb
commit d0ca7e2
Show file tree

Hide file tree

Showing 3 changed files with 25 additions and 5 deletions.
diff --git a/src/pytest_codeblocks/__about__.py b/src/pytest_codeblocks/__about__.py
@@ -1 +1 @@
-__version__ = "0.16.2"
+__version__ = "0.17.0"
diff --git a/src/pytest_codeblocks/plugin.py b/src/pytest_codeblocks/plugin.py
@@ -108,8 +108,8 @@ def runtest(self):
             str1 = output
 
             if self.obj.expected_output_ignore_whitespace:
-                str0 = re.sub(r"^\s+", "", str0, flags=re.MULTILINE)
-                str1 = re.sub(r"^\s+", "", str1, flags=re.MULTILINE)
+                str0 = re.sub(r"\s+", "", str0)
+                str1 = re.sub(r"\s+", "", str1)
 
             if str0 != str1:
                 raise RuntimeError(

diff --git a/tests/test_expected_output.py b/tests/test_expected_output.py
@@ -47,12 +47,32 @@ def test_expected_output_ignore_whitespace(testdir):
 dolor sit amet
 <!--pytest-codeblocks:expected-output-ignore-whitespace-->
 ```
- 4
-     -2
+ 4 -2
 
   3
 ```
 """
     testdir.makefile(".md", string)
     result = testdir.runpytest("--codeblocks")
     result.assert_outcomes(passed=1)
+
+
+def test_expected_output_ignore_whitespace_fail(testdir):
+    string = """
+Lorem ipsum
+```python
+print(1 + 3)
+print(1 - 3)
+print(1 * 3)
+```
+dolor sit amet
+<!--pytest-codeblocks:expected-output-ignore-whitespace-->
+```
+ 4 -2
+
+  5
+```
+"""
+    testdir.makefile(".md", string)
+    result = testdir.runpytest("--codeblocks")
+    result.assert_outcomes(failed=1)