| 70 | class OffScreenRendererTest(absltest.TestCase): |
| 71 | |
| 72 | def setUp(self): |
| 73 | super().setUp() |
| 74 | self.model = mock.MagicMock() |
| 75 | self.model.vis.global_.offwidth = _SCREEN_SIZE.width |
| 76 | self.model.vis.global_.offheight = _SCREEN_SIZE.height |
| 77 | |
| 78 | self.surface = mock.MagicMock() |
| 79 | self.renderer = renderer.OffScreenRenderer(self.model, self.surface) |
| 80 | self.renderer._mujoco_context = mock.MagicMock() |
| 81 | |
| 82 | self.viewport = mock.MagicMock() |
| 83 | self.scene = mock.MagicMock() |
| 84 | |
| 85 | self.viewport.width = 3 |
| 86 | self.viewport.height = 3 |
| 87 | self.viewport.dimensions = np.array([3, 3]) |
| 88 | |
| 89 | def test_render_context_initialization(self): |
| 90 | self.renderer._mujoco_context = None |