a
i‚jZ ã @ sd d Z ddlmZ ddlmZ ddlZddlZe eedƒd¡G dd„ dej ƒƒZ
e d kr`e ¡ dS )
z Tests for the MuJoCo renderer.é )Úabsltest)Ú
parameterizedNZ GLContextz MuJoCo rendering is disabledc @ s< e Zd Zdd„ Zdd„ Zdd„ Zdd„ Zd d
„ Zd d „ Zd
S )ÚMuJoCoRendererTestc C s d}t j |¡}t |¡}t |dd¡T}t ||¡ | td¡ | |d¡ W d ƒ n1 sd0 Y W d ƒ n1 s‚0 Y d S )NúH
é2 zcamera "b" does not existÚb© ÚmujocoÚMjModelÚfrom_xml_stringÚMjDataÚRendererÚ
mj_forwardÚassertRaisesRegexÚ
ValueErrorÚ update_scene©ÚselfÚxmlÚmodelÚdataÚrenderer© r úT/home/ghrups/robot-bench/.venv/lib64/python3.9/site-packages/mujoco/renderer_test.pyÚ!test_renderer_unknown_camera_name s
z4MuJoCoRendererTest.test_renderer_unknown_camera_namec C s d}t j |¡}t |¡}t |dd¡T}t ||¡ | td¡ | |d¡ W d ƒ n1 sd0 Y W d ƒ n1 s‚0 Y d S )Nr r z-2 is out of rangeéþÿÿÿr r r r r Ú test_renderer_camera_under_range+ s
z3MuJoCoRendererTest.test_renderer_camera_under_rangec C s d}t j |¡}t |¡}t |dd¡T}t ||¡ | td¡ | |d¡ W d ƒ n1 sd0 Y W d ƒ n1 s‚0 Y d S )Nr r z1 is out of rangeé r r r r r Ú test_renderer_camera_over_range: s
z2MuJoCoRendererTest.test_renderer_camera_over_rangec C s˜ d}t j |¡}t |¡}t |dd¡\}t ||¡ | |d¡ | ¡ ¡ }d}|D ]}|dkrVd} qlqV| |¡ W d ƒ n1 sŠ0 Y d S )Nú¶
r ÚcloseupFr T)
r r
r r r
r r ÚrenderÚflattenÚ
assertTrue)r r r r r ÚpixelsZ
not_all_blackZpixelr r r Útest_renderer_renders_sceneI s
z.MuJoCoRendererTest.test_renderer_renders_scenec C sî d}t j |¡}t |¡}t ||¡ t |dd¡¦}| |d¡ | ¡ g}d}t|ƒD ]r\}}||j dd d …f< t ||¡ | |d¡ |
| ¡ ¡ | |d |d ¡ | ||d || k
¡ ¡ qXW d ƒ n1 sà0 Y d S ) Nr r r ))ç ð?ç r' r& )r' r& r' r& )r' r' r& r& r r éÿÿÿÿr )r r
r r r r
r r! Ú enumerateZ geom_rgbaÚappendZ assertIsNotÚ assertFalseÚall) r r r r r r$ ÚcolorsÚiÚcolorr r r Ú test_renderer_output_without_outb s
z3MuJoCoRendererTest.test_renderer_output_without_outc C s d}d}t g |¢d‘R t j¡}tj |¡}t |¡}t ||¡ tj|g|¢R Ž ¶}| |d¡ |
t |dk¡¡ |j |d}|
||¡ | t |dk¡¡ d}| ||¡ | t¡0 |j t g |¢d‘R t j¡d W d ƒ n1 sò0 Y W d ƒ n1 s0 Y d S )Nr )r r é r r )Úout)é
r3 )ÚnpZzerosZuint8r r
r r r r
r r# r, r! ZassertIsr+ ZassertNotEqualZ assertRaisesr ) r r Z render_sizeZ
render_outr r r r$ Zfailing_render_sizer r r Ú test_renderer_output_with_out‚ s
z0MuJoCoRendererTest.test_renderer_output_with_outN) Ú__name__Ú
__module__Ú __qualname__r r r r% r0 r5 r r r r r s r Ú__main__)
Ú__doc__Z absl.testingr r r Únumpyr4 Z
skipUnlessÚhasattrZTestCaser r6 Úmainr r r r Ú s
ÿ