diff options
| -rw-r--r-- | tests/test_load_lambda.py | 9 |
1 files changed, 8 insertions, 1 deletions
diff --git a/tests/test_load_lambda.py b/tests/test_load_lambda.py index 830bb00..3ad7158 100644 --- a/tests/test_load_lambda.py +++ b/tests/test_load_lambda.py @@ -4,7 +4,14 @@ import boto3 import botocore.exceptions import os import pytest -from src.load_lambda import * +from src.load_lambda import ( + lambda_handler, + retrieve_secrets, + connect_to_db_and_return_engine, + convert_parquet_files_to_dfs, + get_transform_bucket, + upload_dfs_to_database, +) import tempfile |