a
i‚jí ã @ s d Z ddlZddlZddlZddlZddlmZ ddlmZ ddlZddl Z
dZ dZ dZ
dZejd d
„ ƒZG d d „ d ejƒZed
krŒe ¡ dS )z!Tests for MuJoCo Python bindings.é N)Úabsltest)Ú
parameterizedaQ
zë
aö
ap
c c s | |ƒ d V | d ƒ d S ©N© )ÚsetterÚcallbackr r úT/home/ghrups/robot-bench/.venv/lib64/python3.9/site-packages/mujoco/bindings_test.pyÚtemporary_callbackn s r c
s$ e Zd Z‡ fdd„Zdd„ Zdd„ Zdd„ Zd d
„ Zd d „ Ze j
e j j
d
kdddd„ ƒZdd„ Zdd„ Zdd„ Zdd„ Zdd„ Zd d „ Zd d „ Zd d!„ Zd"d#„ Zd$d%„ Zd&d'„ Zd(d)„ Zd*d+„ Zd,d-„ Z d.d/„ Z d0d1„ Z d2d3„ Z d4d5„ Z d6d7„ Z!d8d9„ Z"d:d;„ Z#dd?„ Z%d@dA„ Z&dBdC„ Z'dDdE„ Z(dFdG„ Z)dHdI„ Z*dJdK„ Z+dLdM„ Z,dNdO„ Z-dPdQ„ Z.e/ 0dRe1j2dSfdTe1j3dUfdVe1j4dUfdWe1j5dXfdYe1j6dZfd[e1j7d\fd]e1j8d^fd_e1j9d`fdae1j:dbfg ¡dcdd„ ƒZ;dedf„ Zdkdl„ Z?dmdn„ Z@dodp„ ZAdqdr„ ZBdsdt„ ZCdudv„ ZDdwdx„ ZEdydz„ ZFd{d|„ ZGd}d~„ ZHdd€„ ZIdd‚„ ZJdƒd„„ ZKd…d†„ ZLd‡dˆ„ ZMd‰dŠ„ ZNd‹dŒ„ ZOddŽ„ ZPe j
e j j
d
kdddd„ ƒZQe j
e j j
d
kddd‘d’„ ƒZRd“d”„ ZSd•d–„ ZTd—d˜„ ZUd™dš„ ZVd›dœ„ ZWddž„ ZXdŸd „ ZYd¡d¢„ ZZd£d¤„ Z[d¥d¦„ Z\d§d¨„ Z]d©dª„ Z^d«d¬„ Z_dd®„ Z`d¯d°„ Zad±d²„ Zbd³d´„ Zcdµd¶„ Zdd·d¸„ Zed¹dº„ Zfd»d¼„ Zge/jhd½d½d¾d¿dÀ„ ƒZidÁd„ ZjdÃdÄ„ ZkdÅdÆ„ ZldÇdÈ„ ZmdÉdÊ„ ZndËdÌ„ ZodÍd΄ ZpdÏdЄ ZqdÑdÒ„ ZrdÓdÔ„ Zs‡ ZtS )ÕÚMuJoCoBindingsTestc s* t ƒ ¡ tj t¡| _t | j¡| _d S r ) ÚsuperÚsetUpÚmujocoÚMjModelÚfrom_xml_stringÚTEST_XMLÚmodelÚMjDataÚdata©Úself©Ú __class__r r r w s
zMuJoCoBindingsTest.setUpc C sr d}d}d}t j |||dœ¡}| t |t jjd¡d¡ | t |t jjd¡d¡ | t |t jjd ¡d
¡ d S ) Nz¦
s; s; )z
model_.xmlz model__.xmlZplaner Úboxé Zballé )r
r r Ú assertEqualÚ
mj_name2idÚmjtObjÚ
mjOBJ_GEOM)r Zxml_1Zxml_2Zxml_3r r r r Ú#test_load_xml_can_handle_name_clash| s
ÿÿÿÿz6MuJoCoBindingsTest.test_load_xml_can_handle_name_clashc C sD | td¡$ tj ddddœ¡ W d ƒ n1 s60 Y d S )Nz
Repeated.*ú s asset1s asset2)z asset.xmlzpath/asset.xml)ÚassertRaisesRegexÚ
ValueErrorr
r r r r r r Ú!test_load_xml_repeated_asset_name” s
ÿz4MuJoCoBindingsTest.test_load_xml_repeated_asset_namec C s4 t j | jjg d¢g d¢g d¢g d¢g d¢g¡ d S )N©r r r )r r çš™™™™™¹?)ç E@r r& )ÚnpÚtestingÚassert_array_equalr Úbody_posr r r r Útest_can_read_arrayœ s þz&MuJoCoBindingsTest.test_can_read_arrayc C s, d| j _tj | j jdgt| j jƒ ¡ d S )Ng|ò°Pkš¿?©r Úqposr' r( r) Úlenr r r r Útest_can_set_array¢ s ÿz%MuJoCoBindingsTest.test_can_set_arrayc C s0 | j j}d| j _tj |dgt| j jƒ ¡ d S )Ngsh‘í|?é?r, )r Zqpos_refr r r Útest_array_is_a_view¨ s z'MuJoCoBindingsTest.test_array_is_a_viewÚpypyzrequires sys.getrefcount)Úreasonc C s˜ t j t¡}|j}|j}d|d d …< ~| |d d¡ g }t |¡d }|j }| t |¡| d¡ ~| t |¡| d¡ ~| t |¡| d¡ d S )Nr r é r )
r
r r r Úqpos0Ú qpos_springr ÚsysÚ getrefcountÚbase)r r r4 r5 ÚdummyZ
base_refcountZcapsuler r r Ú test_array_keeps_struct_alive¯ s z0MuJoCoBindingsTest.test_array_keeps_struct_alivec C s¼ t | jt jjd¡}| | j d¡| j |¡¡ | | j d¡j| j |¡j¡ | | j d¡jj
d¡ | j d¡j}d| jj|< t j
|dg¡ d| j d¡_t j
| jj| dg¡ d S )NZ
myactuator)r é é )r
r r r ZmjOBJ_ACTUATORÚassertIsr ZactuatorZctrlr Úshaper' r( r) )r Z actuator_idZctrl_from_indexerr r r Ú!test_named_indexing_actuator_ctrlÍ s ÿÿ þ z4MuJoCoBindingsTest.test_named_indexing_actuator_ctrlc C s< | td¡ | j d¡ W d ƒ n1 s.0 Y d S ©Nz Ú geom_sizer' r( r) )r Úbox_idZsize_from_indexerr r r Ú test_named_indexing_geom_size s z0MuJoCoBindingsTest.test_named_indexing_geom_sizec C sÈ t | jt jjd¡}| | j d¡| j |¡¡ | | j d¡j| j |¡j¡ | | j d¡jj d¡ | j d¡j}g d¢| jj
|< t j
|g d¢¡ g d¢| j d¡_t j
| jj
| g d¢¡ d S )NrN ©é )r; é
é é )é é é r3 )r
r r r r r= rB Úquatr r> Z geom_quatr' r( r) )r rS Zquat_from_indexerr r r Ú test_named_indexing_geom_quat s z0MuJoCoBindingsTest.test_named_indexing_geom_quatc C sä t | jt jjd¡}| | j d¡| j |¡¡ | | j d¡j| j |¡j¡ | | j d¡jj
d¡ | j d¡j}| jj | }g d¢| jj||d …< t j
|g d¢¡ g d¢| j d¡_t j
| jj||d … g d¢¡ d S )NZmyballrU )rV r; r\ r< rV )r[ é r< r\ )r
r r r Ú mjOBJ_JOINTr= r rE r- r r> Z jnt_qposadrr' r( r) )r Z balljoint_idZqpos_from_indexerZqpos_idxr r r Ú test_named_indexing_ragged_qpos+ s ÿ ÿ ÿz2MuJoCoBindingsTest.test_named_indexing_ragged_qposc C s t | jt jjd¡}| | j d¡| j |¡¡ | | j d¡j| j |¡j¡ | | j d¡jj
d¡ | j d¡j}| jj | }t
tdƒd¡| jj||d …d d …f< t j |t
tdƒd¡¡ d| j d¡_t j | jj||d … dgd gd ¡ d S )NÚmyfree)r\ r\ é$ r\ é* )r
r r r r` r= r rE Zcdofr r> Z
jnt_dofadrr' ÚreshapeÚranger( r) )r Z freejoint_idZcdof_from_indexerZdof_idxr r r Ú!test_named_indexing_ragged2d_cdof? s$ ÿ ÿ (ÿ ÿz4MuJoCoBindingsTest.test_named_indexing_ragged2d_cdofc C s d}| |t| j d¡ƒ¡ d S )Nz{<_MjDataGeomViews
id: 1
name: 'mybox'
xmat: array([0., 0., 0., 0., 0., 0., 0., 0., 0.])
xpos: array([0., 0., 0.])
>rN )r Úreprr rB )r Ú
expected_reprr r r Ú test_named_indexing_repr_in_dataU s z3MuJoCoBindingsTest.test_named_indexing_repr_in_datac C s8 t | j d¡ƒ}| |d¡ | d|¡ | |d¡ d S )NrN z<_MjDataBodyViewszxpos: array([0., 0., 0.])ú>)rh r ÚbodyÚassertStartsWithÚassertInÚassertEndsWith©r Z view_reprr r r Ú%test_named_indexing_body_repr_in_data^ s z8MuJoCoBindingsTest.test_named_indexing_body_repr_in_datac C s8 t | j d¡ƒ}| |d¡ | d|¡ | |d¡ d S )NrN z<_MjModelGeomViewsz size: array([0.1, 0.1, 0.1])rk )rh r rB rm rn ro rp r r r Ú!test_named_indexing_repr_in_modeld s z4MuJoCoBindingsTest.test_named_indexing_repr_in_modelc C sv t j t¡}t |¡}| | jjd¡ | | jjd¡ | |jd¡ | |jd¡ | | jj| jj|j|jhd¡ d S )Nr rV )
r
r r r r Ú
assertGreaterr Ú_addressr Ú assertLen©r Úmodel2Údata2r r r Ú%test_addresses_differ_between_structsj s
üùz8MuJoCoBindingsTest.test_addresses_differ_between_structsc C s | | jjjd¡ tj | jjjg d¢¡ | jj}d| jj_| |jd¡ |j}d| jjjd< tj |g d¢¡ d| jj_tj |g d¢¡ d S ) Nçü©ñÒMb`?©r r ç …ëQ¸ž#Àçü©ñÒMbP?r% r )r r% r| çš™™™™™É?)r~ r~ r~ )r r ÚoptÚtimestepr' r( r) Úgravity)r r r r r r Ú#test_mjmodel_can_read_and_write_opt| s
z6MuJoCoBindingsTest.test_mjmodel_can_read_and_write_optc C s6 | | jjjd¡ | jj}d| jj_| |jd¡ d S )Nr g333333ó?)ÚassertNotEqualr ÚstatZmeanmassr )r r„ r r r Ú$test_mjmodel_can_read_and_write_statŠ s
z7MuJoCoBindingsTest.test_mjmodel_can_read_and_write_statc C s8 | | jjjjd¡ d| jjj_| | jjjjd¡ d S )Né3 éd )r r ÚvisZqualityZ
shadowsizer r r r Ú#test_mjmodel_can_read_and_write_vis‘ s z6MuJoCoBindingsTest.test_mjmodel_can_read_and_write_visc C sn t | jjd| jj d¡… dƒ}| |d¡ | jjd }| jj d|¡}t | jj||… dƒ}| |d¡ d S )Nr ó zutf-8ÚtestÚmyplane)Ústrr ÚnamesÚfindr Z name_geomadr)r Z
model_nameÚ start_indexZ end_indexZ geom_namer r r Ú&test_mjmodel_can_access_names_directly— s ÿ z9MuJoCoBindingsTest.test_mjmodel_can_access_names_directlyc C s | j j}| || j j¡ d S r )r rŽ r= )r rŽ r r r Ú test_mjmodel_names_doesnt_copy¤ s z1MuJoCoBindingsTest.test_mjmodel_names_doesnt_copyc C s8 | | jjjjd¡ d| jjj_| | jjjjd¡ d S )Né2 r‡ )r r rˆ Zglobal_Zfovyr r r r Ú"test_vis_global_exposed_as_global_¨ s z5MuJoCoBindingsTest.test_vis_global_exposed_as_global_c C s. t ¡ }| |jd¡ tj |jg d¢¡ d S )Nrz r{ )r
ÚMjOptionr r€ r' r( r) r )r r r r r Ú test_mjoption_can_make_default s z1MuJoCoBindingsTest.test_mjoption_can_make_defaultc C sr t ¡ }d|_d|_t |¡}| |jd¡ tj |jg d¢¡ d|_d|_| |jd¡ tj |jg d¢¡ d S )Nr} r )r r r g{®Gázt?r; ) r
r• r€ r Úcopyr r' r( r) )r Zopt1Zopt2r r r Útest_mjoption_can_copy² s
z)MuJoCoBindingsTest.test_mjoption_can_copyc C sœ t | j¡}| t |tjjd¡d¡ | t |tjjd¡d¡ | t |tjjd¡d¡ d| jjd< t j
| jjd g d¢¡ t j
|jd g d¢¡ d S ) Nr rb rŒ r rN ç à?)r™ r™ r™ )r% r% r% ) r— r r r
Z
mj_id2namer r` r rR r' r( r) )r Z
model_copyr r r Útest_mjmodel_can_copyÁ s ÿÿÿ z(MuJoCoBindingsTest.test_mjmodel_can_copyc C s dddt d¡ d t t jd ¡t t jd ¡dddddddg | j_t | j | j¡ t
| j¡}| |j d¡ | |j
d j| jj
d j¡ t | j | j¡ t | j | j¡ t | j |¡ | | jj d¡ | |j d¡ t | j |¡ t | j |¡ | |j d¡ d S )Nr r% r r} r_ r rV )r' ÚsqrtÚcosÚpiÚsinr r- r
Ú
mj_forwardr r— r ÚnconÚcontactÚdistÚ mj_resetData)r Z data_copyr r r Útest_mjdata_can_copyÔ s "þ z'MuJoCoBindingsTest.test_mjdata_can_copyc C sR | j j}| |tjj¡ tdƒ| j jd< t | j | j ¡ |
|tjj j d¡ d S )NÚNaNr r )
r Úwarningru r
Z
mjtWarningZ
mjNWARNINGÚfloatr- Z mj_checkPosr r ZmjWARN_BADQPOSÚnumber)r Úwarningsr r r Ú"test_mjdata_can_read_warning_arrayë s
z5MuJoCoBindingsTest.test_mjdata_can_read_warning_arrayc C s t | j| j¡ g }tdƒD ] }| t | jj| ¡¡ q t|dd„ d}t j
|d j d d… ddg¡ t j
|d j d d… d dg¡ t j
|d j d d… d d g¡ t j
|d
j d d… dd g¡ tdƒD ]}d | jj| j d d…< qÒt j
| jjd j d d… d d g¡ t j
| jjd j d d… d d g¡ t j
| jjd j d d… d d g¡ t j
| jjd
j d d… d d g¡ t j
|d j d d… ddg¡ t j
|d j d d… d dg¡ t j
|d j d d… d d g¡ t j
|d
j d d… dd g¡ d S ) NrV c S s t | jd | jd ¡S ©Nr r ©r' Zarctan2Úpos©Úxr r r Úú ó z.©Úkeyr r çš™™™™™¹¿r r% r3 é7 )
r
rŸ r r rf Úappendr— r¡ Úsortedr' r( Úassert_allcloser )r Z contact_copyÚir r r Útest_mjcontact_can_copyò s* ÿ $$$$ z*MuJoCoBindingsTest.test_mjcontact_can_copyc C sø d}| j jd |7 < t | j| j ¡ | jjjd }t d| | ¡}| j j }d| jj_
| | j j d¡ | j j |k rÞ| | j j
d¡ | | j j¡ | | j jjd ¡ | j j }t | j| j ¡ | | j j || jjj
¡ qlt | j| j ¡ | | j j
d¡ | | j jd¡ | | j jjd ¡ t|dd „ d
}tj |d jd d… d d g¡ tj |d jd d… d
d g¡ tj |d jd d… d
d
g¡ tj |d jd d… d d
g¡ t | j| j ¡ | | j j
d¡ | | j j¡ | | j jjd ¡ d S )Ng Ð?r ç `?r ZOWNDATArV é c S s t | jd | jd ¡S r« r¬ r® r r r r° * r± z1MuJoCoBindingsTest.test_mj_step..r² r´ r r% r3 )r r- r
rŸ r r r r' r› r¡ r€ r Útimer Ú assertEmptyZefc_typeZ
assertTrueÚflagsÚmj_stepru Z assertFalser· r( r¸ r r£ )r Z displacementr Zexpected_contact_timer¡ Ú prev_timeZsorted_contactr r r Ú test_mj_step
s<
ÿ z MuJoCoBindingsTest.test_mj_stepc C sv d| j j_| | jjd¡ tdƒD ]>}| jj}tj| j | jdd | | jj|d| j jj ¡ q"| dtjj
¡ d S )Nr» r rW r< )Znstepz Optionally, repeat nstep times.) r r r€ r r r½ rf r
rÀ rn Ú__doc__)r Ú_rÁ r r r Útest_mj_step_multiple6 s
z(MuJoCoBindingsTest.test_mj_step_multiplec C sœ | | jj¡ d}t | j| j¡ | | jj|¡ tj | jjj
ddggd ¡ g }| jjD ]"}| tj j
dd¡ |d |_q\| ||¡ tj | jjj|¡ g }| jjD ]"}| tj j
dd¡ |d |_q¬| ||¡ tj | jjj|¡ g }| jjD ]"}| tj j
dd¡ |d |_qü| ||¡ tj | jjj|¡ g }t| jjƒD ]&\}}| ||d g¡ |d |_
qP| ||¡ tj | jjj
|¡ d S ) NrV r r r3 )rQ rJ r; rc )r¾ r r¡ r
rŸ r ru r' r( r) rB r¶ ÚrandomÚuniformr ZfrictionÚHÚ enumerate)r Z
expected_nconZ expected_posr¡ Zexpected_frictionZ
expected_HZ
expected_geomr¹ r r r Útest_mj_contact_list? s: z'MuJoCoBindingsTest.test_mj_contact_listc C sâ | | jj¡ d}d}tj | j||¡ d}d}tjj| j||d | | jj|¡ | | jjj |f¡ | | jj
j |df¡ d}| tj |¡" tj | jdd¡ W d ƒ n1 s¸0 Y | | jj¡ | | jj¡ d S ) Nr[ rO rP é )r ÚnefcrV z#insufficient arena memory availablei áõ)
r¾ r r¡ r
Z
_functionsZ_realloc_con_efcru r Zefc_idr> Zefc_KBIPÚ assertRaisesWithLiteralMatchÚ
FatalError)r r rÌ Zexpected_errorr r r Útest_realloc_con_efcc s 0z'MuJoCoBindingsTest.test_realloc_con_efcc C s$ t j t¡}t |¡}t | j| j¡ | | jj d¡ t ||¡ | |j d¡ | |j
| jj
¡ t t j
d ¡t t j
d ¡ddg| jjdd…< | jjd t d¡d d d 9 < t | j| j¡ | | jj d¡ | |j
| jj
¡ | |j
| jj
¡ | | jj
| jj¡ | | jj
d
¡ d S ) NrV r_ r r3 r< r r r% çíµ ÷Æ°>r; )r
r r r r rŸ r r r r r¡ r' rœ r rž r- r› rƒ r¦ rv r r r Ú test_mj_struct_list_equalityx s
0&z/MuJoCoBindingsTest.test_mj_struct_list_equalityr• Z toleranceÚ
MjWarningStatr¨ Ú MjTimerStatÚ MjSolverStatZnevalÚ MjContactr¢ Ú MjStatisticZextentÚMjLROptZmaxforceÚ
MjvPerturbÚselectÚ MjvCameraÚ
fixedcamidc C sˆ |ƒ }|ƒ }t ||dƒ | ||¡ t ||dƒ | ||¡ | |d¡ | |¡ | t¡ t|ƒ W d ƒ n1 sz0 Y d S ©Nr r3 )Úsetattrrƒ r ZassertIsNotNoneÚ assertRaisesÚ TypeErrorÚhash)r ÚclsÚattrÚstructZstruct2r r r Útest_mj_struct_equality s
z*MuJoCoBindingsTest.test_mj_struct_equalityc C s@ t ¡ }t ¡ }d|jd< | ||¡ d|jd< | ||¡ d S rÜ )r
rÕ rÈ rƒ r )r Zcontact1Zcontact2r r r Ú test_mj_struct_equality_array© s
z0MuJoCoBindingsTest.test_mj_struct_equality_arrayc C sX |ƒ }t ||dƒ t|ƒ}| |d|j› ¡ | |› d|¡ | t|ƒt|ƒ¡ d S )Nr ú.MjWarningStatSubclassc S s | S r r r r r r ÚptrÉ s zVMuJoCoBindingsTest.test_mj_struct_repr_for_subclass..MjWarningStatSubclass.ptrN)rç Ú
__module__Ú __qualname__Úpropertyré r r r r ÚMjWarningStatSubclassÅ s rí z2)r
rÒ r rh )r rí ri r r r Ú test_mj_struct_repr_for_subclassÄ s z3MuJoCoBindingsTest.test_mj_struct_repr_for_subclassc C sœ g d¢}t t jd ¡ddt t jd ¡g}t g d¢¡t d¡ }t dt j¡}| t
|||¡¡ t j ||¡ t j
|||d t j ||¡ |
t¡( t
t dt j¡||¡ W d ƒ n1 sÊ0 Y |
t¡" t
|g d ¢|¡ W d ƒ n1 s0 Y |
t¡" t
||g d
¢¡ W d ƒ n1 sF0 Y |
t¡, t dt j¡}t
|||¡ W d ƒ n1 sŽ0 Y d S ) N©r r r r_ r )r r r r r3 )Úvecr] ÚresrV )r r r3 rV ©r r r3 )r' rœ r rž Úarrayr› ÚemptyÚfloat64Ú assertIsNoner
Zmju_rotVecQuatr( r¸ rÞ rß ÚzerosÚint32)r rð r] Úexpectedrñ r r r Útest_mju_rotVecQuatÕ s" $ 6 2 2 z&MuJoCoBindingsTest.test_mju_rotVecQuatc C s~ t | j| j¡ dt jjj }d|› d}| t j|¡ t | j|¡ W d ƒ n1 s\0 Y t jj
}t | j|¡}t |d t j
¡}d}| t|¡$ t | j| j||¡ W d ƒ n1 sÈ0 Y t |t j
¡}t | j| j||¡ t | j| j¡ t |t j
¡}t | j| j||¡ t | j| j||¡ t | j| j¡ t |t j
¡}t | j| j||¡ t j ||¡ d S )Nr z!mj_stateSize: invalid state spec z >= 2^mjNSTATEr z-state size should equal mj_stateSize(m, spec))r
rÀ r r ZmjtStateZmjNSTATEÚvaluerÍ rÎ Z mj_stateSizeZmjSTATE_INTEGRATIONr' rô rõ rß Z mj_getStateZ mj_setStater( r) ) r Z invalid_specÚexpected_messageÚspecrQ Zstate_bad_sizeZstate0Zstate1aZstate1br r r Útest_getsetstateø s,
ÿ,2z#MuJoCoBindingsTest.test_getsetstatec C s t | j| j¡ d}d|› d}| t j|¡" t | j| j|¡ W d ƒ n1 sV0 Y d}| jj}| jj ¡ }| jj
¡ }| jj ¡ }t | j| j|¡ t | j| j¡ | || jj¡ t
| j| j|¡ | || jj¡ tj || jj¡ tj || jj
¡ tj || jj ¡ d S )Nr z+mj_setKeyframe: index must be smaller than z (keyframes allocated in model)r )r
rÀ r r rÍ rÎ Zmj_setKeyframer½ r- r— ÚqvelÚactrƒ Zmj_resetDataKeyframer r' r( r) )r Z invalid_keyrü Z valid_keyr½ r- rÿ r r r r Útest_mj_setKeyframe s&
ÿ0 z&MuJoCoBindingsTest.test_mj_setKeyframec C s‚ t | jjt j¡| j_t | j| j¡ t | j| j¡ t
dt j¡}t | j| j|d¡ t j
|| jj | jjdd d …f ¡ d S )N©r3 rW r )r' Úonesr Únvrõ r rÿ r
rŸ Z
mj_subtreeVelrô Z mj_angmomMatr( Úassert_almost_equalZsubtree_angmom)r Úmatr r r Útest_mj_angmomMat; s ÿz$MuJoCoBindingsTest.test_mj_angmomMatc
C s> t | j| j¡ t | jt jjd¡}t dtj ¡}t
| j| j|d |¡ t g d¢g d¢g d¢g¡}tj
||¡ t dtj ¡}t
| j| jd ||¡ t g d¢g d¢g d¢g¡}tj
||¡ d|d d …< d|d d …< t
| j| j|||¡ tj
||¡ tj
||¡ | t¡0 t
| j| jt d|j¡d |¡ W d ƒ n1 sL0 Y | t¡0 t
| j| jd t d|j¡|¡ W d ƒ n1 s˜0 Y | t¡0 t
| j| jt |jt¡d |¡ W d ƒ n1 sä0 Y | t¡0 t
| j| jd t |jt¡|¡ W d ƒ n1 s00 Y d S ) NZmysiter )
r r r r r r rJ r r r )
r r r r r r r r r r )
r r r r r r r r r r r )r3 r\ )rV r< )r
rŸ r r r r Z
mjOBJ_SITEr' rô rõ Z
mj_jacSiteró r( r) rÞ rß Zdtyper÷ r> Úint)r Zsite_idZjacpZ
expected_jacpZjacrZ
expected_jacrr r r Útest_mj_jacSiteF sN ýý ÿ$ ÿ$ ÿ$ ÿz"MuJoCoBindingsTest.test_mj_jacSitec C s$ | tjjd¡ | tjjd¡ d S )Nz]mj_versionString() -> str
Return the current version of MuJoCo as a null-terminated string.
zvmj_Euler(m: mujoco._structs.MjModel, d: mujoco._structs.MjData) -> None
Euler integrator, semi-implicit in velocity.
)r r
Zmj_versionStringrà Zmj_Eulerr r r r Útest_docstringsƒ s þþz"MuJoCoBindingsTest.test_docstringsc C s | tjd¡ | tjd¡ d S )Ng _ BgV瞯Ò<)r r
ÚmjMAXVALZmjMINVALr r r r Útest_float_constant“ s z&MuJoCoBindingsTest.test_float_constantc C sª | tjtjj¡ | tjtjj¡ | tjtj j
¡ | tj tj j
¡ | tjtjj¡ | tjtjj¡ | tjtjj¡ | tjd d¡ | tjtjj d¡ d S )NrO ZRefsafe)ZInertiaÚ0ÚI)ru r
ZmjDISABLESTRINGÚ
mjtDisableBitZ
mjNDISABLEZmjENABLESTRINGÚ mjtEnableBitÚ mjNENABLEZ
mjTIMERSTRINGZmjtTimerZmjNTIMERZ
mjLABELSTRINGZmjtLabelZmjNLABELZ
mjFRAMESTRINGÚmjtFrameZmjNFRAMEZ mjVISSTRINGZ
mjtVisFlagZ
mjNVISFLAGZ mjRNDSTRINGZ
mjtRndFlagZ
mjNRNDFLAGr Z
mjVIS_INERTIAr r r r Útest_string_constants— s þz(MuJoCoBindingsTest.test_string_constantsc C s | tjjd¡ | tjjd¡ | tjjd¡ | tjjd¡ | tjjd¡ | tjj d¡ | tjj
d¡ | tjj d¡ | tj j
d¡ | tj jd¡ | tj jd¡ | tj jd¡ | tj jd¡ | tj jd ¡ | tj jd
¡ | tj jd ¡ d S ) Nr r r r3 rV r< r‡ ée éf él ié )r r
ÚmjtJointÚ
mjJNT_FREEZ
mjJNT_BALLZ mjJNT_SLIDEZ mjJNT_HINGEr ZmjENBL_OVERRIDEZ
mjENBL_ENERGYZ
mjENBL_FWDINVr ÚmjtGeomZ mjGEOM_PLANEZ
mjGEOM_HFIELDZ
mjGEOM_SPHEREÚ mjGEOM_ARROWZ
mjGEOM_ARROW1Z
mjGEOM_ARROW2ZmjGEOM_TRIANGLEZ mjGEOM_NONEr r r r Útest_enum_values¥ s z#MuJoCoBindingsTest.test_enum_valuesc C sš | tjjt d¡¡ | tjjtjdd¡ | t¡ t d¡ W d ƒ n1 sX0 Y | t¡ t d¡ W d ƒ n1 sŒ0 Y d S )Nr r‡ )rû iè rJ )r r
r r r r rÞ r" r r r r Útest_enum_from_int· s ( z%MuJoCoBindingsTest.test_enum_from_intc C s@ g d¢}| |tjj d¡ | |tjjtjj… ddg¡ d S )N) ÚaÚbÚcÚdÚeÚfÚgÚhr¹ ÚjÚkr$ r r )r r
r Ú
mjFRAME_WORLDZ mjFRAME_GEOMZmjFRAME_CAMERA)r r¯ r r r Útest_enum_as_index¿ s þz%MuJoCoBindingsTest.test_enum_as_indexc C sÈ | tjjd¡ | tjjd¡ | dtjj¡ | dtjj¡ | tjjtjj¡ | tjjtjj¡ | tjj d¡ | tjj t¡ | tjjd d¡ | tjjd t¡ | dtjj d¡ | dtjj t¡ | tjjd d ¡ | d
tjj d ¡ | tjjd d ¡ | tjjd t¡ | dtjj d¡ | dtjj t¡ | tjjd
d¡ | dtjj d¡ | tjjd d¡ | tjjd t¡ | dtjj d¡ | dtjj t¡ | tjjd d¡ | dtjj d¡ | tjjd d¡ | dtjj d¡ | tjjd d¡ | tjjd t¡ | tjj d d¡ | tjj d t¡ | dtjj d¡ | dtjj t¡ | dtjj d¡ | dtjj t¡ | tjjd d¡ | tjjd t¡ | tjj d d¡ | tjj d t¡ | d tjj d¡ | d tjj t¡ | dtjj d¡ | d tjj t¡ | tjjd d¡ | tjjd t¡ | tjj d d¡ | tjj d t¡ | tjj d d¡ | tjj d t¡ | tjjd d ¡ | tjjd t¡ | dtjj d¡ | dtjj t¡ | d tjj d!¡ | d tjj t¡ | d tjj d ¡ | d tjj t¡ | dtjj d"¡ | dtjj t¡ | t
¡ tjjd# }W d ƒ n1 sâ0 Y | t
¡ dtjj }W d ƒ n1 s0 Y | t
¡ tjjd# }W d ƒ n1 sR0 Y | t
¡ dtjj }W d ƒ n1 sŠ0 Y | t
¡ tjjd# }W d ƒ n1 sÂ0 Y | t
¡ dtjj }W d ƒ n1 sú0 Y | tj j tj j
B d$¡ | tj jd%B d%¡ | tj jd%@ d&¡ | tj jd%A d¡ | d%tj jB d%¡ | d%tj j@ d&¡ | d%tj jA d¡ | tj jd> tj j¡ | tj jd? tj j¡ d S )'Nr< ç @iùÿÿÿr r_ r r[ g ü?g €!@g @g €#@r; g @g @g € @g ô?r3 é g @g €8@g €1@g @éüÿÿÿrY iìÿÿÿéýÿÿÿç @g 4@g 4ÀrV rJ i÷ÿÿÿéþÿÿÿéûÿÿÿr éH é! é )r r
r r' rƒ Z mjFRAME_NONEÚassertIsInstancer r§ rÞ ÚZeroDivisionErrorr ZmjDSBL_GRAVITYZ mjDSBL_LIMITZmjDSBL_PASSIVEZmjDSBL_CLAMPCTRLZmjDSBL_WARMSTARTZmjDSBL_CONTACT)r rÄ r r r Ú
test_enum_opsÇ s¼ ÿ ÿ , , , , , ,þ
þ
þz MuJoCoBindingsTest.test_enum_opsc C sN | j j| j _| tjd¡ t | j| j ¡ W d ƒ n1 s@0 Y d S )Nz.\Amj_stackAlloc: out of memory, stack overflow)r ZnarenaZpstackr! r
rÎ rŸ r r r r r Útest_can_raise_error1 s
ÿz'MuJoCoBindingsTest.test_can_raise_errorc C sŽ G dd„ dƒ}|ƒ }t tj|ƒL | t ¡ |¡ | |jd¡ t | j| j ¡ |
|jd¡ W d ƒ n1 sr0 Y | t ¡ ¡ d S )Nc @ s e Zd Zdd„ Zdd„ ZdS )z6MuJoCoBindingsTest.test_mjcb_time..CallCounterc S s
d| _ d S ©Nr ©Úcountr r r r Ú__init__< s z?MuJoCoBindingsTest.test_mjcb_time..CallCounter.__init__c S s | j d7 _ | j d S )Nr r8 r r r r Ú__call__? s z?MuJoCoBindingsTest.test_mjcb_time..CallCounter.__call__N©rç rê rë r: r; r r r r Ú CallCounter: s r= r ) r r
Ú
set_mjcb_timer= Z
get_mjcb_timer r9 rŸ r r rs rö )r r= Z call_counterr r r Útest_mjcb_time8 s ,z!MuJoCoBindingsTest.test_mjcb_timec s² G dd„ dt ƒ‰ ‡ fdd„}ttj|ƒf | ˆ ¡ }t | j| j¡ W d ƒ n1 sZ0 Y | |j j
ddddd œf¡ W d ƒ n1 s”0 Y t | j| j¡ d S )
Nc @ s e Zd ZdS )z>MuJoCoBindingsTest.test_mjcb_time_exception..TestErrorN)rç rê rë r r r r Ú TestErrorQ s r@ c s ˆ dddddœƒ‚d S )NÚstringrò r r ©r r r r ©r@ r r Úraises_exceptionT s zEMuJoCoBindingsTest.test_mjcb_time_exception..raises_exceptionrA rò r r rB ) Ú RuntimeErrorr r
r> rÞ rŸ r r r Ú exceptionÚargs)r rD r! r rC r Útest_mjcb_time_exceptionO s .ÿ"z+MuJoCoBindingsTest.test_mjcb_time_exceptionc C sp t tjdd„ ƒL | td¡ t | j| j¡ W d ƒ n1 sD0 Y W d ƒ n1 sb0 Y d S )Nc S s dS )NrA r r r r r r° b r± zEMuJoCoBindingsTest.test_mjcb_time_wrong_return_type..z*mjcb_time callback did not return a number)r r
r> rÍ rß rŸ r r r r r r Ú test_mjcb_time_wrong_return_typea s
ÿz3MuJoCoBindingsTest.test_mjcb_time_wrong_return_typec C s: | td¡ t d¡ W d ƒ n1 s,0 Y d S )Nz%callback is not an Optional[Callable]r )rÍ rß r
r> r r r r Útest_mjcb_time_not_callableh s ÿz.MuJoCoBindingsTest.test_mjcb_time_not_callablec C sš G dd„ dƒ}t j t¡}t |¡}|| ||ƒ}| |jd¡ tt j|ƒ t ||¡ W d ƒ n1 sl0 Y | |jd¡ | |j
d d¡ d S )Nc @ s e Zd Zdd„ Zdd„ ZdS )z;MuJoCoBindingsTest.test_mjcb_sensor..SensorCallbackc S s || _ || _|| _d| _d S r7 )r‹ Úexpected_modelÚ
expected_datar9 )r r‹ rK rL r r r r: r s zDMuJoCoBindingsTest.test_mjcb_sensor..SensorCallback.__init__c S sN | j || j¡ | j || j¡ | j |tjj¡ d|jd< | j d7 _ d S )NrË r r )
r‹ r= rK rL r r
ZmjtStageZ mjSTAGE_VELÚ
sensordatar9 )r Úmr Zstager r r r; x s
zDMuJoCoBindingsTest.test_mjcb_sensor..SensorCallback.__call__Nr< r r r r ÚSensorCallbackp s rO r r rË ) r
r r ÚTEST_XML_SENSORr r r9 r Zset_mjcb_sensorrŸ rM )r rO Zmodel_with_sensorZdata_with_sensorZsensor_callbackr r r Útest_mjcb_sensorn s
*z#MuJoCoBindingsTest.test_mjcb_sensorc C s¸ g }g }t dƒD ]V}t d ¡ | tj d¡¡ | t |d ¡¡ t dd„ ¡ t |d |d ¡ qt d ¡ |r”| ¡ }| t
|¡d¡ ~qr|r´| ¡ }| t
|¡d¡ q”d S )NrW r rJ c S s d S r r )rN r r r r r° – r± zFMuJoCoBindingsTest.test_mjcb_control_not_leak_memory..r ) rf r
Zset_mjcb_controlr¶ r r r rÀ Úpopr r6 r7 )r Zmodel_instancesZdata_instancesrÄ r rN r r r Ú!test_mjcb_control_not_leak_memoryŒ s
z4MuJoCoBindingsTest.test_mjcb_control_not_leak_memoryc C sD t t j d¡¡}|j}| t |j¡d¡ ~| t |¡d¡ d S )Nr r3 r )r
r r r r r r6 r7 )r r r r r r Ú test_mjdata_holds_ref_to_model£ s
z1MuJoCoBindingsTest.test_mjdata_holds_ref_to_modelc C s¬ | t ¡ tj¡ | t ¡ tj¡ | t ¡ tj¡ | t ¡ tj¡ | t ¡ tj¡ | t ¡ tj¡ | t ¡ tj¡ | t | jd¡tj¡ | t ¡ tj ¡ d S )Nr‡ )
r3 r
ÚMjvScenerÚ Z MjvGLCameraZMjvGeomZMjvLightÚ MjvOptionr Z MjvFigurer r r r Ú test_can_initialize_mjv_structs¯ s z2MuJoCoBindingsTest.test_can_initialize_mjv_structsc C sZ t ¡ }t jj|_d|_| |jd¡ | t¡ d|_W d ƒ n1 sL0 Y d S )Niÿÿÿr™ ) r
rÚ Z mjtCameraZmjCAMERA_TRACKINGÚtyperÛ r rÞ rß )r Úcamerar r r Útest_mjv_cameraº s
z"MuJoCoBindingsTest.test_mjv_camerac C s„ t j| jdd}| |jd¡ | |jd¡ | |j|j¡ t | j| j ¡ t
| j| j t ¡ d t ¡ t j
j|¡ | |jd¡ d S )Nr‡ )r Úmaxgeomr )r
rU r r Zngeomr[ ru ZgeomsrŸ r Zmjv_updateScenerV rÚ Z mjtCatBitZ mjCAT_ALLrs ©r Zscener r r Útest_mjv_sceneà s ù z!MuJoCoBindingsTest.test_mjv_scenec C s( t ¡ }| |jd¡ | |jd¡ d S )Nç ð?r )r
rU r Úscaler[ r\ r r r Ú test_mjv_scene_without_model× s z/MuJoCoBindingsTest.test_mjv_scene_without_modelc
C s” t dt j¡}t | j| j¡ t | j| jg d¢g d¢d dd|¡ t | j| jg d¢g d¢g d¢dd|¡ tj| j| jg d¢g d¢d dd|d d S )Nr r$ ©r r r r )r r r r r r )rN r Úpntrð Ú geomgroupÚ
flg_staticÚ bodyexcludeÚgeomid)r' r÷ rø r
rŸ r r Úmj_ray)r rf r r r Ú test_mj_rayÜ s2 ÿø øz MuJoCoBindingsTest.test_mj_rayc
C s* d}t dt j¡}t g d¢¡}t g d¢g d¢g d¢gt j¡}t g d¢¡}t g d¢¡}t |t j¡}t |t j¡}t | j| j¡ tj | j| j||
¡ d dd |||tj d
t d dƒD ]h} |
|| t | j| j||| d dd |¡¡ |
|| |¡ |
|| || ¡ | || || ¡ q¼d S ) Nr3 r )g333333Ó¿r r% rï ra )r r rJ )r~ rJ r% )r rJ r rJ ) rN r rb rð rc rd re rf r¢ ÚnrayÚcutoffr )r' r÷ rø ró rõ r
rŸ r r Z mj_multiRayÚflattenr rf r rg ZassertAlmostEqual)
r ri Zgeom1rb rð Zdist_exZgeom_exrf r¢ r¹ r r r Útest_mj_multi_rayù s< õ þz$MuJoCoBindingsTest.test_mj_multi_rayc
C s. d}d}t | j| j||d d d d d d d ¡ d S ©NrÐ r )r
Ú
mjd_inverseFDr r )r ÚepsÚ flg_centeredr r r Útest_inverse_fd_none s õz'MuJoCoBindingsTest.test_inverse_fd_nonec C sZ t | j| j¡ t dtj¡}t | j| jddd|¡}| |d¡ tj
|t d¡¡ d S )Nr\ r r éÈ ç33333óD@)r& ç rt r& rt rs ) r
rŸ r r r' rô rõ Zmj_geomDistancer r( r) ró )r Zfromtor¢ r r r Útest_geom_distance, s
ÿz%MuJoCoBindingsTest.test_geom_distancec
C s@ d}d}t | jj| jjf¡}t | jj| jjf¡}t | jj| jjf¡}t | jj| jjf¡}t | jj| jjf¡}t | jj| jjf¡}t | jj| jjf¡} t | j| j||||||||| ¡ | t j
|¡|¡ | t j
|¡|¡ | t j
|¡|¡ | t j
|¡|¡ | t j
|¡|¡ | t j
|¡|¡ d S rm ) r' r÷ r r Z nsensordataZnMr
rn r rs ÚlinalgZnorm)
r ro rp Zdf_dqZdf_dvZdf_daZds_dqZds_dvZds_daZdm_dqr r r Útest_inverse_fd5 s8 õ
z"MuJoCoBindingsTest.test_inverse_fdc C s¤ t d¡}t d¡}t dt j¡}t dt j¡}t ||||¡ t dt j¡}t dt j¡}t ||d |¡ t |||d ¡ t j || ¡ ¡ t j || ¡ ¡ d S )N)r~ ç333333Ó?rx çš™™™™™Ù?)r% r~ ry r™ r[ )r3 r3 ) r' ró rô rõ r
Z mjd_subQuatr( r) rk )r Zquat1Zquat2Zd1Zd2Zd3Zd4r r r Útest_mjd_sub_quatS s
z$MuJoCoBindingsTest.test_mjd_sub_quatc C sN d}t d¡}t dt j¡}t dt j¡}t dt j¡}t |||||¡ d S )Nr% )r~ rx rx r[ r3 )r' ró rô rõ r
Zmjd_quatIntegrate)r r_ ZvelZd_quatZd_velZd_hr r r Útest_mjd_quat_intergrate` s
z+MuJoCoBindingsTest.test_mjd_quat_intergratec
C s^ d}d}d}t g d¢g d¢g d¢g d¢g¡}t ||| || ¡}t |||||¡ tdƒD ](}t ||||¡}| || |d ¡ qbt ||f¡}d} t |||||| ¡ t j
||¡ t dgdgdgd gg¡}
t |
¡} d} t | ||
|||| | ¡ t j
| ||
¡ d
}
d
}t
|||||
|¡ t | ||
|||¡ t j | t j ||
¡¡ d S ) NrV r )r^ r r r% )r r- r r~ )r r ç @rx )r% r~ rx ç @r- r| r} r )r' ró r÷ r
Zmju_dense2Bandrf Z mju_bandDiagr Zmju_band2Denser( r) Z
zeros_likeZmju_bandMulMatVecZmju_cholFactorBandZmju_cholSolveBandr rv Zsolve)r Zn_totalZn_bandZn_denseZdenseZbandr¹ ÚindexZdense2Zflg_symrð rñ Zn_vecZdiag_addZdiag_mulr r r Ú test_bandedh s@ ü
ÿ ÿz MuJoCoBindingsTest.test_bandedc
C s„ d}t |¡}t ||d f¡}t |t j¡}t |¡}t |f¡}t |f¡ }t |f¡}t |||||||¡} | | d¡ d S )Nr; r< rJ )r' r÷ rø Úeyer r
Z mju_boxQPrs )
r Únrñ Úrr~ r$ r# ÚlowerÚupperZrankr r r Útest_mju_box_qpŠ s
z"MuJoCoBindingsTest.test_mju_box_qpc C s4 t dt j¡}t |d¡ t j |t dd¡¡ d S )Nr3 ç ø?)r' rô rõ r
Zmju_fillr( r) Úfull)r rñ r r r Ú
test_mju_fill– s z MuJoCoBindingsTest.test_mju_fillc C s0 t dt j¡}t |¡ t j |t d¡¡ d S )N©rV rV rV )r' rô rõ r
Zmju_eyer( r) r€ )r Zeye4r r r Ú test_mju_eye› s
z MuJoCoBindingsTest.test_mju_eyec C sL t ddd¡ dd¡}t dt j¡}t ||¡ t j |d||j ¡ d S )Nr r r¼ rV r‰ r™ )
r' Zlinspacere rô rõ r
Zmju_symmetrizer( r) ÚT)r r rñ r r r Útest_mju_symmetrize s z&MuJoCoBindingsTest.test_mju_symmetrizec C sF | t ddd¡d¡ | t ddd¡d¡ | t ddd¡d¡ d S )Nr† r^ r- r| rt )r r
Zmju_clipr r r r Ú
test_mju_clip¦ s z MuJoCoBindingsTest.test_mju_clipc C sR t g d¢¡}t g d¢¡}t g d¢g d¢g d¢g¡}| t |||¡d¡ d S )N©r^ r- r| )r| r- r^ )r} g @g @)r) g @g "@g €i@)r' ró r r
Zmju_mulVecMatVec)r Zvec1Zvec2r r r r Útest_mju_mul_vec_mat_vec« s z+MuJoCoBindingsTest.test_mju_mul_vec_mat_vecc C sÞ t g d¢g d¢g¡}t g d¢¡}t ddg¡}t ddg¡}t g d¢¡}t d¡}t dt j¡}t dt j¡}t dt j¡} t ||||| ¡}
t j |
d¡ t j ||¡ t j ||¡ t j ||¡ t j | |¡ d S ) N©rt r^ rt ©r- rt r| rŽ r r r ©r r r r3 ) r' ró r÷ rø r
Zmju_dense2sparser( Z assert_equalr) ) r r Z
expected_valsZexpected_rownnzZexpected_rowadrZexpected_colindÚvalsZrow_nnzZrow_adrZcol_indÚstatusr r r Útest_mju_dense_to_sparse± s
z+MuJoCoBindingsTest.test_mju_dense_to_sparsec C sx t g d¢g d¢g¡}t d¡}t ddg¡}t ddg¡}t g d¢¡}t d¡}t |||||¡ t j ||¡ d S ) Nr r‘ rŽ r r r r’ )r r3 )r' ró r÷ r
Zmju_sparse2denser( r) )r rù r ZrownnzZrowadrZcolindrñ r r r Útest_mju_sparse_to_dense s
z+MuJoCoBindingsTest.test_mju_sparse_to_densec C s t d¡}t dt jd dg¡}d}t |||¡ t t d¡dt d¡dg¡}t j ||¡ d}| tj
|¡ t ||d¡ W d ƒ n1 s”0 Y | tj
|¡ t ||d ¡ W d ƒ n1 sÐ0 Y d
}| tj
|¡ t ||d ¡ W d ƒ n1 s0 Y d S ) NrV r r Zxyzr™ rt z5mju_euler2Quat: seq must contain exactly 3 charactersZxyZxyzyz@mju_euler2Quat: seq[2] is 'p', should be one of x, y, z, X, Y, ZZxYp) r' r÷ ró r r
Zmju_euler2Quatr› r( r rÍ rÎ )r r] ZeulerÚseqZ
expected_quatÚerrorr r r Útest_mju_euler_to_quatÌ s
,,z)MuJoCoBindingsTest.test_mju_euler_to_quat)FT)Úflg_htmlÚflg_padc C s | dt ||¡¡ d S )Nr
)rn r
Zmj_printSchema)r rš r› r r r Útest_mj_printSchemaÝ s z&MuJoCoBindingsTest.test_mj_printSchemac C s* t t | j¡¡}d}| || j|¡ d S )N)r½ r- rÿ Ú mocap_pos)ÚpickleÚloadsÚdumpsr Ú_assert_attributes_equal)r rx Úattr_to_comparer r r Útest_pickle_mjdata_cleanã s z+MuJoCoBindingsTest.test_pickle_mjdata_cleanc C sv t | j| j¡ t t | j¡¡}d}| || j|¡ tdƒD ]"}t | j| j¡ t | j|¡ q>| || j|¡ d S ©N)
r½ r- rÿ ZqaccÚxposr r¦ Zenergyr¡ Zefc_JrW ) r
rÀ r r rž rŸ r r¡ rf ©r rx r¢ rÄ r r r Útest_pickle_mjdataè s z%MuJoCoBindingsTest.test_pickle_mjdatac C s„ t jj| jj_t | j| j¡ t t
| j¡¡}d}| || j|¡ t dƒD ]"}t | j| j¡ t | j|¡ qL| || j|¡ d S r¤ )
r
Z mjtJacobianZ mjJAC_SPARSEr r ZjacobianrÀ r rž rŸ r r¡ rf r¦ r r r Útest_pickle_mjdata_sparseý s z,MuJoCoBindingsTest.test_pickle_mjdata_sparsec C s* t t | j¡¡}d}| || j|¡ d S )N)ZnqZnmatr* rŽ )rž rŸ r r r¡ )r rw r¢ r r r Útest_pickle_mjmodel s z&MuJoCoBindingsTest.test_pickle_mjmodelc C s2 d}t j |¡}| | d¡jd¡ | | d¡jd¡ | | d¡jd¡ | | d¡jd¡ | | d¡jd¡ | | d¡jd¡ | | d ¡jd¡ | | d¡jd d¡ | | d¡jd d¡ | | d¡jd d¡ | | d¡jd d ¡ | | d ¡jd d
¡ t |¡}t ||¡ | | d¡jd¡ | | d¡jd¡ | | d¡jd¡ | | d¡jd¡ | | d¡jd¡ | | d¡jd¡ | | d ¡jd¡ | | d¡j
d d¡ | | d¡j
d d¡ | | d¡j
d d¡ | | d¡j
d d ¡ | | d ¡j
d d
¡ d S ) Nz÷
Zmygeomr Z myothergeomr3 r Ú r rV r; ) r
r r r rB ÚidÚnamerQ r rŸ r¥ )r Úxmlr r r r r Útest_indexer_name_id s8
z'MuJoCoBindingsTest.test_indexer_name_idc C s& t j t¡}t |¡}t ||¡ d S r )r
r r ÚTEST_XML_PLUGINr rŸ )r r r r r r Útest_load_pluginG s
z#MuJoCoBindingsTest.test_load_pluginc C sx t j t¡}t |¡}| |j|¡ t ||¡ t |¡}t ||¡ t ||¡ t j
|j |j ¡ | |j|j¡ d S r )
r
r r r¯ r r= r rÀ r— r' r( r) r- ©r r Zdata1rx r r r Ú test_copy_mjdata_with_pluginL s
z/MuJoCoBindingsTest.test_copy_mjdata_with_pluginc C sŒ t j t¡}t |¡}| |j|¡ t ||¡ t |¡}t ||¡ t ||¡ t
j |j
|j
¡ | |j|j¡ | |jj|jj¡ d S r )r
r r r¯ r r= r rÀ r— Údeepcopyr' r( r) r- Z assertIsNotrƒ rt r± r r r Ú test_deepcopy_mjdata_with_pluginW s
z3MuJoCoBindingsTest.test_deepcopy_mjdata_with_pluginc C s& t j t¡}| | d¡jjd¡ d S )NÚtex)é r¶ r3 )r
r r ÚTEST_XML_TEXTUREr rµ r r> )r r r r r Útest_texture_sizec s z$MuJoCoBindingsTest.test_texture_sizec C sŠ |D ]€}t ||ƒ}t ||ƒ}z,t|tjƒr:tj ||¡ n | ||¡ W q ty‚ } z"| d |t
|ƒ¡¡ W Y d }~qd }~0 0 qd S )Nz.Attribute '{}' differs from expected value: {}) ÚgetattrÚ
isinstancer' Zndarrayr( r) r ÚAssertionErrorZfailÚformatr )r Z
actual_objZ expected_objr¢ r¬ Z actual_valueZexpected_valuer! r r r r¡ g s
ÿÿz+MuJoCoBindingsTest._assert_attributes_equal)urç rê rë r r r# r+ r/ r0 r ZskipIfr6 Úimplementationr¬ r: r? rC rF rG rH rL rM rT r^ ra rg rj rq rr ry r‚ r… r‰ r‘ r’ r” r– r˜ rš r¤ rª rº r rÅ rÊ rÏ rÑ r Znamed_parametersr
r• rÒ rÓ rÔ rÕ rÖ r× rØ rÚ rä rå rè rî rú rþ r r r r
r r r r r( r5 r6 r? rH rI rJ rQ rS rT rW rZ r] r` rh rl rq ru rw rz r{ r r… rˆ rŠ rŒ r r r• r– r™ Úproductrœ r£ r§ r¨ r© r® r° r² r´ r¸ r¡ Ú
__classcell__r r r r r
u sþ ÿ
) $
÷
÷
#% =j ÿ
ÿ
"
"
* r
Ú__main__)rà Ú
contextlibr— rž r6 Z absl.testingr r r
Únumpyr' r rP r¯ r· Úcontextmanagerr ZTestCaser
rç Úmainr r r r Ú s: &