← Back
Editing: test_discrete_basic.cpython-311.pyc
� d�c�N � � � d dl mZ d dl mZ d dlZd dlZd dlmZ ddl m Z mZmZm Z mZmZmZmZmZmZmZmZmZmZmZmZ d dlmZmZ d dlmZ g d�g d �fZe ej e� � � dggz Zdd hZ!d� Z"ej# �$ d� � ej# �% d e"� � � � d� � � � � Z&ej# �$ d� � ej# �% de� � d� � � � � Z'ej# �% de� � d� � � Z(ej# �% de� � d� � � Z)ej# �% de� � d� � � Z*d� Z+d� Z,d� Z-d� Z.d� Z/d� Z0ej# �% d g d!�� � ej# �% d"e� � d#� � � � � Z1ej# �% d"e� � d$� � � Z2d%� Z3ej# �% d& e3� � � � d'� � � Z4d(� Z5ej# �% d)e� � d*� � � Z6d+� Z7dS ),� N)�assert_allclose)�stats� )�check_normalization�check_moment�check_mean_expect�check_var_expect�check_skew_expect�check_kurt_expect� check_entropy�check_private_entropy�check_edge_support�check_named_args�check_random_state_property�check_pickling�check_rvs_broadcast�check_freezing�'check_deprecation_warning_gh5982_moment�)check_deprecation_warning_gh5982_interval)�distdiscrete�invdistdiscrete)�rv_discrete_frozen)r � � � )皙�����?皙�����?g333333�?g�������?��values� �zipfian� nhypergeomc # �� K � t � � } t D ]V\ }}|t v r*t j |||t j j �� � V � n |||| vfV � | � |� � �Wd S )N)�marks)�setr �distslow�pytest�param�mark�slow�add)�seen�distname�args �G/usr/lib/python3/dist-packages/scipy/stats/tests/test_discrete_basic.py�cases_test_discrete_basicr0 s� � � � ��5�5�D�%� � � ��#��x����,�x��h�f�k�>N�O�O�O�O�O�O�O��C���!5�5�5�5�5���������� � zignore::RuntimeWarningzdistname,arg,first_casec �� � t t | � � }n# t $ r | }d} Y nw xY wt j � d� � |j |ddi�}t j |� � } |j |� \ }}t |||| dz � � t ||| � � t |||| dz � � t ||� � t ||| � � t ||| � � d}t ||||| dz � � |r�d } |j |j |j |j |j g} d ddd d�}|� | d� � }t+ |||| | � � | dk rt- |� � t/ ||� � t1 ||� � t3 ||� � t5 ||| � � |j j t j j k rt= ||t j � � d S d S d S )N�sample distributioniP� �sizei� z cdf_ppfz oth�{�G�z�?z chisquare)r � r r � )�randint� hypergeom� bernoulli�nchypergeom_walleniusr )�getattrr � TypeError�np�random�seed�rvs�unique� check_cdf_ppf� check_pmf_cdf� check_othr r r �check_discrete_chisquare�pmf�logpmf�cdf�logcdf�logsf�getr �check_scale_docstringr r r r � __class__�_entropy�rv_discreter ) r- r. � first_case�distfnrA �supp�m�v�alpha�locscale_defaults�meths�spec_k�ks r/ �test_discrete_basicr[ $ s@ � �)����)�)����� )� )� )���(����)���� �I�N�N�7���� �&�*�� %�$� %� %�C� �9�S�>�>�D��6�<���D�A�q��&�#�t�X� �%:�;�;�;��&�#�x�(�(�(� �f�c�4��F�!2�3�3�3��v�s�#�#�#�+�F�C��B�B�B�-�f�c�8�D�D�D��E��V�S�#�u��l�"�$� $� $� � B� ����V�]�F�J�� ����� �a�a�+,�.� .���J�J�x��#�#�����C�):�E�B�B�B��,�,�,�!�&�)�)�)�#�F�C�0�0�0��v�s�#�#�#��v�s�#�#�#� �f�c�8�,�,�,���$��(9�(B�B�B�!�&�#�u�/@�A�A�A�A�A�'B� B�$ C�B� � �)�)zdistname,argc � � t t | � � }n# t $ r | }d} Y nw xY w |j |ddi�\ }}}}t ||| � � t ||||| � � t |||| � � t ||||| � � t |||||| � � | dvrt |||||| � � t |||d� � t |||||z z d� � d S )Nr3 �moments�mvsk)�zipf� yulesimonr r ) r<