TDD Pandas - [RED] Second Test for creating grouped result

diligejy · Jan 19, 2024 · 214f70a · 214f70a
1 parent de14ccb
commit 214f70a
Showing 1 changed file with 22 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
@@ -14,4 +14,25 @@ def test_results_are_grouped_by_student_group():
     students_df = pd.DataFrame(data=students).set_index("ID")
 
     result = generate_gradebook(students_df=students_df)
-    assert list(result.keys()) == [1]
+    assert list(result.keys()) == [1]
+
+def test_results_are_grouped_by_student_group_for_students_in_multiple_group():
+    students = [{
+        "ID" : 1,
+        "Name" : "Doe, John",
+        "NetID" : "JXD12345",
+        "Email Address" : "JOHN.DOE@EXAMPLE.EDU",
+        "Group" : 1,
+    },{
+        "ID" : 2,
+        "Name" : "Alec, Curry	",
+        "NetID" : "AMC53511",
+        "Email Address" : "ALEC.CURRY@EXAMPLE.EDU",
+        "Group" : 2,
+    }]
+    students_df = pd.DataFrame(data=students).set_index("ID")
+    result = generate_gradebook(students_df=students_df)
+    assert list(result.keys()) == [1, 2]
+
+
+