Merge pull request #86 from FastTrackiverse/67-candidatetracks-should…

…-be-indexable-iterable

candidate tracks now indexable

src/fasttrackpy/tracks.py

@@ -14,6 +14,7 @@
 from aligned_textgrid.sequences.tiers import TierGroup
 
 import polars as pl
+from collections.abc import Sequence
 
 from typing import Union, Literal
 import warnings
@@ -381,7 +382,7 @@ def spectrogram(self, **kwargs):
 
 
 
-class CandidateTracks(Track):
+class CandidateTracks(Track, Sequence):
     """A class for candidate tracks for a single formant
 
     You can provide *either*
@@ -510,6 +511,12 @@ def __init__(
 
         self.winner_idx = np.argmin(self.smooth_errors)
         self.winner = self.candidates[self.winner_idx]
+
+    def __getitem__(self, idx:int) -> OneTrack:
+        return self.candidates[idx]
+
+    def __len__(self) -> int:
+        return len(self.candidates)
 
     @property
     def file_name(self):

tests/test_tracks.py

@@ -99,4 +99,14 @@ def test_candidate_tracks_default(self):
         big_df4 = candidates.to_df(which = "all", output="param")
 
         assert isinstance(big_df3, pl.DataFrame)
-        assert isinstance(big_df4, pl.DataFrame)        
+        assert isinstance(big_df4, pl.DataFrame)
+
+    def test_candidate_sequence(self):
+        candidates = CandidateTracks(
+            sound = SOUND
+        )
+
+        assert len(candidates) == 20
+
+        one_cand = candidates[0]
+        assert isinstance(one_cand, OneTrack)