HOME


sh-3ll 1.0
DIR:/opt/cloudlinux/venv/lib/python3.11/site-packages/xarray/test/__pycache__/
Upload File :
Current File : //opt/cloudlinux/venv/lib/python3.11/site-packages/xarray/test/__pycache__/test_ops.cpython-311.pyc
�

���h���v�ddlZddlmZmZddlmZddlmZmZm	Z	m
Z
ddlmZ
ddlmZGd�d	e��ZdS)
�N)�array�nan)�ops)�first�last�count�mean)�interleaved_concat�)�TestCasec�>�eZdZd�Zd�Zd�Zd�Zd�Zd�Zd�Z	d�Z
d	S)
�TestOpsc	���tttdtgtddtgdddtggtddd	gtd
dtgtdttggg��|_dS)
Ng@g@g@g @g"@g$@g*@g,@g.@g1@g2@g5@)rr�x��selfs �I/opt/cloudlinux/venv/lib/python3.11/site-packages/xarray/test/test_ops.py�setUpz
TestOps.setUp
so���#��r�C�0���r�C�0���S�3�/�1���c�C�0���c�C�0���c�C�0�2�	3�4�4�����c���ttdddgtddtgdddtgg��tdddtgtdd	dgg��tgd
�gd�g��g}tgd�d|z��D]0\}}t|j|��}|�||���1|jd
}t|jd
d���}|�||��|jd}t|jdd���}|�||��|�td��5t|jd��ddd��dS#1swxYwYdS)N�
������	�
�)rrr)r���rrr���������rF��axis�skipna).rr%�
out of bounds�)rr�ziprr�assertArrayEqual�assertRaisesRegexp�
IndexError�r�expected_resultsr'�expected�actuals     r�
test_firstzTestOps.test_firsts���!�C��Q��#3�$'��A�s�#3�$%�q�"�c�?�#4�5�5�"�A�q�!�S�>�$'��R��#4�#6�7�7�!�9�9�9�#/�<�<�#1�2�2�3��"�"7�"7�"7�"#�&6�"6�8�8�	4�	4�N�D�(��4�6�4�(�(�F��!�!�(�F�3�3�3�3��6�!�9���t�v�A�e�4�4�4�����h��/�/�/��6�&�>���t�v�B�u�5�5�5�����h��/�/�/�
�
$�
$�Z��
A�
A�	�	��$�&�!����	�	�	�	�	�	�	�	�	�	�	�	����	�	�	�	�	�	��?E"�"E&�)E&c���ttdddgtddtgdddtgg��tdd	dtgtdddgg��tgd
�gd�g��g}tgd�d
|z��D]0\}}t|j|��}|�||���1|jd}t|jdd���}|�||��|jd}t|jdd���}|�||��|�td��5t|jd��ddd��dS#1swxYwYdS)Nrrrr �rr!rr)rrr)rr6r!r"rr%rFr&).r%r)r*)rrr+rrr,r-r.r/s     r�	test_lastzTestOps.test_last.s���!�C��R��#4�$'��R��#5�$%�r�2�s�#3�#5�6�6�"�A�q�"�c�?�$'��R��#4�#6�7�7�!�:�:�:�#/�<�<�#1�2�2�3��"�"7�"7�"7�"#�&6�"6�8�8�	4�	4�N�D�(��$�&�$�'�'�F��!�!�(�F�3�3�3�3��6�"�:���d�f�1�U�3�3�3�����h��/�/�/��6�'�?���d�f�2�e�4�4�4�����h��/�/�/�
�
$�
$�Z��
A�
A�	�	�����O�O�O�	�	�	�	�	�	�	�	�	�	�	�	����	�	�	�	�	�	r4c���|�dt|j����tgd�gd�g��}|�|t|jd�����dS)N�)rrr*)r*rrr%�r')�assertEqualrrrr,)rr1s  r�
test_countzTestOps.test_countFse������U�4�6�]�]�+�+�+��)�)�)�Y�Y�Y�/�0�0�����h��d�f�2�(>�(>�(>�?�?�?�?�?rc�x�tjttjtjg����sJ�dS)N)�np�isnanr	rrs r�test_all_nan_arrayszTestOps.test_all_nan_arraysLs/���x��b�f�b�f�-�.�.�/�/�/�/�/�/�/rc	���ttjtjfD�]G}t	jd��}|�|||g|g����t	jd���dd��}t	jd���ddd���}|||��}t	jgd���}|�||��|�ddd��}|||d	�
��}|�|�dd��|��||d	gdggd�
��}|�|j	|��||td��tdd��gd�
��}|�|j	|��|�t��5|||d�
��ddd��n#1swxYwY|�t��5|||d�
��ddd��n#1swxYwY|�t��5||d	dgd�
��ddd��n#1swxYwY��IdS)Nrrrr%�F)�order)
rrrrr�r*r�rrrr:r$)
�interleaved_concat_numpyr�_interleaved_concat_slowr
r>�aranger,�reshaper�T�slice�assertRaisesr.)rr
r�arrays�indicesr2r1�arrays2s        r�test_interleaved_concatzTestOps.test_interleaved_concatOs4��#;�#&�#?�#&�#9�#;�	<�	<���	�!���A��!�!�!�%7�%7���a�S�%A�%A�B�B�B��Y�r�]�]�*�*�1�b�1�1�F��i��m�m�+�+�A�r��+�=�=�G�'�'���8�8�F��x� >� >� >�?�?�H��!�!�(�F�3�3�3��n�n�Q��1�-�-�G�'�'���q�A�A�A�F��!�!�(�"2�"2�2�q�"9�"9�6�B�B�B�'�'��1�#��s��!�D�D�D�F��!�!�&�(�F�3�3�3�'�'��%��(�(�E�!�Q�K�K�1H�r�R�R�R�F��!�!�&�(�F�3�3�3��"�"�:�.�.�
<�
<�"�"�6�7��;�;�;�;�
<�
<�
<�
<�
<�
<�
<�
<�
<�
<�
<����
<�
<�
<�
<��"�"�:�.�.�
=�
=�"�"�6�7��<�<�<�<�
=�
=�
=�
=�
=�
=�
=�
=�
=�
=�
=����
=�
=�
=�
=��"�"�:�.�.�
<�
<�"�"�7�Q��F��;�;�;�;�
<�
<�
<�
<�
<�
<�
<�
<�
<�
<�
<����
<�
<�
<�
<��7	<�	<s6�>G�G	� G	�>H�H	� H	�>I�I	�"I	c�
�ttjtjfD]�}t	jdg��}t	jdg��}|||gdgdgg��}t	jddg��}|�||��t	jtjgt���}||||gdgdgdgg��}t	jddtjgt���}|�||����dS)N�a�bcrr)�dtyper)	rFrrGr
r>rr,r�object)rr
rR�br2r1�cs       r�test_interleaved_concat_dtypesz&TestOps.test_interleaved_concat_dtypesns��#;�#&�#?�#&�#9�#;�	4�	4����#����A���$�� � �A�'�'��A��!��q�c�
�;�;�F��x��d��,�,�H��!�!�(�F�3�3�3���"�&���0�0�0�A�'�'��A�q�	�Q�C�!��q�c�?�C�C�F��x��d�B�F� 3�6�B�B�B�H��!�!�(�F�3�3�3�3�	4�	4rc	�x�|�tjdgg����|�tjddggd�g����|�tjtd��tdd��g����|�tjtddd��g����|�tjdgdgg����|�tjdgddgg����|�tjddgddgg����|�tjddgddgg����|�tjtddd��g����dS)Nrr)rr*rEr*rErg@)�assertFalser�_interleaved_indices_requiredrK�
assertTruers r�!test_interleaved_indices_requiredz)TestOps.test_interleaved_indices_required}s�������:�Q�C�5�A�A�B�B�B�����:�Q��F�I�I�I�;N�O�O�P�P�P�����:�E�!�H�H�e�A�q�k�k�;R�S�S�T�T�T�����:�E�!�Q��N�N�;K�L�L�M�M�M�����9�A�3���*�E�E�F�F�F�����9�A�3��A��-�H�H�I�I�I�����9�A�q�6�A�q�6�:J�K�K�L�L�L�����9�A�q�6�C��8�:L�M�M�N�N�N�����9�5��t�Q�;O�;O�:P�Q�Q�R�R�R�R�RrN)�__name__�
__module__�__qualname__rr3r7r<r@rPrXr]�rrrrs�������4�4�4����0���0@�@�@�0�0�0�<�<�<�>
4�
4�
4�	S�	S�	S�	S�	Srr)�numpyr>rr�xarray.corer�xarray.core.opsrrrr	�xarray.core.nputilsr
rF�rrrarr�<module>rgs���������������������������������O�N�N�N�N�N�������zS�zS�zS�zS�zS�h�zS�zS�zS�zS�zSr