Merge 043d63f into 81b3db1

WPI-MMR · Feb 2, 2021 · 1de6f80 · 1de6f80
2 parents 81b3db1 + 043d63f
commit 1de6f80
Showing 1 changed file with 9 additions and 3 deletions.
diff --git a/gym_solo/envs/test_solo8v2vanilla_realtime.py b/gym_solo/envs/test_solo8v2vanilla_realtime.py
@@ -12,6 +12,15 @@
 
 
 class TestSolo8v2VanillaRealtimeEnv(unittest.TestCase):
+  @classmethod
+  def setUpClass(cls):
+    cls.display = pyvirtualdisplay.Display(visible=False, size=(1400, 900))
+    cls.display.start()
+
+  @classmethod
+  def tearDownClass(cls):
+    cls.display.stop()
+
   def test_init_non_realtime(self):
     config = solo_env.RealtimeSolo8VanillaConfig()
     config.dt = 21
@@ -38,9 +47,6 @@ def test_client_patching(self):
     self.assertEqual(env.client.stepSimulation(), solo_types.no_op)
 
   def test_step_and_reset(self):
-    display = pyvirtualdisplay.Display(visible=False, size=(1400, 900))
-    display.start()
-
     # Realtime only works in GUI mode
     env = solo_env.RealtimeSolo8VanillaEnv(use_gui=True)
     env.obs_factory.register_observation(obs.TorsoIMU(env.robot))