passing tests create_dim_design and create_dim_staff

1 files changed, 40 insertions, 0 deletions

diff --git a/tests/test_fact_sales_order.py b/tests/test_fact_sales_order.py
new file mode 100644
index 0000000..13196d5
--- /dev/null
+++ b/tests/test_fact_sales_order.py
@@ -0,0 +1,40 @@
+from src.fact_sales_order import create_dim_design, create_dim_staff
+import pandas as pd
+
+class TestCreateDimDesign:
+    def test_dim_design_returns_dataframe(self):
+        d = {"test": ["Hello", "Bye"], "design_id": ["Hello", "Bye"], "design_name": ["Hello", "Bye"], 
+                "file_name": ["Hello", "Bye"], "file_location": ["Hello", "Bye"], "Hello": ["Hello", "Bye"]}
+        test_df = {"design": pd.DataFrame(data=d)}
+        result = create_dim_design(test_df)
+        assert isinstance(result, pd.DataFrame)
+
+    def test_dim_design_returns_correct_columns_and_values(self):
+        d = {"test": ["Hello", "Bye"], "design_id": ["Hello", "Bye"], "design_name": ["Hello", "Bye"], 
+                "file_name": ["Hello", "Bye"], "file_location": ["Hello", "Bye"], "Hello": ["Hello", "Bye"]}
+        test_df = {"design": pd.DataFrame(data=d)}
+        result = create_dim_design(test_df)
+        d2 = {"design_id": ["Hello", "Bye"], "design_name": ["Hello", "Bye"], "file_name": ["Hello", "Bye"], 
+            "file_location": ["Hello", "Bye"]}
+        expected_df = pd.DataFrame(data=d2)
+        expected_result = expected_df.copy()
+        assert result.equals(expected_result)
+
+class TestCreateDimStaff:
+    def test_dim_staff_returns_dataframe(self):
+        d = {"staff_id": ["Hello", "Bye"], "first_name": ["Hello", "Bye"], "last_name": ["Hello", "Bye"], "department_id": ["Hello", "Bye"]}
+        d2 = {"department_name": ["Hello", "Bye"], "location": ["Hello", "Bye"], "email_address": ["Hello", "Bye"], "department_id": ["Hello", "Bye"]}
+        test_df = {"staff": pd.DataFrame(data=d), "department": pd.DataFrame(data=d2)}
+        result = create_dim_staff(test_df)
+        assert isinstance(result, pd.DataFrame)  
+
+    def test_dim_staff_returns_correct_columns_and_values(self):
+        d = {"staff_id": ["Hello", "Bye"], "first_name": ["Hello", "Bye"], "last_name": ["Hello", "Bye"], "department_id": ["Hello", "Bye"]}
+        d2 = {"department_name": ["Hello", "Bye"], "location": ["Hello", "Bye"], "email_address": ["Hello", "Bye"], "department_id": ["Hello", "Bye"]}
+        test_df = {"staff": pd.DataFrame(data=d), "department": pd.DataFrame(data=d2)}
+        result = create_dim_staff(test_df)
+        expected_d = {"staff_id": ["Hello", "Bye"], "first_name": ["Hello", "Bye"], "last_name": ["Hello", "Bye"], "department_name": ["Hello", "Bye"], "location": ["Hello", "Bye"], "email_address": ["Hello", "Bye"]}
+        expected_df = pd.DataFrame(data=expected_d)
+        expected_result = expected_df.copy()
+        assert result.equals(expected_result)     
+    
\ No newline at end of file