
    h"                     &    S SK Jr  S SKJrJr  S rg)    )	MagicMock)Linear
with_debugc                  @   [        5       n [        5       n[        5       n[        [        5       XUS9nU R                  5         UR                  5         UR                  5         UR                  R                  SS5      nUR                  R                  SS5      nUR                  XES9  U R                  X4U5        UR                  5         UR                  5         U" USS9u  pgUR                  X4S5        UR                  5         U" U5        UR                  U5        g )N)on_init
on_forwardon_backprop   )XYT)is_train)r   r   r   assert_not_calledopsalloc2f
initializeassert_called_once_with)r   r   r	   modelr   r   Yhbackprops           \/home/james-whalen/.local/lib/python3.13/site-packages/thinc/tests/layers/test_with_debug.pytest_with_debugr      s    kGJ+K'kE   "!!#		!QA		!QA	q##Ea0  "!!#T*LB&&u6!!#QK''*    N)mockr   	thinc.apir   r   r    r   r   <module>r      s     (+r   