diff --git a/tests/test_dressing_env.py b/tests/test_dressing_env.py index a657f43b..3d982d29 100644 --- a/tests/test_dressing_env.py +++ b/tests/test_dressing_env.py @@ -7,3 +7,19 @@ def test_dressing_env_initialization(): """Tests that the dressing environment initializes without crashing.""" env = DressingEnv(graphics=False) assert isinstance(env, DressingEnv) + +def test_dressing_env_cloth_access(): + """Tests that the cloth is present in the dressing environment.""" + env = DressingEnv(graphics=False) + cloth = env.get_cloth() + assert cloth is not None + +def test_dressing_env_cloth_particles(): + """Tests that the cloth particles are present in the dressing environment.""" + env = DressingEnv(graphics=False) + cloth = env.get_cloth() + cloth.GetParticles() + env.step() + cloth_particles = cloth.data['particles'] # Assuming cloth has a 'data' attribute with 'particles' + assert cloth_particles is not None + assert len(cloth_particles) > 0 # Ensure there are particles present \ No newline at end of file