Current File : //lib64/python3.9/site-packages/numpy/random/tests/__pycache__/test_seed_sequence.cpython-39.pyc
a
z[yc� � @ s8 d dl Zd dlmZmZ d dlmZ dd� Zdd� ZdS )� N)�assert_array_equal�assert_array_compare)� SeedSequencec C s* g d�g d�g d�g d�g d�g d�g d�g d�g d �g d
�g
} g d �g d �g d
�g d�g d�g d�g d�g d�g d�g d�g
}ddgddgddgdd gd d gd d gd!d"gd#d$gd%d&gd'd(gg
}t | ||�D ]X\}}}tj|tjd)�}t|�}|�t|��}t||� |jt|�tjd)�}t||� q�d*S )+z� Check that SeedSequence generates data the same as the C++ reference.
https://gist.github.com/imneme/540829265469e673d045
)l �>[= i�ʭ i��
ixV4)l AM5 l �l�p i1 cl �^<1 )i�� l O!�p iF�6ii��M)i(��l Nw�U i]e+ ivp2p)i�v(oi��fl �F�N it��i)l �m l �-�
i�� i��5)i^w�Cl �x{ l �9 l I6�; )i�|�Il -Ker i� |Li��r)l gT<> irT�"l o�j l �P�J )l �b�I l �x�W l s-_` id�f3)l �M�R i�b"l � n, l �=� )l �u l � 8 iK�jQl �b< )l S�( l E{| iH5lDl !bJ< )i�٦Sl � , i�k�Li�&W)i��@l ;: i>f!l �'4 )l l l D� i��2l �z9e )l �4 l Kp�` l �a$ i^
)i��)9l 't�M i��npi�Rp)l �2DZ iA�l �DI i~G)i55�l �\| ih
� l ^r l �M�R� l � n,swM' l �u 2R,`
l KZ�"R�q l S�(i�q l H5��+q
l �YM' �0
l �k�=�R l �|�D� l >�B�W�P
l l]_ l �e�k� l �4.A� l �a${E�Pl �`Sr�P�7 l �K�`EJ�+ l �2DZe8N l �DI x;Bl 55T#LsNp l h
�x!� �ZdtypeN) �zip�np�array�uint32r �generate_state�lenr Zuint64) ZinputsZoutputsZ outputs64ZseedZexpectedZ
expected64�ss�stateZstate64� r �K/usr/lib64/python3.9/site-packages/numpy/random/tests/test_seed_sequence.py�test_reference_data sP �
�
�
r c C sr t d�} t d�}ttj| �d�|�d�� tjg d�tjd�}tt d��d�|� ttjt ddd��d�|� dS ) zD Ensure that the implicit zero-padding does not cause problems.
�* l � � )l �
� l m.<> i���yl �s�* r )r )Z spawn_keyN)r r r Z not_equalr
r r r )Zss0Zss1Z
expected42r r r �test_zero_padding9 s �
��r ) Znumpyr Z
numpy.testingr r Z numpy.randomr r r r r r r �<module> s 2
Mini Shell