From 27ce6913c6c4a513efe0f5d02134cdcb5c5b31ac Mon Sep 17 00:00:00 2001
From: matze-dd <45763831+matze-dd@users.noreply.github.com>
Date: Fri, 6 Nov 2020 11:01:04 +0100
Subject: [PATCH] core: added tests for #98

---
 tests/test_packages/test_babel.py          | 45 ++++++++++++++++++++--
 tests/test_packages/test_latex_builtins.py |  4 +-
 2 files changed, 43 insertions(+), 6 deletions(-)

diff --git a/tests/test_packages/test_babel.py b/tests/test_packages/test_babel.py
index 8167714d..507cce54 100644
--- a/tests/test_packages/test_babel.py
+++ b/tests/test_packages/test_babel.py
@@ -6,22 +6,59 @@
 preamble = '\\usepackage{babel}\n'
 
 def get_plain(latex):
-    parms = parameters.Parameters()
+    parms = parameters.Parameters('en')
+    parms.multi_language = True
     p = parser.Parser(parms)
     plain, nums = utils.get_txt_pos(p.parse(preamble + latex))
     return plain
 
 
-data_test_macros_latex = [
+data_test_macros_python = [
 
     (r'A\foreignlanguage{russian}{T}B', 'ATB'),
     (r'A\foreignlanguage[opt]{russian}{T}B', 'ATB'),
+    (r"""
+A
+\foreignlanguage{german}{ }
+B
+""", r"""
+A
+B
+"""),
+    (r"""
+\usepackage{amsthm}
+\begin{proof}
+\foreignlanguage{german}{\begin{proof}}
+""", r"""
+
+
+Proof.
+
+Beweis.
+"""),
     (r'A\selectlanguage{russian}B', 'AB'),
+    (r"""
+A
+\selectlanguage{english}
+B
+""", r"""
+A
+B
+"""),
+    (r"""
+\usepackage{amsthm}
+\selectlanguage{german}
+\begin{proof}
+""", r"""
+
+
+Beweis.
+"""),
 
 ]
 
-@pytest.mark.parametrize('latex,plain_expected', data_test_macros_latex)
-def test_macros_latex(latex, plain_expected):
+@pytest.mark.parametrize('latex,plain_expected', data_test_macros_python)
+def test_macros_python(latex, plain_expected):
     plain = get_plain(latex)
     assert plain == plain_expected
 
diff --git a/tests/test_packages/test_latex_builtins.py b/tests/test_packages/test_latex_builtins.py
index 519c6908..450ef04d 100644
--- a/tests/test_packages/test_latex_builtins.py
+++ b/tests/test_packages/test_latex_builtins.py
@@ -72,8 +72,8 @@ def test_macros_latex(latex, plain_expected):
     (r'A\hspace*{1em}B', 'A B'),
     (r'A\newcommand{\xxx}{X}B', 'AB'),
     (r'A\newcommand*{\xxx}[1][x]{X}B', 'AB'),
-    (r'A\newtheorem{xxx}{XXX} B', 'AB'),
-    (r'A\newtheorem{xxx}[c]{XXX}[d] B', 'A B'),
+    (r'A\newtheorem{xxx}{XYZ} B', 'AB'),
+    (r'A\newtheorem{xxx}[c]{XYZ}[d] B', 'A B'),
     (r'\part[hi]{ho}', 'ho.'),
     (r'\part*{ho}', 'ho.'),
     (r'\part{ho?}', 'ho?'),