summaryrefslogtreecommitdiff
path: root/Qwen2.5-Eval/evaluation/latex2sympy/tests/linalg_test.py
diff options
context:
space:
mode:
authorYuren Hao <yurenh2@timan108.cs.illinois.edu>2025-09-04 22:16:22 -0500
committerYuren Hao <yurenh2@timan108.cs.illinois.edu>2025-09-04 22:16:22 -0500
commitfc6d57ffb8d5ddb5820fcc00b5491a585c259ebc (patch)
treee9841f93a353e2107225cfc721d1ce57c0e594dc /Qwen2.5-Eval/evaluation/latex2sympy/tests/linalg_test.py
Initial commit
Diffstat (limited to 'Qwen2.5-Eval/evaluation/latex2sympy/tests/linalg_test.py')
-rwxr-xr-xQwen2.5-Eval/evaluation/latex2sympy/tests/linalg_test.py15
1 files changed, 15 insertions, 0 deletions
diff --git a/Qwen2.5-Eval/evaluation/latex2sympy/tests/linalg_test.py b/Qwen2.5-Eval/evaluation/latex2sympy/tests/linalg_test.py
new file mode 100755
index 0000000..d6dea56
--- /dev/null
+++ b/Qwen2.5-Eval/evaluation/latex2sympy/tests/linalg_test.py
@@ -0,0 +1,15 @@
+from .context import assert_equal
+import pytest
+from sympy import MatMul, Matrix
+
+
+def test_linalg_placeholder():
+ assert_equal("\\begin{pmatrix}1&2\\\\3&4\\end{pmatrix}\\cdot\\variable{v}", MatMul(Matrix([[1, 2], [3, 4]]), Matrix([1, 2])), {'v': Matrix([1, 2])})
+
+
+def test_linalg_placeholder_multiple():
+ assert_equal("\\variable{M}\\cdot\\variable{v}", MatMul(Matrix([[1, 2], [3, 4]]), Matrix([1, 2])), {'M': Matrix([[1, 2], [3, 4]]), 'v': Matrix([1, 2])})
+
+
+def test_linalg_placeholder_multiple_mul():
+ assert_equal("\\begin{pmatrix}3&-1\\end{pmatrix}\\cdot\\variable{M}\\cdot\\variable{v}", MatMul(Matrix([[3, -1]]), Matrix([[1, 2], [3, 4]]), Matrix([1, 2])), {'M': Matrix([[1, 2], [3, 4]]), 'v': Matrix([1, 2])})