[feat] add ids list

Author: bgallois

fastanalysis/load.py

@@ -60,6 +60,15 @@ def getObjectNumber(self):
         maxObj = len(set(self.tracking.id.values))
         return maxObj
 
+    def getIds(self):
+        """Get the list of object's IDs.
+
+        :return: [List of IDs]
+        :rtype: [int]
+        """
+        obj = list(set(self.tracking.id.values))
+        return obj
+
     def getObjects(self, ids):
         """Get the tracking data for a list of objects.
 

fastanalysis/tests/test_load.py

@@ -24,6 +24,12 @@ def test_object_number():
     assert objectNumber == 207
 
 
+def test_object_ids():
+    """Test number of objects."""
+    objectNumber = load.Load("tests/tracking.txt").getIds()
+    assert objectNumber == list(range(207))
+
+
 def test_get_keys():
     """Test get list of keys."""
     keys = load.Load("tests/tracking.txt").getKeys()