o oiD@sddlZdddZdS)Nư>cCst|tr+t|tr+t|t|ksJ|D] }t|||||qdSt|trQt|trQt|t|ks?Jt||D] \}}t|||qDdSt|tj rit|tj ritj |||dsgJdS||ksoJdS)N)atolT) isinstancedictsetkeysassert_dict_equallistlenziptorchTensorallclose)ab precisionkv1v2rI/home/ubuntu/SoloSpeech/.venv/lib/python3.10/site-packages/tests/utils.pyrs   r)r)r rrrrrs