TDD Pandas - [REFACTOR] Extract generate grade book function

diligejy · Jan 19, 2024 · 520f504 · 520f504
1 parent 5fe340c
commit 520f504
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/PyCon_PL/Test_Driven_Pandas/tests/test_gradebook.py b/PyCon_PL/Test_Driven_Pandas/tests/test_gradebook.py
@@ -1,5 +1,8 @@
 import pandas as pd 
 
+def generate_gradebook(students_df:pd.DataFrame) -> dict[int, pd.DataFrame]:
+    return {}
+
 def test_results_are_grouped_by_student_group():
     students = [{
         "ID" : 1,
@@ -10,5 +13,5 @@ def test_results_are_grouped_by_student_group():
     }]
     students_df = pd.DataFrame(data=students).set_index("ID")
 
-    result = {}
+    result = generate_gradebook(students_df=students_df)
     assert list(result.keys()) == [1]