12345qiupeng
/
gtsam
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Rename of test

release/4.3a0
Frank Dellaert 2022-10-23 16:26:41 -07:00
parent f68f0dbe41
commit 90c990dc33
1 changed files with 6 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
python/gtsam/tests/test_dsf_map.py → python/gtsam/tests/test_DSFMap.py
View File

@ -15,8 +15,7 @@ from __future__ import print_function
import unittest import unittest
from typing import Tuple from typing import Tuple
import gtsam from gtsam import DSFMapIndexPair, IndexPair, IndexPairSetAsArray
from gtsam import IndexPair
from gtsam.utils.test_case import GtsamTestCase from gtsam.utils.test_case import GtsamTestCase
@ -29,10 +28,10 @@ class TestDSFMap(GtsamTestCase):
def key(index_pair) -> Tuple[int, int]: def key(index_pair) -> Tuple[int, int]:
return index_pair.i(), index_pair.j() return index_pair.i(), index_pair.j()
dsf = gtsam.DSFMapIndexPair() dsf = DSFMapIndexPair()
pair1 = gtsam.IndexPair(1, 18) pair1 = IndexPair(1, 18)
self.assertEqual(key(dsf.find(pair1)), key(pair1)) self.assertEqual(key(dsf.find(pair1)), key(pair1))
pair2 = gtsam.IndexPair(2, 2) pair2 = IndexPair(2, 2)
# testing the merge feature of dsf # testing the merge feature of dsf
dsf.merge(pair1, pair2) dsf.merge(pair1, pair2)
@ -45,7 +44,7 @@ class TestDSFMap(GtsamTestCase):
k'th detected keypoint in image i. For the data below, merging such k'th detected keypoint in image i. For the data below, merging such
measurements into feature tracks across frames should create 2 distinct sets. measurements into feature tracks across frames should create 2 distinct sets.
""" """
dsf = gtsam.DSFMapIndexPair() dsf = DSFMapIndexPair()
dsf.merge(IndexPair(0, 1), IndexPair(1, 2)) dsf.merge(IndexPair(0, 1), IndexPair(1, 2))
dsf.merge(IndexPair(0, 1), IndexPair(3, 4)) dsf.merge(IndexPair(0, 1), IndexPair(3, 4))
dsf.merge(IndexPair(4, 5), IndexPair(6, 8)) dsf.merge(IndexPair(4, 5), IndexPair(6, 8))
@ -56,7 +55,7 @@ class TestDSFMap(GtsamTestCase):
for i in sets: for i in sets:
set_keys = [] set_keys = []
s = sets[i] s = sets[i]
for val in gtsam.IndexPairSetAsArray(s): for val in IndexPairSetAsArray(s):
set_keys.append((val.i(), val.j())) set_keys.append((val.i(), val.j()))
merged_sets.add(tuple(set_keys)) merged_sets.add(tuple(set_keys))