
    h                         S SK r S SKrS SKJrJr  S SKJrJrJr  S SK	J
r
  S SKJr  S SKJr  Sr\SS \S	-
  4S
 j5       r\R"                  " SS9SS j5       rS r\" \" S \S	-
  S9S9\" SS9S 5       5       rg)    N)givensettings)	compositeintegerslists)assert_allclose)Embed)uniqued
         c           	          U " [        [        X#S95      5      nU[        U5      U-  S  n[        R                  " USS9nUR                  SU45      $ )N)	min_value	max_valueuint64)dtype)r   r   lennumpyarrayreshape)drawcolumnslohiint_listr   s         Y/home/james-whalen/.local/lib/python3.13/site-packages/thinc/tests/layers/test_uniqued.pylists_of_integersr      sS    
 E(R>?@H H/12HKK1E=="g''    module)scopec                 <    [        U [        SS9R                  5       $ )Nr   column)r	   ROWS
initialize)nOs    r   modelr(   !   s    T!$//11r   c                     ^ / m[        SSSS9n U4S jU l        U R                  5         TS/:X  d   e[        U 5      nUR                  5         TSS/:X  d   eg )N   r   r#   c                  &   > TR                  S5      $ )NT)append)argskwargscallss     r   <lambda>)test_uniqued_calls_init.<locals>.<lambda>)   s    d);r   T)r	   initr&   r
   )embeduembedr/   s     @r   test_uniqued_calls_initr5   &   s^    E!Qq!E;EJ	TF??U^F
T4L   r   )r   r   )X)deadlinec                     [        X R                  S   S9R                  5       nU " USS9u  p4U" USS9u  pV[        X55        U" U5      nU" U5      n[        Xx5        UR                  (       a  g g )Nr$   r#   T)is_train)r
   attrsr&   r   size)	r(   r6   umodelYbp_YYubp_YudXdXus	            r   !test_uniqued_doesnt_change_resultrC   1   sm     U;;x#89DDFFA%GAq4(IBA	aB
)CBvv r   )   )r   pytest
hypothesisr   r   hypothesis.strategiesr   r   r   numpy.testingr   thinc.layersr	   thinc.layers.uniquedr
   r%   r   fixturer(   r5   rC    r   r   <module>rM      s      & < < )  (	 $%!q ( ( h2  2! 	aD1H	-.	4	  /	r   