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

add docstring to test and fix assert

release/4.3a0
Varun Agrawal 2022-10-26 12:42:45 -04:00
parent b0533bb198
commit 3da27a5df7
1 changed files with 4 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
python/gtsam/tests/test_DsfTrackGenerator.py
View File

@ -7,7 +7,8 @@ import unittest
import gtsam import gtsam
import numpy as np import numpy as np
from gtsam import IndexPair, KeypointsVector, MatchIndicesMap, Point2, SfmMeasurementVector, SfmTrack2d from gtsam import (IndexPair, KeypointsVector, MatchIndicesMap, Point2,
SfmMeasurementVector, SfmTrack2d)
from gtsam.gtsfm import Keypoints from gtsam.gtsfm import Keypoints
from gtsam.utils.test_case import GtsamTestCase from gtsam.utils.test_case import GtsamTestCase
@ -84,12 +85,12 @@ class TestSfmTrack2d(GtsamTestCase):
"""Tests for SfmTrack2d.""" """Tests for SfmTrack2d."""
def test_sfm_track_2d_constructor(self) -> None: def test_sfm_track_2d_constructor(self) -> None:
""" """ """Test construction of 2D SfM track."""
measurements = SfmMeasurementVector() measurements = SfmMeasurementVector()
measurements.append((0, Point2(10, 20))) measurements.append((0, Point2(10, 20)))
track = SfmTrack2d(measurements=measurements) track = SfmTrack2d(measurements=measurements)
track.measurement(0) track.measurement(0)
track.numberMeasurements() == 1 assert track.numberMeasurements() == 1
if __name__ == "__main__": if __name__ == "__main__":