1 files changed, 52 insertions, 0 deletions
diff --git a/Qwen2.5-Eval/evaluation/latex2sympy/tests/grouping_test.py b/Qwen2.5-Eval/evaluation/latex2sympy/tests/grouping_test.py
new file mode 100755
index 0000000..139f3f1
--- /dev/null
+++ b/Qwen2.5-Eval/evaluation/latex2sympy/tests/grouping_test.py
@@ -0,0 +1,52 @@
+from .context import assert_equal, _Pow, _Add, _Mul
+import pytest
+from sympy import Integral, sin, Symbol, Mul, Integer, Pow
+from latex2sympy.latex2sympy2 import latex2sympy as process_sympy
+
+a = Symbol('a', real=True)
+b = Symbol('b', real=True)
+x = Symbol('x', real=True)
+theta = Symbol('theta', real=True)
+
+
+func_arg_examples = [
+    ('\\int ', 'x dx', Integral(x, x)),
+    ('\\sin', '\\theta ', sin(theta))
+]
+
+example_groups = [
+    ('1+2', '3-4', _Mul(_Add(1, 2), _Add(3, _Mul(-1, 4))))
+]
+
+modifiable_delimiter_pairs = {
+    '(': ')',
+    '\\lgroup': '\\rgroup',
+    '\\{': '\\}',
+    '\\lbrace': '\\rbrace',
+    '[': ']',
+    '\\lbrack': '\\rbrack',
+}
+
+
+@pytest.mark.parametrize('func, args, output', func_arg_examples)
+def test_func_arg_groupings(func, args, output):
+    # none
+    assert_equal("{func} {args}".format(func=func, args=args), output)
+    # normal brace (not modifiable)
+    assert_equal("{func}{{{args}}}".format(func=func, args=args), output)
+    # rest of delimiters, with modifications
+    for left, right in modifiable_delimiter_pairs.items():
+        assert_equal("{func}{left}{args}{right}".format(left=left, right=right, func=func, args=args), output)
+        assert_equal("{func}\\left{left}{args}\\right{right}".format(left=left, right=right, func=func, args=args), output)
+        assert_equal("{func}\\mleft{left}{args}\\mright{right}".format(left=left, right=right, func=func, args=args), output)
+
+
+@pytest.mark.parametrize('group1, group2, output', example_groups)
+def test_delimiter_groupings(group1, group2, output):
+    # normal brace (not modifiable)
+    assert_equal("{{{group1}}}{{{group2}}}".format(group1=group1, group2=group2), output)
+    # rest of delimiters, with modifications
+    for left, right in modifiable_delimiter_pairs.items():
+        assert_equal("{left}{group1}{right}{left}{group2}{right}".format(left=left, right=right, group1=group1, group2=group2), output)
+        assert_equal("\\left{left}{group1}\\right{right}\\left{left}{group2}\\right{right}".format(left=left, right=right, group1=group1, group2=group2), output)
+        assert_equal("\\mleft{left}{group1}\\mright{right}\\mleft{left}{group2}\\mright{right}".format(left=left, right=right, group1=group1, group2=group2), output)