Test for add_edges

bevyengine · Jan 8, 2025 · e0aaca9 · e0aaca9
1 parent 1b67f1d
commit e0aaca9
Showing 1 changed file with 19 additions and 0 deletions.
diff --git a/crates/bevy_input_focus/src/directional_navigation.rs b/crates/bevy_input_focus/src/directional_navigation.rs
@@ -512,6 +512,25 @@ mod tests {
         assert_eq!(map.get_neighbor(c, CompassOctant::West), None);
     }
 
+    #[test]
+    fn edges() {
+        let mut world = World::new();
+        let a = world.spawn_empty().id();
+        let b = world.spawn_empty().id();
+        let c = world.spawn_empty().id();
+
+        let mut map = DirectionalNavigationMap::default();
+        map.add_edges(&[a, b, c], CompassOctant::East);
+
+        assert_eq!(map.get_neighbor(a, CompassOctant::East), Some(b));
+        assert_eq!(map.get_neighbor(b, CompassOctant::East), Some(c));
+        assert_eq!(map.get_neighbor(c, CompassOctant::East), None);
+
+        assert_eq!(map.get_neighbor(a, CompassOctant::West), None);
+        assert_eq!(map.get_neighbor(b, CompassOctant::West), Some(a));
+        assert_eq!(map.get_neighbor(c, CompassOctant::West), Some(b));
+    }
+
     #[test]
     fn looping_edges() {
         let mut world = World::new();