Fix test failure.

huawei-noah · Jan 30, 2024 · 14452d0 · 14452d0
1 parent b731d8e
commit 14452d0
Showing 1 changed file with 16 additions and 3 deletions.
diff --git a/smarts/core/tests/test_sensors.py b/smarts/core/tests/test_sensors.py
@@ -30,7 +30,10 @@
 from helpers.scenario import temp_scenario
 
 from smarts.core import glsl
-from smarts.core.agent_interface import CustomRenderCameraDependency
+from smarts.core.agent_interface import (
+    CustomRenderCameraDependency,
+    CustomRenderVariableDependency,
+)
 from smarts.core.coordinates import Heading, Pose, RefLinePoint
 from smarts.core.observations import CustomRenderData
 from smarts.core.plan import Plan
@@ -275,10 +278,16 @@ def test_custom_render_sensor():
             resolution=1,
             fragment_shader_path=frag_shader,
             renderer=renderer,
-            render_dependencies=(),
+            render_dependencies=(
+                CustomRenderVariableDependency(
+                    value=1.0,
+                    variable_name="scale",
+                ),
+            ),
             ogm_sensor=None,
             top_down_rgb_sensor=None,
-            dagm_sensor=None,
+            drivable_area_grid_map_sensor=None,
+            occlusion_map_sensor=None,
         )
         sensor = sensor_gen(name="simplex")
         sensor2 = sensor_gen(
@@ -287,6 +296,10 @@ def test_custom_render_sensor():
                 CustomRenderCameraDependency(
                     camera_dependency_name=sensor.name, variable_name="iChannel0"
                 ),
+                CustomRenderVariableDependency(
+                    value=10.0,
+                    variable_name="scale",
+                ),
             ),
         )