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

tests for dunder methods

release/4.3a0
Varun Agrawal 2024-01-16 15:17:26 -05:00
parent f4ea551153
commit b2e4fa2112
1 changed files with 35 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

36
python/gtsam/tests/test_Utilities.py
View File

@ -12,13 +12,14 @@ Author: Varun Agrawal
import unittest import unittest
import numpy as np import numpy as np
from gtsam.utils.test_case import GtsamTestCase
import gtsam import gtsam
from gtsam.utils.test_case import GtsamTestCase
class TestUtilites(GtsamTestCase): class TestUtilites(GtsamTestCase):
"""Test various GTSAM utilities.""" """Test various GTSAM utilities."""
def test_createKeyList(self): def test_createKeyList(self):
"""Test createKeyList.""" """Test createKeyList."""
I = [0, 1, 2] I = [0, 1, 2]
@ -28,6 +29,17 @@ class TestUtilites(GtsamTestCase):
kl = gtsam.utilities.createKeyList("s", I) kl = gtsam.utilities.createKeyList("s", I)
self.assertEqual(kl.size(), 3) self.assertEqual(kl.size(), 3)
def test_KeyList_iteration(self):
"""Tests for KeyList iteration"""
I = [0, 1, 2]
kl = gtsam.utilities.createKeyList(I)
self.assertEqual(len(kl), len(I))
for i, key in enumerate(kl):
self.assertTrue(key in kl)
self.assertEqual(I[i], key)
def test_createKeyVector(self): def test_createKeyVector(self):
"""Test createKeyVector.""" """Test createKeyVector."""
I = [0, 1, 2] I = [0, 1, 2]
@ -37,6 +49,17 @@ class TestUtilites(GtsamTestCase):
kl = gtsam.utilities.createKeyVector("s", I) kl = gtsam.utilities.createKeyVector("s", I)
self.assertEqual(len(kl), 3) self.assertEqual(len(kl), 3)
def test_KeyVector_iteration(self):
"""Tests for KeyVector iteration"""
I = [0, 1, 2]
kv = gtsam.utilities.createKeyVector(I)
self.assertEqual(len(kv), len(I))
for i, key in enumerate(kv):
self.assertTrue(key in kv)
self.assertEqual(I[i], key)
def test_createKeySet(self): def test_createKeySet(self):
"""Test createKeySet.""" """Test createKeySet."""
I = [0, 1, 2] I = [0, 1, 2]
@ -46,6 +69,17 @@ class TestUtilites(GtsamTestCase):
kl = gtsam.utilities.createKeySet("s", I) kl = gtsam.utilities.createKeySet("s", I)
self.assertEqual(kl.size(), 3) self.assertEqual(kl.size(), 3)
def test_KeySet_iteration(self):
"""Tests for KeySet iteration"""
I = [0, 1, 2]
ks = gtsam.utilities.createKeySet(I)
self.assertEqual(len(ks), len(I))
for i, key in enumerate(ks):
self.assertTrue(key in ks)
self.assertEqual(I[i], key)
def test_extractPoint2(self): def test_extractPoint2(self):
"""Test extractPoint2.""" """Test extractPoint2."""
initial = gtsam.Values() initial = gtsam.Values()