Fix test

sea-bass · Jan 6, 2025 · 2bf7f93 · 2bf7f93
1 parent 2a3c3dc
commit 2bf7f93
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/pyrobosim/test/core/test_yaml_utils.py b/pyrobosim/test/core/test_yaml_utils.py
@@ -12,7 +12,7 @@
 from pyrobosim.core.world import World
 from pyrobosim.core.yaml_utils import WorldYamlLoader, WorldYamlWriter
 from pyrobosim.utils.general import get_data_folder
-from pyrobosim.utils.polygon import polygon_and_height_from_footprint
+from pyrobosim.utils.polygon import polygon_and_height_from_footprint, transform_polygon
 from pyrobosim.utils.pose import Pose
 
 
@@ -129,7 +129,9 @@ def test_create_rooms_from_yaml():
 
         assert loader.world.rooms[1].name == "bedroom"
         poly, _ = polygon_and_height_from_footprint(rooms_dict["rooms"][1]["footprint"])
-        assert loader.world.rooms[1].polygon == poly
+        assert loader.world.rooms[1].polygon == transform_polygon(
+            poly, Pose(x=2.625, y=3.5)
+        )
         assert loader.world.rooms[1].wall_width == 0.2
         assert loader.world.rooms[1].viz_color == (0, 1, 0)
         assert loader.world.rooms[1].nav_poses == [