test: new test for lambda_handler function

tests/test_lambda.py

@@ -0,0 +1,31 @@
+import os
+import sys
+import pytest
+import json
+
+current_dir = os.path.dirname(os.path.abspath(__file__))
+parent_dir = os.path.dirname(current_dir)
+sys.path.insert(0, os.path.dirname(parent_dir))
+
+from lambda_function import lambda_handler
+
+@pytest.fixture
+def event():
+    json_event = {"body": "{\"features\": [[6.5, 3.0, 5.8, 2.2], [6.1, 2.8, 4.7, 1.2]]}"}
+    return json_event
+
+@pytest.fixture
+def context():
+    return None
+
+@pytest.fixture
+def response_prediction():
+    return ["virginica", "versicolor"]
+
+
+def test_lambda_handler(event, context, response_prediction):
+    lambda_response = lambda_handler(event, context)
+
+    assert lambda_response["statusCode"] == 200
+    assert json.loads(lambda_response["body"])["predictions"] == response_prediction
+