From f247aa0701814a19998177be2f3a3a29d9731c8a Mon Sep 17 00:00:00 2001 From: Adil Zouitine Date: Thu, 4 Sep 2025 17:34:06 +0200 Subject: [PATCH] refactor(tests): update processor test assertions to reflect new preprocessor and postprocessor names (#1869) - Changed assertions in multiple processor test files to verify the updated names from "robot_preprocessor" and "robot_postprocessor" to "policy_preprocessor" and "policy_postprocessor" for consistency with recent refactoring. --- tests/processor/test_act_processor.py | 4 ++-- tests/processor/test_diffusion_processor.py | 4 ++-- tests/processor/test_pi0_processor.py | 4 ++-- tests/processor/test_sac_processor.py | 4 ++-- tests/processor/test_smolvla_processor.py | 4 ++-- tests/processor/test_tdmpc_processor.py | 4 ++-- tests/processor/test_vqbet_processor.py | 4 ++-- 7 files changed, 14 insertions(+), 14 deletions(-) diff --git a/tests/processor/test_act_processor.py b/tests/processor/test_act_processor.py index b102aef80..b52df9a27 100644 --- a/tests/processor/test_act_processor.py +++ b/tests/processor/test_act_processor.py @@ -81,8 +81,8 @@ def test_make_act_processor_basic(): preprocessor, postprocessor = make_act_pre_post_processors(config, stats) # Check processor names - assert preprocessor.name == "robot_preprocessor" - assert postprocessor.name == "robot_postprocessor" + assert preprocessor.name == "policy_preprocessor" + assert postprocessor.name == "policy_postprocessor" # Check steps in preprocessor assert len(preprocessor.steps) == 4 diff --git a/tests/processor/test_diffusion_processor.py b/tests/processor/test_diffusion_processor.py index 7fcb92e3a..8ab477fab 100644 --- a/tests/processor/test_diffusion_processor.py +++ b/tests/processor/test_diffusion_processor.py @@ -84,8 +84,8 @@ def test_make_diffusion_processor_basic(): preprocessor, postprocessor = make_diffusion_pre_post_processors(config, stats) # Check processor names - assert preprocessor.name == "robot_preprocessor" - assert postprocessor.name == "robot_postprocessor" + assert preprocessor.name == "policy_preprocessor" + assert postprocessor.name == "policy_postprocessor" # Check steps in preprocessor assert len(preprocessor.steps) == 4 diff --git a/tests/processor/test_pi0_processor.py b/tests/processor/test_pi0_processor.py index 82db59921..2c8c0921a 100644 --- a/tests/processor/test_pi0_processor.py +++ b/tests/processor/test_pi0_processor.py @@ -110,8 +110,8 @@ def test_make_pi0_processor_basic(): ) # Check processor names - assert preprocessor.name == "robot_preprocessor" - assert postprocessor.name == "robot_postprocessor" + assert preprocessor.name == "policy_preprocessor" + assert postprocessor.name == "policy_postprocessor" # Check steps in preprocessor assert len(preprocessor.steps) == 6 diff --git a/tests/processor/test_sac_processor.py b/tests/processor/test_sac_processor.py index d60bb42a6..7a2d2d19c 100644 --- a/tests/processor/test_sac_processor.py +++ b/tests/processor/test_sac_processor.py @@ -86,8 +86,8 @@ def test_make_sac_processor_basic(): ) # Check processor names - assert preprocessor.name == "robot_preprocessor" - assert postprocessor.name == "robot_postprocessor" + assert preprocessor.name == "policy_preprocessor" + assert postprocessor.name == "policy_postprocessor" # Check steps in preprocessor assert len(preprocessor.steps) == 4 diff --git a/tests/processor/test_smolvla_processor.py b/tests/processor/test_smolvla_processor.py index 05bccec80..9a187fc7e 100644 --- a/tests/processor/test_smolvla_processor.py +++ b/tests/processor/test_smolvla_processor.py @@ -117,8 +117,8 @@ def test_make_smolvla_processor_basic(): ) # Check processor names - assert preprocessor.name == "robot_preprocessor" - assert postprocessor.name == "robot_postprocessor" + assert preprocessor.name == "policy_preprocessor" + assert postprocessor.name == "policy_postprocessor" # Check steps in preprocessor assert len(preprocessor.steps) == 6 diff --git a/tests/processor/test_tdmpc_processor.py b/tests/processor/test_tdmpc_processor.py index 14858b8b1..eb38143d7 100644 --- a/tests/processor/test_tdmpc_processor.py +++ b/tests/processor/test_tdmpc_processor.py @@ -89,8 +89,8 @@ def test_make_tdmpc_processor_basic(): ) # Check processor names - assert preprocessor.name == "robot_preprocessor" - assert postprocessor.name == "robot_postprocessor" + assert preprocessor.name == "policy_preprocessor" + assert postprocessor.name == "policy_postprocessor" # Check steps in preprocessor assert len(preprocessor.steps) == 4 diff --git a/tests/processor/test_vqbet_processor.py b/tests/processor/test_vqbet_processor.py index 891e88056..787a3d524 100644 --- a/tests/processor/test_vqbet_processor.py +++ b/tests/processor/test_vqbet_processor.py @@ -89,8 +89,8 @@ def test_make_vqbet_processor_basic(): ) # Check processor names - assert preprocessor.name == "robot_preprocessor" - assert postprocessor.name == "robot_postprocessor" + assert preprocessor.name == "policy_preprocessor" + assert postprocessor.name == "policy_postprocessor" # Check steps in preprocessor assert len(preprocessor.steps) == 4