diff options
| author | lian-manonog <160282780+lian-manonog@users.noreply.github.com> | 2024-08-23 15:04:07 +0100 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2024-08-23 15:04:07 +0100 |
| commit | 5c236c6ee0d0b7e0612a51bf8eaa59322d192a6e (patch) | |
| tree | f77e625f8faca75f4d325021f5d20326b381516f /tests/test_transform_lambda.py | |
| parent | 032760a745353b0584bc635bd5c51aa928677fea (diff) | |
| parent | 59035d00133eed3f258f75e3a99ce57cae35989d (diff) | |
| download | de-project-bentley-5c236c6ee0d0b7e0612a51bf8eaa59322d192a6e.tar.gz de-project-bentley-5c236c6ee0d0b7e0612a51bf8eaa59322d192a6e.zip | |
Merge pull request #94 from ajschofield/feature/transform-fact-sales-order
PR: rebased the fact-sales branch with the datafrrames branch - now merging into development
Diffstat (limited to 'tests/test_transform_lambda.py')
| -rw-r--r-- | tests/test_transform_lambda.py | 18 |
1 files changed, 14 insertions, 4 deletions
diff --git a/tests/test_transform_lambda.py b/tests/test_transform_lambda.py index 4c689f7..37ca08f 100644 --- a/tests/test_transform_lambda.py +++ b/tests/test_transform_lambda.py @@ -40,8 +40,13 @@ 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" @@ -58,8 +63,13 @@ 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( |