pass in IndexPair arguments

release/4.3a0
senselessdev1 2022-09-26 10:56:44 -04:00
parent 8007271f4b
commit a2109a2437
1 changed files with 2 additions and 2 deletions

View File

@ -28,8 +28,8 @@ class TestDsfTrackGenerator(GtsamTestCase):
# For each image pair (i1,i2), we provide a (K,2) matrix of corresponding image indices (k1,k2). # For each image pair (i1,i2), we provide a (K,2) matrix of corresponding image indices (k1,k2).
matches_dict = MatchIndicesMap() matches_dict = MatchIndicesMap()
matches_dict[(0, 1)] = np.array([[0, 0], [1, 1]]) matches_dict[gtsam.IndexPair(0, 1)] = np.array([[0, 0], [1, 1]])
matches_dict[(1, 2)] = np.array([[2, 0], [1, 1]]) matches_dict[gtsam.IndexPair(1, 2)] = np.array([[2, 0], [1, 1]])
tracks = DsfTrackGenerator().generate_tracks_from_pairwise_matches(matches_dict, keypoints_list) tracks = DsfTrackGenerator().generate_tracks_from_pairwise_matches(matches_dict, keypoints_list)
assert len(tracks) == 3 assert len(tracks) == 3