Adds indexing by name for ClassLists (#54)

* added get item to classlists with wider range of inputs

* added test for completeness

Author: alexhroom

RATapi/classlist.py

@@ -79,6 +79,15 @@ def __str__(self):
                 output = str(self.data)
         return output
 
+    def __getitem__(self, index: Union[int, slice, str, object]) -> object:
+        """Get an item by its index, name, a slice, or the object itself."""
+        if isinstance(index, (int, slice)):
+            return self.data[index]
+        elif isinstance(index, (str, object)):
+            return self.data[self.index(index)]
+        else:
+            raise IndexError("ClassLists can only be indexed by integers, slices, name strings, or objects.")
+
     def __setitem__(self, index: int, item: object) -> None:
         """Replace the object at an existing index of the ClassList."""
         self._setitem(index, item)

tests/test_classlist.py

@@ -880,3 +880,13 @@ def test_determine_class_handle(input_list: ClassList, expected_type: type) -> N
     for all elements in the ClassList.
     """
     assert ClassList._determine_class_handle(input_list) == expected_type
+
+
+def test_get_item(two_name_class_list):
+    """Test item names can be gotten by name, index, object or slice."""
+    assert two_name_class_list[0] == two_name_class_list["Alice"]
+    assert two_name_class_list[1] == two_name_class_list["Bob"]
+    assert two_name_class_list[:] == two_name_class_list
+    alice = InputAttributes(name="Alice")
+    assert two_name_class_list[alice] == two_name_class_list["Alice"]
+    assert two_name_class_list[alice] == two_name_class_list[0]