o i"@s(ddlmZddlmZmZddZdS)) MagicMock)Linear with_debugcCst}t}t}tt|||d}||||jdd}|jdd}|j||d||||||||dd\}}|||d|||||dS)N)on_init on_forward on_backprop)XYT)is_train)rrrassert_not_calledopsalloc2f initializeassert_called_once_with)rrrmodelr r YhbackproprV/home/ubuntu/.local/lib/python3.10/site-packages/thinc/tests/layers/test_with_debug.pytest_with_debugs( rN)mockr thinc.apirrrrrrrs