
    wi0                     h    d dl Z d dlZd dlmZmZ d dlmZ d dlmZ d Zd Z	d Z
d Zd	 Zd
 Zd ZdS )    N)Unpacker
BufferFull)pack)	OutOfDatac                  d   t                      } |                     d           t          j        t                    5  t          t          |                      d d d            n# 1 swxY w Y   |                     d           t          j        t                    5  t          t          |                      d d d            n# 1 swxY w Y   |                     d           t          j        t                    5  t          t          |                      d d d            n# 1 swxY w Y   |                     d           t          j        t                    5  t          t          |                      d d d            n# 1 swxY w Y   |                     d           t          j        t                    5  t          t          |                      d d d            n# 1 swxY w Y   |                     d           t          t          |                     dk    sJ d S )N      h   a   l   os   hallo)r   feedpytestraisesStopIterationnextiterunpackers    r/root/.openclaw/workspace/chatterbox_venv_py311/lib/python3.11/site-packages/srsly/tests/msgpack/test_sequnpack.pytest_partialdatar      s   zzHMM'	}	%	%  T(^^              MM$	}	%	%  T(^^              MM$	}	%	%  T(^^              MM$	}	%	%  T(^^              MM$	}	%	%  T(^^              MM$X8++++++sY   A&&A*-A*CCCD**D.1D.#FFFG..G25G2c                     t          dd          } |                     d           |                                 t          d          k    sJ |                                 t          d          k    sJ |                                 t          d          k    sJ |                                 t          d          k    sJ |                                 t          d          k    sJ |                                 t          d	          k    sJ t	          j        t                    5  |                                  d d d            n# 1 swxY w Y   |                     d
           |                     d           d}t          | d          D ]\  }}|t          |          k    sJ |dz  } |t          d          k    sJ d S )N      	read_sizeuse_list   foobar   fr      br
      rs   foos   barr   	foobarbaz)	r   r   unpackordr   r   r   ziplen)r   koes       r   test_foobarr)      s   !a000HMM)??D		))))??D		))))??D		))))??D		))))??D		))))??D		))))	y	!	!                 MM&MM&	AHk**  1CFF{{{{	QIs   *EEEc                  @   t          dd          } |                     d           |                                 t          d          k    sJ |                                  |                                 t          d          k    sJ |                                  |                                 t          d          k    sJ |                                  t          j        t                    5  |                                  d d d            d S # 1 swxY w Y   d S )Nr   r   r   r   r   r   r
   )r   r   r"   r#   skipr   r   r   r   s    r   test_foobar_skipr,   3   s4   !a000HMM)??D		))))MMOOO??D		))))MMOOO??D		))))MMOOO	y	!	!                   s   1DDDc                     t          j        t                    5  t          dd           d d d            n# 1 swxY w Y   t          ddd          } |                     d           t          j        t
                    5  |                     d           d d d            n# 1 swxY w Y   |                     d           t          d	          t          |           k    sJ |                     d
           t          d          t          |           k    sJ t          d          t          |           k    sJ t          d          t          |           k    sJ d S )N   r   )r   max_buffer_sizer   )r   r/   r   s   fos   obr   fr   r'   b)r   r   
ValueErrorr   r   r   r#   r   r   s    r   test_maxbuffersizer3   @   s   	z	"	" 1 11a00001 1 1 1 1 1 1 1 1 1 1 1 1 1 1!QCCCHMM%	z	"	"  e              MM$s88tH~~%%%%MM$s88tH~~%%%%s88tH~~%%%%s88tH~~%%%%%%s   8<<B%%B),B)c                     t          d          } |                     d           |                                 t          d          k    sJ |                     d          dk    sJ |                                 t          d          k    sJ |                                 t          d          k    sJ t          t          j        d          d          } |                                 t          d          k    sJ |                     d          dk    sJ |                                 t          d          k    sJ |                                 t          d          k    sJ d S )Nr   )r   r   r   s   oobr
   r    )r   r   r"   r#   
read_bytesioBytesIOr   s    r   test_readbytesr8   O   sL   !$$$HMM)??D		))))q!!V++++??D		))))??D		)))) 
9--;;;H??D		))))q!!V++++??D		))))??D		))))))    c                     t                      } |                     d           t          |           dk    sJ t          |           dk    sJ |                     d           t          |           dk    sJ t          |           dk    sJ |                     d           t          |           dk    sJ t          |           dk    sJ d S )Ns   ?!)   ?   ! s   ?)r;   r<   )r<   )r   r   tupler   s    r   test_issue124r?   _   s    zzHMM-   ??l****??b    MM,??g%%%%??b    MM$??g%%%%??b      r9   c            	         t          j                    } d t          d          D             }|d t          dd          D             z  }|dddz  t          t          d                    d	 t          d
          D             d t          d          D             gz  }g }|D ]9}t	          ||            |                    |                                            :|                     d           t          |           }t          ||          D ]6\  }}t          |          }||k    sJ ||                                k    sJ 7d S )Nc                     g | ]
}d |z  dz
  S )   r   r=   .0is     r   
<listcomp>z$test_unpack_tell.<locals>.<listcomp>n   s     ...qQ
...r9   A   c                     g | ]}d |z   	S )rB   r=   rC   s     r   rF   z$test_unpack_tell.<locals>.<listcomp>o   s    111q16111r9   r   @   s   helloi     c                 4    i | ]}|t          |          |z  S r=   bytesrC   s     r   
<dictcomp>z$test_unpack_tell.<locals>.<dictcomp>t   $    ,,,QE!HHqL,,,r9   
   c                 4    i | ]}|t          |          |z  S r=   rL   rC   s     r   rN   z$test_unpack_tell.<locals>.<dictcomp>u   rO   r9       r   )r6   r7   rangelistr   appendtellseekr   r$   r   )streammessagesoffsetsmr   r'   m2s          r   test_unpack_tellr]   l   sT   Z\\F..E"II...H11E!RLL1111H4U2YY,,%)),,,,,%)),,, H G & &Qv{{}}%%%%
KKNNNHHg&& $ $1(^^BwwwwHMMOO#####$ $r9   )r6   r   srsly.msgpackr   r   r   srsly.msgpack.exceptionsr   r   r)   r,   r3   r8   r?   r]   r=   r9   r   <module>r`      s    				  . . . . . . . .       . . . . . ., , ,*  ,
 
 
& & &* * * 
! 
! 
!$ $ $ $ $r9   