Add new test cases for advanced equations in conftest.py

Author: FridrichMethod

tests/conftest.py

@@ -64,6 +64,29 @@ class RawCase:
             ("v = 5.0 +- 0.1", "v"),
         ],
     ),
+    RawCase(
+        name="trig_exp",
+        equation=["I", "I0 * exp(-t/tau) * sin(omega*t + phi)"],
+        variables=[
+            ("I0 = 2.0 +- 0.05", r"I_0"),
+            ("t = 0.25 +- 0.01", "t"),
+            ("tau = 1.2 +- 0.02", r"\tau"),
+            ("omega = 3.14 +- 0.01", r"\omega"),
+            ("phi = 0.1 +- 0.02", r"\phi"),
+        ],
+    ),
+    RawCase(
+        name="rational_mixed",
+        equation=["Z", "(a*b + c)/(d - e*f)"],
+        variables=[
+            ("a = 1.5 +- 0.01", "a"),
+            ("b = 2.5 +- 0.02", "b"),
+            ("c = 0.75 +- 0.005", "c"),
+            ("d = 10 +- 0.1", "d"),
+            ("e = 0.5 +- 0.01", "e"),
+            ("f = 4 +- 0.2", "f"),
+        ],
+    ),
 ]
 
 

tests/test_calculator.py

@@ -2,8 +2,6 @@
 
 from __future__ import annotations
 
-import pytest
-
 from tests.conftest import RawCase
 from tests.legacy_calculator import run_legacy_calculator
 from uncertainty_calculator import Digits, Equation, UncertaintyCalculator, Variable
@@ -20,7 +18,6 @@ def test_calculator_output_matches_legacy(
     include_equation_number,
 ):
     """Refactored calculator should match the legacy implementation."""
