16
16
import unittest
17
17
from pathlib import Path
18
18
19
+ import numpy as np
19
20
import pytest
20
21
from pydantic import ValidationError
21
22
from rai .types import (
@@ -43,9 +44,9 @@ def test_position():
43
44
assert isinstance (position .y , float )
44
45
assert isinstance (position .z , float )
45
46
46
- assert position .x == 1.1
47
- assert position .y == 2.2
48
- assert position .z == 3.3
47
+ assert np . isclose ( position .x , 1.1 )
48
+ assert np . isclose ( position .y , 2.2 )
49
+ assert np . isclose ( position .z , 3.3 )
49
50
50
51
51
52
def test_quaternion ():
@@ -56,10 +57,10 @@ def test_quaternion():
56
57
assert isinstance (quaternion .z , float )
57
58
assert isinstance (quaternion .w , float )
58
59
59
- assert quaternion .x == 0.1
60
- assert quaternion .y == 0.2
61
- assert quaternion .z == 0.3
62
- assert quaternion .w == 0.4
60
+ assert np . isclose ( quaternion .x , 0.1 )
61
+ assert np . isclose ( quaternion .y , 0.2 )
62
+ assert np . isclose ( quaternion .z , 0.3 )
63
+ assert np . isclose ( quaternion .w , 0.4 )
63
64
64
65
65
66
def test_pose ():
@@ -71,8 +72,8 @@ def test_pose():
71
72
assert isinstance (pose .position , Point )
72
73
assert isinstance (pose .orientation , Quaternion )
73
74
74
- assert pose .position .x == 1.1
75
- assert pose .orientation .w == 0.4
75
+ assert np . isclose ( pose .position .x , 1.1 )
76
+ assert np . isclose ( pose .orientation .w , 0.4 )
76
77
77
78
78
79
@pytest .fixture
0 commit comments