a
i‚jK ã @ sN d Z ddlZddlZddlmZ ddlZG dd„ dejƒZedkrJe ¡ dS )zCTest that copying mujoco.MjData multiple times doesn't leak memory.é N)Úabsltestc @ s$ e Zd Zdd„ Zeedœdd„ZdS )ÚMemoryLeakTestc C s| t d¡}| d¡}zVtj |¡}t |¡}d}tdƒD ]$}t ||¡ t
|¡}||j 7 }q8W | |¡ n | |¡ 0 d S )Naª
l r iè ) ÚtextwrapÚdedentÚ
_memory_limitÚmujocoZMjModelZfrom_xml_stringZMjDataÚrangeZmj_stepÚcopyÚdeepcopyÚtime)ÚselfZ model_xmlÚlimitÚmodelÚdataÚtotalÚ_Z last_data© r úW/home/ghrups/robot-bench/.venv/lib64/python3.9/site-packages/mujoco/memory_leak_test.pyÚtest_deepcopy_mjdata_leak s
z(MemoryLeakTest.test_deepcopy_mjdata_leak)Úlimit_in_bytesÚreturnc C sN d}z.ddl }| |j¡\}}| |j||f¡ W n ttfyH Y n0 |S )z4Limits max memory usage, and returns previous limit.éÿÿÿÿr N)ÚresourceZ getrlimitZ RLIMIT_ASZ setrlimitÚ ImportErrorÚ
ValueError)r r Zsoftr Úhardr r r r 7 s z MemoryLeakTest._memory_limitN)Ú__name__Ú
__module__Ú __qualname__r Úintr r r r r r s r Ú__main__)
Ú__doc__r r Z absl.testingr r ZTestCaser r Úmainr r r r Ú s -