allow multiple expressions for homework

Author: beachdweller

tests/test_ai_tutor.py

@@ -3,7 +3,7 @@
 import pathlib
 import sys
 
-from typing import Dict, Union, List
+from typing import Callable, Dict, List, Tuple, Union
 
 import pytest
 
@@ -63,18 +63,23 @@ def explanation_in(request) -> str:
 
 
 @pytest.fixture
-def homework(explanation_in:str) -> str:
+def homework(explanation_in:str) -> Tuple[str]:
     d = {
-        'Korean': '숙제',
-        'English': 'Homework',
-        'Japanese': '宿題',
-        'Chinese': '作业',
-        'Spanish': 'Tarea',
-        'French': 'Devoir',
-        'German': 'Hausaufgabe',
-        'Thai': 'การบ้าน',
+        'Korean': ('숙제',),
+        'English': ('Homework',),
+        'Japanese': ('宿題',),
+        'Chinese': ('作业',),
+        'Spanish': ('Tarea',),
+        'French': ('Devoir',),
+        'German': ('Hausaufgabe',),
+        'Thai': ('การบ้าน',),
     }    
-    return d[explanation_in].lower()
+    return tuple(
+        map(
+            lambda x: x.lower(),
+            d[explanation_in]
+        )
+    )
 
 
 @pytest.fixture
@@ -92,14 +97,19 @@ def msg(explanation_in:str) -> str:
     return d[explanation_in].lower()
 
 
-def test_get_instruction(explanation_in, homework,):
+def test_get_instruction(explanation_in:str, homework:Tuple[str],):
     result = ai_tutor.get_directive(explanation_in=explanation_in)
 
-    assert homework in result.lower()
+    assert any(
+        map(
+            lambda x: x in result.lower(),
+            homework
+        )
+    )
 
 
 @pytest.mark.parametrize("func", (ai_tutor.get_report_header, ai_tutor.get_report_footer))
-def test_get_report__header__footer(explanation_in, msg, func):
+def test_get_report__header__footer(explanation_in:str, msg:str, func:Callable):
 
     result = func(explanation_in=explanation_in)
 
@@ -135,7 +145,7 @@ def test_get_code_instruction(
         sample_student_code_path:pathlib.Path,
         sample_readme_path:pathlib.Path,
         explanation_in:str,
-        homework:str,
+        homework:Tuple[str],
         instruction:str,
     ):
 
@@ -145,7 +155,12 @@ def test_get_code_instruction(
         explanation_in=explanation_in
     ).lower()
 
-    assert homework in result
+    assert any(
+        map(
+            lambda x: x in result,
+            homework
+        )
+    )
     assert instruction in result
 
 
@@ -155,7 +170,8 @@ def test_get_the_question(
         sample_student_code_path:pathlib.Path,
         sample_readme_path:pathlib.Path,
         explanation_in:str,
-        homework:str, msg:str,
+        homework:Tuple[str],
+        msg:str,
         instruction:str,
     ):
     result = ai_tutor.get_the_question(
@@ -165,7 +181,12 @@ def test_get_the_question(
         explanation_in=explanation_in,
     ).lower()
 
-    assert homework in result
+    assert any(
+        map(
+            lambda x: x in result,
+            homework
+        )
+    )
     assert msg in result
     assert instruction in result