chore: apply tests and preview options

Author: DamianKobylinski

app/evaluation_tests.py

@@ -32,7 +32,7 @@ class TestEvaluationFunction(unittest.TestCase):
     def test_connection(self):
         id_connection = initialize_test_connection()
 
-        response, answer, params = id_connection, 0.0, {"api_endpoint": "resistance/"}
+        response, answer, params = id_connection, id_connection, {"api_endpoint": "resistance/", "correct_answer": 0.0}
         result = evaluation_function(response, answer, params)
 
         self.assertEqual(result.get("is_correct"), False)

app/preview.py

@@ -7,12 +7,11 @@
 class Result(TypedDict):
     preview: Any
 
-
 def preview_function(response: Any, params: Any) -> Result:
     """
     Function used to preview a student response.
     ---
-    The handler function passes three arguments to preview_function():
+    The handler function passes two arguments to preview_function():
 
     - `response` which are the answers provided by the student.
     - `params` which are any extra parameters that may be useful,
@@ -21,21 +20,18 @@ def preview_function(response: Any, params: Any) -> Result:
     The output of this function is what is returned as the API response
     and therefore must be JSON-encodable. It must also conform to the
     response schema.
-
-    Any standard python library may be used, as well as any package
-    available on pip (provided it is added to requirements.txt).
-
-    The way you wish to structure you code (all in this function, or
-    split into many) is entirely up to you.
     """
     try:
         api_endpoint = params.get("api_endpoint", 'resistance/')
 
+        if len(response) != 6:
+            raise Exception("Connection ID must be 6 characters long")
+
         api_response = requests.get(f"{os.environ.get('API_CONNECTION')}/{api_endpoint}{response}")
         api_response.raise_for_status()
         api_data = api_response.json()
     except requests.RequestException as e:
         print(f"Error API connection: {e}")
         api_data = None
 
-    return Result(preview=api_data)
+    return Result(preview=api_data)

app/preview_tests.py

@@ -30,15 +30,15 @@ class TestPreviewFunction(unittest.TestCase):
 
     def test_api_endpoint_resistance(self):
         id_connection = initialize_test_connection()
-        response, params = id_connection, {"api_endpoint": "resistance/"}
+        response, params = id_connection, {"api_endpoint": "resistance/", "correct_answer": 0.0}
         result = preview_function(response, params)
 
         self.assertIn("preview", result)
         self.assertIsNotNone(result["preview"])
 
     def test_api_endpoint_resistors(self):
         id_connection = initialize_test_connection()
-        response, params = id_connection, {"api_endpoint": "resistors/"}
+        response, params = id_connection, {"api_endpoint": "resistors/", "correct_answer": 0.0}
         result = preview_function(response, params)
 
         self.assertIn("preview", result)