Update raycaster.py with recent changes

Kinds-of-Intelligence-CFI · Sep 9, 2023 · 35ccf07 · 35ccf07
2 parents b0a700b + e40f473
commit 35ccf07
Showing 1 changed file with 18 additions and 2 deletions.
diff --git a/animalai/animalai/envs/raycastparser.py b/animalai/animalai/envs/raycastparser.py
@@ -111,7 +111,7 @@ def prettyPrint(self, raycast) -> str:
 
         if isinstance(raycast, dict):
             raycast = raycast['rays']
-        
+
         parsedRaycast = self.parse(raycast)
         for i in range(parsedRaycast.shape[0]):
             print(self.listOfObjects[i].name, ":", parsedRaycast[i])
@@ -159,4 +159,20 @@ def prettyPrint(self, raycast) -> str:
     parsedRaycast = rayParser.parse(test_raycast)
     print("Parsed Raycast for Test 3:")
     print(parsedRaycast)
-    rayParser.prettyPrint(test_raycast)
+    rayParser.prettyPrint(test_raycast)
+
+    # Test 4: Mix of objects detected and not detected, including PILLARBUTTON
+    # Description: This test checks if the parser correctly identifies some objects including PILLARBUTTON while ignoring others.
+    rayParser = RayCastParser(
+        [RayCastObjects.ARENA, RayCastObjects.PILLARBUTTON, RayCastObjects.MOVABLE], 7)
+    test_raycast = [1, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1, 0, 0.1,
+                    0, 1, 0, 0, 0, 0, 0, 0, 0, 0, 1, 0, 0.2,
+                    0, 0, 1, 0, 0, 0, 0, 0, 0, 0, 1, 0, 0.3,
+                    0, 0, 0, 1, 0, 0, 0, 0, 0, 0, 1, 0, 0.4,
+                    0, 0, 0, 0, 1, 0, 0, 0, 0, 0, 1, 0, 0.5,
+                    0, 0, 0, 0, 0, 1, 0, 0, 0, 0, 1, 0, 0.6,
+                    0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1, 0, 0]
+    parsedRaycast = rayParser.parse(test_raycast)
+    print("Parsed Raycast for Test 4:")
+    print(parsedRaycast)
+    rayParser.prettyPrint(test_raycast)