-
     expected_output = run_legacy_calculator(
         equation=raw_case.equation,
         variables=raw_case.variables,
@@ -47,7 +44,6 @@ def test_calculator_output_matches_legacy(
 
 def test_run_can_be_called_multiple_times_with_new_inputs():
     """Calculator should not leak state between runs when inputs change."""
-
     digits = Digits(mu=2, sigma=2)
     calc = UncertaintyCalculator(
         equation=Equation(lhs="y", rhs="x"),
@@ -72,7 +68,6 @@ def test_run_can_be_called_multiple_times_with_new_inputs():
 
 def test_variable_dataclass_input():
     """Calculator should accept dataclass inputs and return a string output."""
-
     equation_obj = Equation(lhs=r"\zeta", rhs=r"K*x")
     variables_obj = [
         Variable(name="K", value="2", uncertainty="0.1", latex_name="K"),
@@ -95,7 +90,6 @@ def test_variable_dataclass_input():
 
 def test_mixed_input_types():
     """Numeric values provided as float/int should be accepted."""
-
     equation = Equation(lhs="y", rhs="x")
     variables = [Variable(name="x", value=10.5, uncertainty=0.5, latex_name="x")]
     digits = Digits(mu=2, sigma=2)
@@ -111,4 +105,3 @@ def test_mixed_input_types():
     )
 
     calc.run()
-

tests/test_compute.py

@@ -18,7 +18,6 @@ def _simple_parse_state(value: str, uncertainty: str = "0.1"):
 
 def test_compute_zero_uncertainty_sets_zero_derivative_and_sigma():
     """Zero uncertainty should produce zero derivatives and zero sigma."""
-
     parse_state = _simple_parse_state(value="2", uncertainty="0")
     compute_state = compute(parse_state, digits=Digits(mu=2, sigma=2))
 
@@ -27,4 +26,3 @@ def test_compute_zero_uncertainty_sets_zero_derivative_and_sigma():
     assert pdv_expr == S.Zero
     assert pdv_num == S.Zero
     assert compute_state.result_sigma == "0"
-

tests/test_formatting.py

@@ -10,7 +10,5 @@
 
 def test_latex_number_basic_formatting():
     """latex_number should render rational and integer inputs."""
-
     assert latex_number(sympify("3/2")) == "\\frac{3}{2}"
     assert latex_number(2) == "2"
-

tests/test_parsing.py

@@ -11,7 +11,6 @@
 
 def test_parse_equation_trims_and_builds_dataclass():
     """Whitespace should be stripped when building Equation."""
-
     raw = ["  A  ", "  B + C  "]
     equation = parse_equation(raw)
     assert equation.lhs == "A"
@@ -20,14 +19,12 @@ def test_parse_equation_trims_and_builds_dataclass():
 
 def test_parse_equation_invalid_length_raises():
     """Invalid equation length should raise ValueError."""
-
     with pytest.raises(ValueError):
         parse_equation(["only-one-side"])
 
 
 def test_parse_variables_parses_value_and_uncertainty():
     """Legacy tuple definitions should convert to Variable dataclasses."""
-
     raw = [
         ("X = 1 +- 0.1", "X"),
         ("Y = 2 +- 0.2", "Y"),
@@ -41,19 +38,16 @@ def test_parse_variables_parses_value_and_uncertainty():
 
 def test_parse_variables_invalid_definition():
     """Bad legacy variable definitions should be rejected."""
-
     with pytest.raises(ValueError):
         parse_variables([("invalid", "x")])
 
 
 def test_parsing_rejects_duplicate_variable_names():
     """Duplicate variable names should raise a ValueError."""
-
     equation = Equation(lhs="y", rhs="x + z")
     variables = [
         Variable(name="x", value=1, uncertainty=0.1, latex_name="x"),
         Variable(name="x", value=2, uncertainty=0.2, latex_name="x_dup"),
     ]
     with pytest.raises(ValueError, match="Duplicate variable name detected"):
         parse_inputs(equation, variables)
-

tests/test_rendering.py

@@ -17,10 +17,11 @@ def _simple_parse_state(value: str, uncertainty: str = "0.1"):
 
 def test_render_output_combined_contains_sigma_line():
     """Combined rendering should include sigma line and equation alignment."""
-
     parse_state = _simple_parse_state(value="2", uncertainty="0.5")
     compute_state = compute(parse_state, digits=Digits(mu=2, sigma=2))
-    options = RenderOptions(last_unit=None, separate=False, insert=False, include_equation_number=False)
+    options = RenderOptions(
+        last_unit=None, separate=False, insert=False, include_equation_number=False
+    )
 
     output = render_output(parse_state, compute_state, options)
     assert "\\begin{aligned}" in output
@@ -30,7 +31,6 @@ def test_render_output_combined_contains_sigma_line():
 
 def test_zero_uncertainty_renders_sigma_without_empty_sqrt():
     """Sigma rendering should short-circuit when uncertainties are zero."""
-
     digits = Digits(mu=2, sigma=2)
     calc = UncertaintyCalculator(
         equation=Equation(lhs="y", rhs="x"),
@@ -45,4 +45,3 @@ def test_zero_uncertainty_renders_sigma_without_empty_sqrt():
     output = calc.run()
     assert "\\sigma_{y}&=0" in output
     assert "sqrt{" not in output
-

tests/test_types.py

@@ -9,17 +9,14 @@
 
 def test_digits_dataclass_behaves_like_config():
     """Digits should store configured precision values."""
-
     digits = Digits(mu=4, sigma=5)
     assert digits.mu == 4
     assert digits.sigma == 5
 
 
 def test_digits_requires_positive_integers():
     """Digits must enforce positive integer validation."""
-
     with pytest.raises(ValueError, match="mu must be a positive integer"):
         Digits(mu=0, sigma=2)
     with pytest.raises(ValueError, match="sigma must be a positive integer"):
         Digits(mu=2, sigma=-1)
-

tests/test_validation.py

@@ -9,7 +9,6 @@
 
 def test_missing_variable_validation():
     """Validation should fail when equation references undefined symbols."""
-
     equation = Equation(lhs="y", rhs="m*x + b")  # 'b' is missing
     variables = [
         Variable(name="m", value=1, uncertainty=0.1, latex_name="m"),
@@ -29,4 +28,3 @@ def test_missing_variable_validation():
 
     with pytest.raises(ValueError, match="Symbol 'b' used in equation but not defined"):
         calc.run()
-