Enhance EquationFixer to support bracket blocks with optional blank lines and indentation; add corresponding test case in test.py and bump version to 1.1.7

Author: dynstat

src/markdown_equations_fixer/__init__.py

@@ -1 +1 @@
-__version__ = "1.1.6"
+__version__ = "1.1.7"

src/markdown_equations_fixer/cli.py

@@ -34,20 +34,27 @@ def fix_equations(self, content: str) -> str:
         """Fix mathematical equations in markdown content."""
         try:
             # Pattern to match block equations delimited by standalone lines with [ and ]
-            # Example:
-            # [\n
-            # equation content\n
-            # ]
-            # This converts to:
-            # $$\n
-            # equation content\n
-            # $$
-            content = re.sub(
-                r"(?ms)^\[\s*$\n(.*?)\n^\]\s*$",
-                r"$$\n\1\n$$",
-                content,
+            # Allow optional blank/whitespace-only lines after [ and before ]
+            # and preserve inner content trimmed of leading/trailing blank lines.
+            bracket_block_pattern = re.compile(
+                r"(?ms)^[\t ]*\[[\t ]*\r?\n(.*?)(?:\r?\n)^[\t ]*\][\t ]*(?:\r?\n|$)",
             )
 
+            def _replace_bracket_block(match: re.Match) -> str:
+                inner = match.group(1)
+                # Trim leading/trailing blank lines in the captured block
+                lines = inner.splitlines()
+                start = 0
+                end = len(lines)
+                while start < end and lines[start].strip() == "":
+                    start += 1
+                while end > start and lines[end - 1].strip() == "":
+                    end -= 1
+                inner_trimmed = "\n".join(lines[start:end])
+                return f"$$\n{inner_trimmed}\n$$\n"
+
+            content = bracket_block_pattern.sub(_replace_bracket_block, content)
+
             # Pattern to match block equations: \[ ... \]
             # The re.DOTALL flag allows the match to span multiple lines.
             content = re.sub(

src/test.py

@@ -49,6 +49,23 @@ def test_fix_bracket_line_block(fixer):
     assert fixer.fix_equations(input_text) == expected
 
 
+def test_fix_bracket_block_with_blank_lines_and_indent(fixer):
+    """Blocks with extra blank lines and indentation should be converted."""
+    input_text = (
+        "\n[\n\n   Z₁ = XW₁ + b₁\n\n]\n\n"  # first block
+        "[\n   A₁ = ReLU(Z₁)\n]\n\n"      # second block
+        "[\n   Z₂ = A₁W₂ + b₂\n]\n\n"      # third block
+        "[\n   \\hat{y} = Z₂\n]\n\n"       # fourth block
+        "[\n   L = \\frac{1}{2m} \\sum (y - \\hat{y})^2\n]\n"  # fifth block
+    )
+    output = fixer.fix_equations(input_text)
+    # Ensure each becomes a $$ block and inner content preserved sans surrounding blanks
+    assert "$$\nZ₁ = XW₁ + b₁\n$$" in output
+    assert "$$\nA₁ = ReLU(Z₁)\n$$" in output
+    assert "$$\nZ₂ = A₁W₂ + b₂\n$$" in output
+    assert "$$\n\\hat{y} = Z₂\n$$" in output
+    assert "$$\nL = \\frac{1}{2m} \\sum (y - \\hat{y})^2\n$$" in output
+
 def test_validate_paths_recursive(temp_dir):
     """Test recursive path validation."""
     paths = [temp_dir]