def test_all_frame_ids_single_track(self):
# From a single track
n = 10
t = Track(1)
for i in range(n):
t.append(TrackState(i))
ts = TrackSet([t])
nt.assert_equal(ts.all_frame_ids(), set(range(10)))
def test_all_frame_ids_multitrack(self):
# Across multiple tracks
n = 10
t1 = Track(1)
for i in range(0, n):
t1.append(TrackState(i))
t2 = Track(2)
for i in range(n, n + 5):
t2.append(TrackState(i))
ts = TrackSet([t1, t2])
nt.assert_equal(ts.all_frame_ids(), set(range(n + 5)))
def test_all_frame_ids_single_track(self):
# From a single track
n = 10
t = Track(1)
for i in range(n):
t.append(TrackState(i))
ts = TrackSet([t])
nt.assert_equal(
ts.all_frame_ids(),
set(range(10))
)
def test_all_frame_ids_multitrack(self):
# Across multiple tracks
n = 10
t1 = Track(1)
for i in range(0, n):
t1.append(TrackState(i))
t2 = Track(2)
for i in range(n, n+5):
t2.append(TrackState(i))
ts = TrackSet([t1, t2])
nt.assert_equal(
ts.all_frame_ids(),
set(range(n+5))
)
