Edit File: test_SHAKE.cpython-311.pyc
� ��Fhk � �� � d Z ddlZddlmZmZ ddlmZ ddlmZ ddl m Z m Z ddl m Z mZmZmZ G d� d ej � � Z G d � d e� � Z G d � d e� � Z G d� dej � � Z eddddd� i� � pg Z ee� � D ]G\ ZZej dk r e d� � Zn eej � � Zeej fd�Z e edez e � � �H eddddd� i� � pg Z ee � � D ]G\ ZZej dk r e d� � Zn eej � � Zeej fd�Z e ed ez e � � �Hi fd �Z!e"d k rddlZd � Z# ej$ d �!� � dS dS )"z5Self-test suite for Crypto.Hash.SHAKE128 and SHAKE256� N)�hexlify� unhexlify)�load_test_vectors)�list_test_cases)�SHAKE128�SHAKE256)�b�bchr�bord�tobytesc �, � e Zd Zd� Zd� Zd� Zd� Zd� ZdS )� SHAKETestc � � | j � � � }| j � t d� � �� � }| j � � � � t d� � � � }| � |� d� � |� d� � � � |� d� � | � |� d� � |� d� � � � d S )N�90��data� )�shake�newr �update�assertNotEqual�read� assertEqual)�self�xof1�xof2�xof3s ��/builddir/build/BUILD/imunify360-venv-2.5.3/opt/imunify360/venv/lib64/python3.11/site-packages/Crypto/SelfTest/Hash/test_SHAKE.py�test_new_positivezSHAKETest.test_new_positive, s� � ��z�~�~� � ���z�~�~�1�T�7�7�~�+�+���z�~�~� � �&�&�q��w�w�/�/�� ���D�I�I�b�M�M�4�9�9�R�=�=�9�9�9� � � �"� � � � ������2� � �� � �"� � �6�6�6�6�6� c �� � t d� � dz t d� � dz g}| j � � � }|� |d � � � |d � � |� d� � }| j � � � }|� |d |d z � � | � |� d� � |� � d S )Nr �� � i, r � )r r r r r r )r �pieces�h�digests r � test_updatezSHAKETest.test_update6 s� � ��r�(�(�S�.�$�r�(�(�S�.�1�� �J�N�N� � �� ��������"�"�6�!�9�-�-�-�������� �J�N�N� � �� ������V�A�Y�&�'�'�'� ���������V�,�,�,�,�,r c �z � | j � � � }| � t |j d� � d S )N�string)r r � assertRaises� TypeErrorr )r r& s r �test_update_negativez SHAKETest.test_update_negative? s3 � � �J�N�N� � �� ���)�Q�X�y�9�9�9�9�9r c �" � | j � � � }|� d� � }| � t |t t d� � � � � � � � | � t |� � d� � d S )N�Z r' ) r r r � assertTrue� isinstance�typer r �len)r r& r' s r � test_digestzSHAKETest.test_digestC sp � � �J�N�N� � ��������� ��� �6�4��(� � �+<�+<�=�=�>�>�>� ����V���b�)�)�)�)�)r c � � | j � � � }|� t d� � � � |� d� � | � t |j t d� � � � d S )N�rrrrr/ �ttt)r r r r r r+ r, )r �macs r �test_update_after_readz SHAKETest.test_update_after_readK s` � ��j�n�n� � �� � � �1�V�9�9� � � � ���� � � � ���)�S�Z��5���:�:�:�:�:r N)�__name__� __module__� __qualname__r r( r- r4 r9 � r r r r * s_ � � � � � �7� 7� 7�-� -� -�:� :� :�*� *� *�;� ;� ;� ;� ;r r c � � e Zd ZeZdS )� SHAKE128TestN)r: r; r< r r r= r r r? r? R � � � � � � ����r r? c � � e Zd ZeZdS )� SHAKE256TestN)r: r; r< r r r= r r rB rB V r@ r rB c � � e Zd ZdS )� SHAKEVectorsN)r: r; r< r= r r rD rD Z s � � � � � ��Dr rD )�Hash�SHA3zShortMsgKAT_SHAKE128.txtzShort Messages KAT SHAKE128r3 c � � t | � � S �N��int��xs r �<lambda>rM a � � �3�q�6�6� r � c � � t j |�� � }|� t |� � � � }| � ||� � d S �Nr )r r r r3 r �r r �result�hobjr' s r �new_testrU i �G � ��|��&�&�&�����3�v�;�;�'�'�� �����(�(�(�(�(r z test_128_%dzShortMsgKAT_SHAKE256.txtzShort Messages KAT SHAKE256c � � t | � � S rH rI rK s r rM rM t rN r c � � t j |�� � }|� t |� � � � }| � ||� � d S rQ )r r r r3 r rR s r rU rU | rV r z test_256_%dc � � g }|t t � � z }|t t � � z }|t t � � z }|S rH )r r? rB rD )�config�testss r � get_testsr\ � sB � � �E� �_�\� *� *�*�E� �_�\� *� *�*�E� �_�\� *� *�*�E� �Lr �__main__c �B � t j t � � � � S rH )�unittest� TestSuiter\ r= r r rM rM � s � �H�&�y�{�{�3�3� r �suite)� defaultTest)%�__doc__r_ �binasciir r �Crypto.SelfTest.loaderr �Crypto.SelfTest.st_commonr � Crypto.Hashr r �Crypto.Util.py3compatr r r r �TestCaser r? rB rD �test_vectors_128� enumerate�idx�tvr3 r �msg�mdrU �setattr�test_vectors_256r\ r: ra �mainr= r r �<module>rs s ��> <� ;� ���� '� '� '� '� '� '� '� '� 4� 4� 4� 4� 4� 4� 5� 5� 5� 5� 5� 5� *� *� *� *� *� *� *� *� 8� 8� 8� 8� 8� 8� 8� 8� 8� 8� 8� 8�%;� %;� %;� %;� %;��!� %;� %;� %;�P� � � � �9� � � �� � � � �9� � � � � � � � �8�$� � � � %�$�%5� 9� <�!&�)9�)9� ;�>� >� D� BD� � �y�)�*�*� 9� 9�G�C�� �v��{�{��q��u�u����w�r�v���� ��� )� )� )� )� �G�L�-�#�-�x�8�8�8�8� %�$�%5� 9� <�!&�)9�)9� ;�>� >� D� BD� � �y�)�*�*� 9� 9�G�C�� �v��{�{��q��u�u����w�r�v���� ��� )� )� )� )� �G�L�-�#�-�x�8�8�8�8� � � � � � �z����O�O�O� 3� 3�E��H�M�g�&�&�&�&�&�&� �r