U
    g                     @   sR   d Z ddlT ddlZG dd dejZi fddZedkrNd	d
 Zejdd dS )z&Self-tests for Cryptodome.Util.Counter    )*Nc                   @   sD   e Zd Zdd Zdd Zdd Zdd Zd	d
 Zdd Zdd Z	dS )CounterTestsc                 C   s   ddl ma d S )Nr   )Counter)ZCryptodome.Utilr   )self r   I/tmp/pip-unpacked-wheel-_q8s9isk/Cryptodome/SelfTest/Util/test_Counter.pysetUp    s    zCounterTests.setUpc                 C   s   t d}t jddd}dS )z
Big endian   FZlittle_endianNr   newr   cr   r   r   test_BE$   s    
zCounterTests.test_BEc                 C   s   t jddd}dS )zLittle endianr	   Tr
   Nr   r   r   r   r   test_LE)   s    zCounterTests.test_LEc                 C   s    t jdd}| tt jd d S )Nr	   )nbits   r   r   assertRaises
ValueErrorr   r   r   r   
test_nbits-   s    zCounterTests.test_nbitsc                 C   s   t jdtdd}d S )Nr	   xx)prefixr   r   br   r   r   r   test_prefix1   s    zCounterTests.test_prefixc                 C   s   t jdtdd}d S )Nr	   r   )suffixr   r   r   r   r   test_suffix4   s    zCounterTests.test_suffixc                 C   s&   t jddd}| jtt jddd d S )Nr	      )initial_value   i r   r   r   r   r   test_iv7   s    zCounterTests.test_ivN)
__name__
__module____qualname__r   r   r   r   r   r   r!   r   r   r   r   r      s   r   c                 C   s   ddl m} |tS )Nr   )list_test_cases)ZCryptodome.SelfTest.st_commonr%   r   )configr%   r   r   r   	get_tests;   s    r'   __main__c                   C   s   t t S )N)unittestZ	TestSuiter'   r   r   r   r   <lambda>@       r*   suite)ZdefaultTest)	__doc__ZCryptodome.Util.py3compatr)   ZTestCaser   r'   r"   r,   mainr   r   r   r   <module>   s   