Add unit tests

Author: Val Brodsky

libs/labelbox/src/labelbox/schema/user_group_v2.py

@@ -192,7 +192,7 @@ def upload_members(
 
     def export_members(self, group_id: str) -> Optional[List[Member]]:
         warnings.warn(
-            "The upload_members for UserGroupV2 is in beta. The method name and signature may change in the future.”",
+            "The export_members for UserGroupV2 is in beta. The method name and signature may change in the future.",
         )
 
         if not group_id:
@@ -216,17 +216,18 @@ def export_members(self, group_id: str) -> Optional[List[Member]]:
             raise ResourceNotFoundError(message="The user group is not found.")
         data = result["userGroupV2"]
 
-        # Parse CSV string into list of members
-        csv_lines = data["membersAsCSV"].strip().split("\n")
-        members_list = []
+        return self._parse_members_csv(data["membersAsCSV"])
+
+    def _parse_members_csv(self, csv_data: str) -> List[Member]:
+        csv_lines = csv_data.strip().split("\n")
+        if not csv_lines:
+            return []
 
+        members_list = []
         # Skip header row
         for email in csv_lines[1:]:
-            members_list.append(
-                Member(
-                    email=email.strip(),
-                )
-            )
+            if email.strip():  # Skip empty lines
+                members_list.append(Member(email=email.strip()))
 
         return members_list
 

libs/labelbox/tests/unit/schema/test_user_group_v2.py

@@ -0,0 +1,50 @@
+from unittest.mock import MagicMock
+
+import pytest
+
+from labelbox.schema.user_group_v2 import Member, UserGroupV2
+
+
+@pytest.fixture
+def client():
+    return MagicMock()
+
+
+def test_parse_members_csv_empty(client):
+    group = UserGroupV2(client)
+    assert group._parse_members_csv("") == []
+    assert group._parse_members_csv("\n") == []
+
+
+def test_parse_members_csv_header_only(client):
+    group = UserGroupV2(client)
+    assert group._parse_members_csv("email\n") == []
+
+
+def test_parse_members_csv_single_member(client):
+    group = UserGroupV2(client)
+    result = group._parse_members_csv("email\n[email protected]")
+    assert len(result) == 1
+    assert isinstance(result[0], Member)
+    assert result[0].email == "[email protected]"
+
+
+def test_parse_members_csv_multiple_members(client):
+    group = UserGroupV2(client)
+    csv_data = "email\n[email protected]\n[email protected]\n[email protected]"
+    result = group._parse_members_csv(csv_data)
+    assert len(result) == 3
+    assert [m.email for m in result] == [
+        "[email protected]",
+        "[email protected]",
+        "[email protected]",
+    ]
+
+
+def test_parse_members_csv_handles_whitespace(client):
+    group = UserGroupV2(client)
+    csv_data = "email\n  [email protected]  \n\n[email protected]\n"
+    result = group._parse_members_csv(csv_data)
+    assert len(result) == 2
+    assert result[0].email == "[email protected]"
+    assert result[1].email == "[email protected]"