
    ir                         d dl Z d dlZd dlmZmZmZ ej                  d        Zej                  d        Zd Z	d Z
d Zd Zd	 Zy)
    N)
premap_ids	remap_idsremap_ids_v2c                  0    t        j                  g d      S )N)	                  	         )numpyarray     p/var/www/vps2.regionflexible.com/Desarrollo/venv/lib/python3.12/site-packages/thinc/tests/layers/test_mappers.pykeysr      s    ;;344r   c                 p    t        |       D ci c]  \  }}t        |      t        |       c}}S c c}}w )N)	enumerateint)r   vks      r   mapperr      s*    '07tq!CFCFN777s   2c           
          t        |d      } || d      \  }}t        j                  j                  |j	                         t        j
                  t        t        |                          y Nc   )defaultF)r   r   testingassert_equalsqueezeasarrayrangelen)r   r   premapvalues_s        r   test_premapr)      sK    +FtU#IFA	MMv~~/uSY?O1PQr   c           
          t        |d      } || d      \  }}t        j                  j                  |j	                         t        j
                  t        t        |                          y r   )r   r   r    r!   r"   r#   r$   r%   r   r   remapr'   r(   s        r   
test_remapr-      sK    fb)EdE"IFA	MMv~~/uSY?O1PQr   c           
          t        |d      } || d      \  }}t        j                  j                  |j	                         t        j
                  t        t        |                          y r   )r   r   r    r!   r"   r#   r$   r%   r+   s        r   test_remap_v2r/      sK    ,EdE"IFA	MMv~~/uSY?O1PQr   c                    t        |d      }t        |d      }t        |d      } || d      \  }} || d      \  }} || d      \  }}t        j                  j                  ||       t        j                  j                  ||       y r   )r   r   r   r   r    r!   )	r   r   r,   remap_v2r&   values1r(   values2values3s	            r   test_remap_premap_eqr5   #   s~    fb)EFB/H+FtU#JGQ$&JGQe$JGQ	MMw0	MMw0r   c           
         t        j                  t        |       dfd      }| |d d df<   t        |d      }t	        |d      }t         j
                  j                   ||d      d   j                         t        j                  t        t        |                          t         j
                  j                   ||d      d   j                         t        j                  t        t        |                          y )Nr   r   )dtyper   )columnFr   )
r   zerosr%   r   r   r    r!   r"   r#   r$   )r   r   idxr1   r&   s        r   test_columnr;   .   s    
++s4y!nE
2CC1IF1-Hq)F	MMeQ'')5==s4y9I+J 
MMsE1%%'uSY7G)Hr   )r   pytestthinc.layersr   r   r   fixturer   r   r)   r-   r/   r5   r;   r   r   r   <module>r?      s]      < < 5 5 8 8RRR1
r   