diff options
| author | T-Aji <tolujbd2@gmail.com> | 2024-08-23 09:33:32 +0100 |
|---|---|---|
| committer | T-Aji <tolujbd2@gmail.com> | 2024-08-23 09:33:32 +0100 |
| commit | f7dedf78465d27abec5f467d377ec67741b44fb3 (patch) | |
| tree | c4b3c8983f9c236b2505485ede3bc9154c22e91d /tests/test_transform_lambda.py | |
| parent | a8cadadfe2b96c84a29a252110822ec535a0da7e (diff) | |
| parent | f4bd9e3c85341c0805821728d42d74c19cb16bde (diff) | |
| download | de-project-bentley-f7dedf78465d27abec5f467d377ec67741b44fb3.tar.gz de-project-bentley-f7dedf78465d27abec5f467d377ec67741b44fb3.zip | |
Merge branch 'feature/transform-fact-sales-order' of https://github.com/ajschofield/de-project-bentley into feature/transform-fact-sales-order
Diffstat (limited to 'tests/test_transform_lambda.py')
| -rw-r--r-- | tests/test_transform_lambda.py | 10 |
1 files changed, 6 insertions, 4 deletions
diff --git a/tests/test_transform_lambda.py b/tests/test_transform_lambda.py index 5121905..516f83b 100644 --- a/tests/test_transform_lambda.py +++ b/tests/test_transform_lambda.py @@ -39,8 +39,8 @@ class TestReadFromS3: ) print(result) expected_df = pd.DataFrame( - np.array([["Vegetable", "Sour", "Green"], ["Berry", "Sweet", "Red"]]), - columns=["Food_type", "Flavour", "Colour"], + np.array([["Vegetable", "Sour", "Green", "2022-11-03 14:20:49.962"], ["Berry", "Sweet", "Red", "2022-11-03 14:20:49.962"]]), + columns=["Food_type", "Flavour", "Colour", "last_updated"], ) assert isinstance(result, dict) assert list(result.keys())[0] == "Foods" @@ -56,8 +56,8 @@ class TestReadFromS3: tables, bucket="dummy_buc", client=s3_client ) expected_foods_df = pd.DataFrame( - np.array([["Vegetable", "Sour", "Green"], ["Berry", "Sweet", "Red"]]), - columns=["Food_type", "Flavour", "Colour"], + np.array([["Vegetable", "Sour", "Green", "2022-11-03 14:20:49.962"], ["Berry", "Sweet", "Red", "2022-11-03 14:20:49.962"]]), + columns=["Food_type", "Flavour", "Colour", "last_updated"], ) expected_cars_df = pd.DataFrame( np.array( @@ -72,3 +72,5 @@ class TestReadFromS3: assert list(result.keys()) == tables assert result["Foods"].eq(expected_foods_df, axis="columns").all(axis=None) assert result["Cars"].eq(expected_cars_df, axis="columns").all(axis=None) + + |