Skip to content

Commit 220c9e4

Browse files
committed
fix: compare floats with np.isclose
1 parent 93623dd commit 220c9e4

File tree

1 file changed

+10
-9
lines changed

1 file changed

+10
-9
lines changed

tests/rai_sim/test_simulation_bridge.py

Lines changed: 10 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -16,6 +16,7 @@
1616
import unittest
1717
from pathlib import Path
1818

19+
import numpy as np
1920
import pytest
2021
from pydantic import ValidationError
2122
from rai.types import (
@@ -43,9 +44,9 @@ def test_position():
4344
assert isinstance(position.y, float)
4445
assert isinstance(position.z, float)
4546

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)
4950

5051

5152
def test_quaternion():
@@ -56,10 +57,10 @@ def test_quaternion():
5657
assert isinstance(quaternion.z, float)
5758
assert isinstance(quaternion.w, float)
5859

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)
6364

6465

6566
def test_pose():
@@ -71,8 +72,8 @@ def test_pose():
7172
assert isinstance(pose.position, Point)
7273
assert isinstance(pose.orientation, Quaternion)
7374

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)
7677

7778

7879
@pytest.fixture

0 commit comments

Comments
 (0)