Return the np.array after execution

sansmoraxz · May 17, 2023 · f19cad8 · f19cad8
1 parent a639712
commit f19cad8
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/src/lambda_handler.py b/src/lambda_handler.py
@@ -27,3 +27,4 @@ def lambda_handler(event, context):
     with io.BytesIO(obj['Body'].read()) as file_bytes:
         arr = np.load(file_bytes)
         print("Array from S3: ", arr)
+        return arr
diff --git a/tests/test_handler.py b/tests/test_handler.py
@@ -91,8 +91,8 @@ def test_created_object_exists(self):
 
     def test_lambda_handler(self):
         """Test lambda_handler"""
-        lambda_handler(event, None)
-        assert True
+        arr = lambda_handler(event, None)
+        assert arr.shape == (int(ARR_SIZE),)
 
     def tearDown(self):
         """Tear down method