
    wi	
                     >    d dl Z d dlmZ d dlmZ d Zd Zd Zd ZdS )    N)msgpackExtTypec                  .   d }  | d          dk    sJ  | d          dk    sJ  | d          dk    sJ  | d          d	k    sJ  | d
          dk    sJ  | d          dk    sJ  | d          dk    sJ  | ddz            dddz  z   k    sJ d S )Nc                 |    t          j                    }|                    d|            |                                S )NB   )r   Packerpack_ext_typebytes)spackers     r/root/.openclaw/workspace/chatterbox_venv_py311/lib/python3.11/site-packages/srsly/tests/msgpack/test_extension.pypztest_pack_ext_type.<locals>.p   s3    !!T1%%%||~~       A   BA   AB   BAB   ABCD   BABCD   ABCDEFGH
   BABCDEFGH   AAAAAAAAAAAAAAAA   BAAAAAAAAAAAAAAAA   ABC   BABC#  AAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAA'  #BAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAE#     #EB )r   s    r   test_pack_ext_typer"      s      
 1T77l""""1U88}$$$$1W::((((1[>>000001Y<<222221V99*****1]BBBBB	$
 ;dZ>O OOOOOOOr   c                     d }  | dt          dd                      | dt          dd                      | dt          dd                      | d	t          dd
                      | dt          dd                      | dt          dd                      | dt          dd                      | dddz  z   t          dddz                       d S )Nc                 :    t          j        |           |k    sJ d S )N)r   unpackb)bexpecteds     r   checkz#test_unpack_ext_type.<locals>.check   s#    q!!X------r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r    r   r   )r(   s    r   test_unpack_ext_typer)      s   . . . 
E,d++,,,	E-u--...	E/7411222	E
wt[99:::	E
!74#;#;<<<	E
gdF33444	E
-wt]/K/KLLL	E#dZ&77dZ'((    r   c                      d } d }ddt          j         dg d          g}t          j        ||           }t          j        ||          }||k    sJ d S )	Nc                     t          d|            t          | t          j                  r&d}|                                 }t	          ||          S t          d|           )Nzdefault called{   zUnknown type object )print
isinstancearraytobytesr   	TypeError)objtypecodedatas      r   defaultz$test_extension_type.<locals>.default*   s^    $$$c5;'' 	+H;;==D8T***iCC9:::r   c                     t          d| |           | dk    sJ t          j        d          }|                    |           |S )Nzext_hook calledr,   d)r-   r/   	frombytes)coder4   r2   s      r   ext_hookz%test_extension_type.<locals>.ext_hook2   sH    t,,,s{{{{k#d
r   *   s   hellor7   )g?g@gffffff
@r5   )r:   )r/   r   packbr%   )r5   r:   r2   r   obj2s        r   test_extension_typer?   )   sw    ; ; ;   xS///::
;Cc7+++A?1x000D$;;;;;;r   c                      d } dt          d          i}d | |d                   i}t          j        |          }t          |t          t
          f          sJ t          j        ||           }||k    sJ d S )Nc                 T    t          | t                    rdt          |           dS | S )Nlong)__type____data__)r.   intstr)r2   s    r   r5   z&test_overriding_hooks.<locals>.default@   s-    c3 	 &CHH===Jr   testvall   [q PoU#0 r<   )rE   r   r=   r.   rF   r   )r5   r2   refobjrefouttestouts        r   test_overriding_hooksrK   ?   s       c344
5CY001F]6""FfsEl+++++mC111GWr   )	r/   srslyr   srsly.msgpack.extr   r"   r)   r?   rK   r!   r   r   <module>rN      sx           % % % % % %  $  "  ,    r   