demathpy 0.0.1__tar.gz → 0.0.2__tar.gz

{demathpy-0.0.1 → demathpy-0.0.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: demathpy
-Version: 0.0.1
+Version: 0.0.2
 Summary: PDE/ODE math backend
 Author: Misekai
 Author-email: Misekai <mcore-us@misekai.net>

{demathpy-0.0.1 → demathpy-0.0.2}/pyproject.toml

@@ -1,7 +1,7 @@
 
 [project]
 name = "demathpy"
-version = "0.0.1"
+version = "0.0.2"
 authors = [
   { name="Misekai", email="mcore-us@misekai.net" },
 ]
@@ -22,3 +22,8 @@ license-files = ["LICEN[CS]E*"]
 requires = ["uv_build >= 0.9.26, <0.10.0"]
 build-backend = "uv_build"
 
+[dependency-groups]
+dev = [
+    "pytest>=9.0.2",
+]
+

{demathpy-0.0.1 → demathpy-0.0.2}/src/demathpy/symbols.py

@@ -287,25 +287,40 @@ def normalize_symbols(expr: str) -> str:
         "omicron|pi|rho|sigma|tau|upsilon|phi|chi|psi|omega"
     )
     fn_words = "sin|cos|tan|sinh|cosh|tanh|exp|log|log10|log2|sqrt|abs|sech|sign"
+    # Use word boundary \b to avoid matching inside words
     expr = re.sub(rf"\b({greek_words})(?=({fn_words})\b)", r"\1*", expr)
-    # Greek word directly followed by a single-letter variable (e.g., beta u)
-        # Greek word directly followed by a single-letter variable (e.g., beta u) or concatenated (betau).
+    # Greek word directly followed by a single-letter variable (e.g., beta u) or concatenated (betau).
     # BUT do not split axis-suffixed coefficients like alphax/alphaz (common in anisotropic diffusion terms).
     _axis_coeffs = {
         "alphax", "alphaz", "betax", "betaz", "gammax", "gammaz", "deltax", "deltaz",
         "sigmax", "sigmaz", "thetax", "thetaz", "kappax", "kappaz", "lambdax", "lambdaz",
         "rhox", "rhoz", "mux", "muz", "nux", "nuz",
+        # Also protect eps/epsilon from being split
+        "epsilon", "eps",
     }
 
     def _split_greek_letter_var(m: re.Match) -> str:
         greek = m.group(1)
         letter = m.group(2)
         combined = f"{greek}{letter}"
+        # Don't split if this creates a known coefficient (e.g., alphax, sigmaz)
         if combined in _axis_coeffs:
             return combined
+        # Don't split if combined is the start of a greek word (to avoid "epsi*lon" from "epsilon")
+        # Check if any greek word starts with the combined string
+        all_greek_words = {"alpha", "beta", "gamma", "delta", "epsilon", "zeta", "eta", "theta", 
+                          "iota", "kappa", "lam", "mu", "nu", "xi", "omicron", "pi", "rho", 
+                          "sigma", "tau", "upsilon", "phi", "chi", "psi", "omega"}
+        for gw in all_greek_words:
+            if gw.startswith(combined) and len(gw) > len(combined):
+                # combined is a prefix of a longer greek word, so don't split
+                return combined
+        # Otherwise, insert multiplication: betau -> beta*u
         return f"{greek}*{letter}"
 
-    expr = re.sub(rf"({greek_words})([A-Za-z])", _split_greek_letter_var, expr)
+    # Only match at word boundaries to avoid splitting within words like 'epsilon'
+    # Use lookahead to detect when a greek word is followed by a single letter that starts a new token
+    expr = re.sub(rf"\b({greek_words})([A-Za-z])(?=([^a-zA-Z]|$))", _split_greek_letter_var, expr)
 
     # Absolute value for variables/parentheses
     expr = re.sub(r"\|\s*([a-zA-Z_][a-zA-Z0-9_]*)\s*\|", r"abs(\1)", expr)