def test_ts_feat_desc(self):
t = Track()
f0 = Feature(mag=0)
d0 = Descriptor(4)
d0[:] = 0
t.append(TrackState(0, f0, d0))
f1 = Feature(mag=1)
d1 = Descriptor(4)
d1[:] = 1
t.append(TrackState(1, f1, d1))
nose.tools.assert_equal(t.size, 2)
nose.tools.assert_equal(len(t), 2)
ts0 = t.find_state(0)
nose.tools.assert_equal(ts0.feature, f0)
numpy.testing.assert_equal(ts0.descriptor, d0)
ts1 = t.find_state(1)
nose.tools.assert_equal(ts1.feature, f1)
numpy.testing.assert_equal(ts1.descriptor, d1)
# Delete local descriptor references, get track states and check
# descriptor values manually
del f0, f1, d0, d1, ts0, ts1
# Now, only the track-state in C++ has feature/descriptor references
numpy.testing.assert_equal(t.find_state(0).descriptor, [0, 0, 0, 0])
numpy.testing.assert_equal(t.find_state(1).descriptor, [1, 1, 1, 1])
def test_track_find(self):
t = Track()
t.append(TrackState(0))
t.append(TrackState(1))
t.append(TrackState(5))
t.append(TrackState(9))
ts = t.find_state(0)
nose.tools.assert_is_not_none(ts)
nose.tools.assert_equal(ts.frame_id, 0)
ts = t.find_state(1)
nose.tools.assert_is_not_none(ts)
nose.tools.assert_equal(ts.frame_id, 1)
ts = t.find_state(5)
nose.tools.assert_is_not_none(ts)
nose.tools.assert_equal(ts.frame_id, 5)
ts = t.find_state(9)
nose.tools.assert_is_not_none(ts)
nose.tools.assert_equal(ts.frame_id, 9)
nose.tools.assert_raises(
IndexError,
t.find_state, 10
)
t.append(TrackState(10))
nose.tools.assert_is_not_none(t.find_state(10))
nose.tools.assert_equal(t.find_state(10).frame_id, 10)
def test_track_find(self):
t = Track()
t.append(TrackState(0))
t.append(TrackState(1))
t.append(TrackState(5))
t.append(TrackState(9))
ts = t.find_state(0)
nose.tools.assert_is_not_none(ts)
nose.tools.assert_equal(ts.frame_id, 0)
ts = t.find_state(1)
nose.tools.assert_is_not_none(ts)
nose.tools.assert_equal(ts.frame_id, 1)
ts = t.find_state(5)
nose.tools.assert_is_not_none(ts)
nose.tools.assert_equal(ts.frame_id, 5)
ts = t.find_state(9)
nose.tools.assert_is_not_none(ts)
nose.tools.assert_equal(ts.frame_id, 9)
nose.tools.assert_raises(IndexError, t.find_state, 10)
t.append(TrackState(10))
nose.tools.assert_is_not_none(t.find_state(10))
nose.tools.assert_equal(t.find_state(10).frame_id, 10)
