diff options
Diffstat (limited to 'tests')
| -rw-r--r-- | tests/test_fact_sales_order.py | 7 |
1 files changed, 3 insertions, 4 deletions
diff --git a/tests/test_fact_sales_order.py b/tests/test_fact_sales_order.py index 7592f68..48426b4 100644 --- a/tests/test_fact_sales_order.py +++ b/tests/test_fact_sales_order.py @@ -6,6 +6,7 @@ from src.fact_sales_order import ( create_dim_currency, ) + class TestCreateDimDesign: def test_dim_design_returns_dataframe(self): d = { @@ -56,8 +57,7 @@ class TestCreateDimStaff: "email_address": ["Hello", "Bye"], "department_id": ["Hello", "Bye"], } - test_df = {"staff": pd.DataFrame( - data=d), "department": pd.DataFrame(data=d2)} + test_df = {"staff": pd.DataFrame(data=d), "department": pd.DataFrame(data=d2)} result = create_dim_staff(test_df) assert isinstance(result, pd.DataFrame) @@ -74,8 +74,7 @@ class TestCreateDimStaff: "email_address": ["Hello", "Bye"], "department_id": ["Hello", "Bye"], } - test_df = {"staff": pd.DataFrame( - data=d), "department": pd.DataFrame(data=d2)} + test_df = {"staff": pd.DataFrame(data=d), "department": pd.DataFrame(data=d2)} result = create_dim_staff(test_df) expected_d = { "staff_id": ["Hello", "Bye"], |