From ac166fa8d706a150666f37a3830505c5fd5c0d4c Mon Sep 17 00:00:00 2001
From: "Xin (Eric) Wang" <aifumu.wx@gmail.com>
Date: Wed, 8 May 2019 17:14:42 -0700
Subject: [PATCH] Correct pytest code for the vector env render (#82)
Correct pytest code for the vector env render
---
test/test_habitat_env.py | 14 +++++++++++---
1 file changed, 11 insertions(+), 3 deletions(-)
diff --git a/test/test_habitat_env.py b/test/test_habitat_env.py
index 628e94c0c..5c1c5961e 100644
--- a/test/test_habitat_env.py
+++ b/test/test_habitat_env.py
@@ -221,9 +221,17 @@ def test_rl_vectorized_envs():
assert len(rewards) == num_envs
assert len(dones) == num_envs
assert len(infos) == num_envs
- assert envs.render(
- mode="rgb_array"
- ).all(), "vector env render is broken"
+
+ tiled_img = envs.render(mode="rgb_array")
+ new_height = int(np.ceil(np.sqrt(NUM_ENVS)))
+ new_width = int(np.ceil(float(NUM_ENVS) / new_height))
+ h, w, c = observations[0]["rgb"].shape
+ assert tiled_img.shape == (
+ h * new_height,
+ w * new_width,
+ c,
+ ), "vector env render is broken"
+
if (i + 1) % configs[0].ENVIRONMENT.MAX_EPISODE_STEPS == 0:
assert all(dones), "dones should be true after max_episode steps"
--
GitLab