From cf5b245641344f268266eca9cbfdd20336fefefa Mon Sep 17 00:00:00 2001
From: bal7hazar <bal7hazar@proton.me>
Date: Mon, 9 Sep 2024 15:20:17 -0400
Subject: [PATCH] =?UTF-8?q?=E2=9C=85=20Update=20test=20to=20remove=20lengt?=
 =?UTF-8?q?h?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 crates/pathfinding/src/helpers/astar.cairo | 44 +++++++++++++++++-----
 1 file changed, 35 insertions(+), 9 deletions(-)

diff --git a/crates/pathfinding/src/helpers/astar.cairo b/crates/pathfinding/src/helpers/astar.cairo
index 5d9b2cb..71d0457 100644
--- a/crates/pathfinding/src/helpers/astar.cairo
+++ b/crates/pathfinding/src/helpers/astar.cairo
@@ -56,6 +56,7 @@ pub impl Astar of AstarTrait {
             path.append(current.position);
             current = heap.at(current.source);
         };
+        println!("path: {:?}", path);
 
         // [Return] The path from the start to the target
         path.span()
@@ -137,8 +138,8 @@ mod test {
 
     #[test]
     fn test_astar_search_small() {
-        // x 1 1
-        // 1 0 1
+        // x───┐
+        // 1 0 │
         // 0 1 s
         let grid: felt252 = 0x1EB;
         let width = 3;
@@ -146,15 +147,14 @@ mod test {
         let from = 0;
         let to = 8;
         let mut path = Astar::search(grid, width, height, from, to);
-        assert_eq!(path.len(), 4);
         assert_eq!(path, array![8, 7, 6, 3].span());
     }
 
     #[test]
     fn test_astar_search_medium() {
-        // 1 x 0 0
-        // 1 0 1 1
-        // 1 1 1 1
+        // ┌─x 0 0
+        // │ 0 1 1
+        // └─────┐
         // 1 1 1 s
         let grid: felt252 = 0xCBFF;
         let width = 4;
@@ -162,7 +162,6 @@ mod test {
         let from = 0;
         let to = 14;
         let mut path = Astar::search(grid, width, height, from, to);
-        assert_eq!(path.len(), 7);
         assert_eq!(path, array![14, 15, 11, 7, 6, 5, 4].span());
     }
 
@@ -180,7 +179,7 @@ mod test {
         // 0 0 0 0 1 1 1 1 1 1 1 1 1 1 1 1 │ 0
         // 0 0 0 1 1 1 1 1 1 1 1 1 1 1 1 1 │ 0
         // 0 0 0 0 0 1 1 1 1 1 1 1 1 1 1 1 │ 0
-        // 0 0 0 0 0 0 1 1 1 1 1 1 1 1 1 1 x 0
+        // 0 0 0 0 0 0 1 1 1 1 1 1 1 1 1 1 s 0
         // 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0
         let grid: felt252 = 0x7F003F800FB001FC003C481F1F0FFFE1EEF83FFE1FFF81FFE03FF80000;
         let width = 18;
@@ -188,6 +187,33 @@ mod test {
         let from = 19;
         let to = 224;
         let mut path = Astar::search(grid, width, height, from, to);
-        assert_eq!(path.len(), 22);
+        assert_eq!(
+            path,
+            array![
+                224,
+                225,
+                207,
+                189,
+                171,
+                172,
+                154,
+                136,
+                118,
+                117,
+                116,
+                115,
+                114,
+                113,
+                112,
+                94,
+                93,
+                75,
+                57,
+                56,
+                55,
+                37
+            ]
+                .span()
+        );
     }
 }