← Back
Editing: test_crosstab.cpython-311.pyc
� d�c* � �@ � d dl Z d dlZd dlmZmZ d dlmZ e j � dddg� � d� � � Z d� Z d � Z e j � dddg� � d � � � Z e j � dddg� � d � � � Zd � Zd � Zd� Zd� Zd� ZdS )� N)�assert_array_equal� assert_equal)�crosstab�sparseFTc � � g d�}g d�}ddg}g d�}t j g d�g d�g� � }t ||| �� � \ \ }}}t ||� � t ||� � | rt |j |� � d S t ||� � d S ) N�r r � r r r r )� � � r r r r r r �r r r )r r r )r r r �r ��np�arrayr r �A� r �a�b�expected_avals�expected_bvals�expected_count�avals�bvals�counts �A/usr/lib/python3/dist-packages/scipy/stats/tests/test_crosstab.py�test_crosstab_basicr s� � � � � �A� � � �A���V�N� �Y�Y�N��X�y�y�y� (�y�y� *� +� +�N� $�Q��&� 9� 9� 9��N�U�E�E��u�n�-�-�-��u�n�-�-�-� � 2��5�7�N�3�3�3�3�3��5�.�1�1�1�1�1� c � � g d�} g d�}t j g d�� � }t | � � \ \ }}t ||� � t ||� � d S )N�r r r r r r r r )r r r �r r r r )�x�expected_xvalsr �xvalsr s r �test_crosstab_basic_1dr% s` � � � � �A� �Y�Y�N��X�i�i�i�(�(�N� �q�k�k�O�H�U�e��u�n�-�-�-��u�n�-�-�-�-�-r c �> � d} d}g d�}| | | | |||| g}g d�}ddg}| |g}g d�}t j g d�g d �gg d �g d �gg� � }t |||� � \ \ } } } } t | |� � t | |� � t | |� � t | |� � d S ) Nr r )r r r r r r r r )r r r r r r r r r r r )r r r �r r r )r r r )r r r r! ) r r r" �y�zr# �expected_yvals�expected_zvalsr r$ �yvals�zvalsr s r �test_crosstab_basic_3dr. " s� � � �A� �A� � � �A� �A�q�!�Q��1�a� �A� � � �A���V�N���V�N� �Y�Y�N��X� � � � )� � � +� )� � � )� � � +� ,� -� -�N� $,�A�q�!�#4�#4� ��U�E�5�5��u�n�-�-�-��u�n�-�-�-��u�n�-�-�-��u�n�-�-�-�-�-r c �& � g d�}g d�}ddg}g d�}t j g d�g d�g� � }t ||d g d�g| �� � \ \ }}}t ||� � t ||� � | rt |j |� � d S t ||� � d S ) Nr r r r �r r r r �r r r r �r r r r ��levelsr r r s r �test_crosstab_levelsr5 7 s� � � � � �A� � � �A���V�N�!�\�\�N��X�|�|�|� +�|�|� -� .� .�N� $�Q��4� � � �2F�,2� 4� 4� 4��N�U�E�E��u�n�-�-�-��u�n�-�-�-� � 2��5�7�N�3�3�3�3�3��5�.�1�1�1�1�1r c �* � g d�}g d�}ddg}g d�}t j g d�g d�g� � }t ||ddgg d�g| �� � \ \ }}}t ||� � t ||� � | rt |j |� � d S t ||� � d S ) N)r r r r r r r ���)r r r r r r r r r r r0 r1 r2 r3 r r s r �test_crosstab_extra_levelsr8 I s� � � "�!�!�A� � � �A���V�N�!�\�\�N��X�|�|�|� +�|�|� -� .� .�N� $�Q��A�q�6�<�<�<�2H�,2� 4� 4� 4��N�U�E�E��u�n�-�-�-��u�n�-�-�-� � 2��5�7�N�3�3�3�3�3��5�.�1�1�1�1�1r c � � t j t d�� � 5 t � � d d d � � d S # 1 swxY w Y d S )Nz At least one��match)�pytest�raises� TypeErrorr � r r � test_validation_at_least_oner@ ] s� � � � �y�� 7� 7� 7� � �� � � �� � � � � � � � � � � ���� � � � � � s �8�<�<c � � t j t d�� � 5 t ddgg d�� � d d d � � d S # 1 swxY w Y d S )Nzmust have the same lengthr: r r )r r r � �r< r= � ValueErrorr r? r r � test_validation_same_lengthsrE b s� � � � �z�)D� E� E� E� '� '��!�Q�� � � �&�&�&�'� '� '� '� '� '� '� '� '� '� '� '���� '� '� '� '� '� 's �>�A�Ac � � t j t d�� � 5 t g d�g d�g d�d�� � d d d � � d S # 1 swxY w Y d S )Nzonly two input sequencesr: r' �� rH r )r r r Tr rC r? r r �$test_validation_sparse_only_two_argsrI g s� � � � �z�)C� D� D� D� ?� ?�����I�I�I�y�y�y��>�>�>�>�?� ?� ?� ?� ?� ?� ?� ?� ?� ?� ?� ?���� ?� ?� ?� ?� ?� ?� �A�A� Ac � � t j t d�� � 5 t g d�g d�g d�f�� � d d d � � d S # 1 swxY w Y d S )Nznumber of input sequencesr: r' rG r0 )r4 rC r? r r �'test_validation_len_levels_matches_argsrL l s� � � � �z�)D� E� E� E� ?� ?�����I�I�I�|�|�|�o�>�>�>�>�?� ?� ?� ?� ?� ?� ?� ?� ?� ?� ?� ?���� ?� ?� ?� ?� ?� ?rJ c �f � t ddgddg� � } t | j | j f| � � d S )Nr r r )r r �elementsr )�ress r � test_resultrP q s8 � � �A�q�6�A�q�6� "� "�C��#�,�� �*�C�0�0�0�0�0r )r< �numpyr � numpy.testingr r �scipy.stats.contingencyr �mark� parametrizer r% r. r5 r8 r@ rE rI rL rP r? r r �<module>rV sM �� � � � � � � � � :� :� :� :� :� :� :� :� ,� ,� ,� ,� ,� ,� �����E�4�=�1�1� 2� 2� 2�1� 2� .� .� .�.� .� .�* �����E�4�=�1�1�2� 2� 2�1�2�" �����E�4�=�1�1�2� 2� 2�1�2�&� � � '� '� '� ?� ?� ?� ?� ?� ?� 1� 1� 1� 1� 1r
Save File
Cancel