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__/__init__.cpython-311.pyc
�

���h����ddlZddlZddlmZddlZddlmZddlm	Z	m
Z
mZddlm
Z
ddlmZ	ddlZn#e$rddlZYnwxYw	ddlZdZn
#e$rdZYnwxYw	ddlZdZn
#e$rdZYnwxYw	ddlZdZn
#e$rdZYnwxYw	ddlZdZn
#e$rdZYnwxYw	ddlZdZn
#e$rdZYnwxYw	ddlZddlZej ej!�	��dZ"n
#e$rdZ"YnwxYw	ddl#Z#dZ$n
#e$rdZ$YnwxYwd
�Z%d�Z&d�Z'd
�Z(d�Z)d�Z*d�Z+d�Z,d�Z-d�Z.d d�Z/Gd�dej0��Z0Gd�de1��Z2Gd�de	j3��Z4Gd�de5��Z6d�Z7dS)!�N)�contextmanager��assert_array_equal)�utils�nputils�ops)�as_variable)�PY3TF)�getc�N�tr|ntjd��|��S)Nzrequires scipy)�	has_scipy�unittest�skip��tests �I/opt/cloudlinux/venv/lib/python3.11/site-packages/xarray/test/__init__.py�requires_scipyrB�'���G�4�4�"A�(�-�0@�"A�"A�$�"G�"G�G�c�N�tr|ntjd��|��S)Nzrequires pydap.client)�	has_pydaprrrs r�requires_pydaprFs'���N�4�4�"H�(�-�0G�"H�"H��"N�"N�Nrc�N�tr|ntjd��|��S)Nzrequires netCDF4)�has_netCDF4rrrs r�requires_netCDF4rJs'���K�4�4�$E�H�M�2D�$E�$E�d�$K�$K�Krc�N�tr|ntjd��|��S)Nzrequires h5netcdf)�has_h5netcdfrrrs r�requires_h5netcdfrNs'���M�4�4�%G�X�]�3F�%G�%G��%M�%M�Mrc�N�tr|ntjd��|��S)Nzrequires pynio)�	has_pyniorrrs r�requires_pynior!Rrrc�\�tstr|ntjd��|��S)Nzrequires scipy or netCDF4)r
rrrrs r�requires_scipy_or_netCDF4r#Vs5���
B��
B�D�D�;���:�;�;�D�A�A�Crc�N�tr|ntjd��|��S)Nz
requires dask)�has_daskrrrs r�
requires_daskr&[s&���E�4�4�!?����!?�!?��!E�!E�Erc�N�tr|ntjd��|��S)Nzrequires matplotlib)�has_matplotlibrrrs r�requires_matplotlibr)_s'��!�Q�4�4�'K�x�}�5J�'K�'K�D�'Q�'Q�Qrc��tjd}tjd}d}|dkr|dkrd}|s|ntjd��|��S)z'
    Test won't work in Python 2.6
    r�F��Tzerror on Python 2.6)�sys�version_inforr)r�major�minor�py26s    r�incompatible_2_6r3cs_��
��Q��E���Q��E��D���z�z��A�:�:��D��K�4�4�!E���/D�!E�!E�d�!K�!K�Krc�r�|jjdkr&tjj�|dd��S|S)N�Szutf-8�replace)�dtype�kind�np�core�defchararray�decode)�datas r�decode_string_datar>ps3���z��#����w�#�*�*�4��)�D�D�D��Kr��h㈵��>�:�0�yE>c���td�||fD����rt|��}t|��}gd��t�fd�||fD����rtj||��Stj||||���S)Nc3�6K�|]}|jjdkV��dS)r5N�r7r8)�.0�arrs  r�	<genexpr>z)data_allclose_or_equiv.<locals>.<genexpr>ws+����
9�
9�S�3�9�>�S� �
9�
9�
9�
9�
9�
9r)�M�m�O�Uc3�4�K�|]}|jj�vV��dS�NrC)rDrE�exact_dtypess  �rrFz)data_allclose_or_equiv.<locals>.<genexpr>{s,�����
B�
B�c�3�9�>�\�)�
B�
B�
B�
B�
B�
Br��rtol�atol)�anyr>r�array_equiv�allclose_or_equiv)�arr1�arr2rOrPrMs    @r�data_allclose_or_equivrVvs����
�
9�
9�T�4�L�
9�
9�
9�9�9�(�!�$�'�'��!�$�'�'��'�'�'�L�
�
B�
B�
B�
B�d�D�\�
B�
B�
B�B�B�G���t�T�*�*�*��$�T�4�d��F�F�F�Frc��eZdZerdd�Zed���Zd�Zd�Zdd�Z	d	�Z
d
�Zd�Zd�Z
d
�Zdd�Zd�Zd�Zd�Zdd�ZdS)�TestCaseNc�0�|�|||��SrL)�assertCountEqual)�self�first�second�msgs    r�assertItemsEqualzTestCase.assertItemsEqual�s���(�(����<�<�<rc#��K�tjd���5}tjd���dV�t|��dksJ�t	�fd�|D����sJ�	ddd��dS#1swxYwYdS)NT)�record�alwaysrc3�D�K�|]}�t|j��vV��dSrL)�str�message)rD�wires  �rrFz'TestCase.assertWarns.<locals>.<genexpr>�s0�����>�>�b�w�#�b�j�/�/�1�>�>�>�>�>�>r)�warnings�catch_warnings�filterwarnings�len�all)r[re�ws ` r�assertWarnszTestCase.assertWarns�s������
�
$�D�
1�
1�
1�	?�Q��#�H�g�6�6�6��E�E�E��q�6�6�A�:�:�:�:��>�>�>�>�A�>�>�>�>�>�>�>�>�>�		?�	?�	?�	?�	?�	?�	?�	?�	?�	?�	?�	?����	?�	?�	?�	?�	?�	?s�AA3�3A7�:A7c�b�t|���|��sJ||f���dSrL)r	�equals�r[�v1�v2s   r�assertVariableEqualzTestCase.assertVariableEqual�s4���2���%�%�b�)�)�3�3�B��8�3�3�)�3�3rc�b�t|���|��sJ||f���dSrL)r	�	identicalrps   r�assertVariableIdenticalz TestCase.assertVariableIdentical�s4���2���(�(��,�,�6�6�r�2�h�6�6�,�6�6rr?r@c��|�|j|j��t|j|j||���}|sJ|j|jf���dS�NrN)�assertEqual�dimsrV�values)r[rqrrrOrP�allcloses      r�assertVariableAllClosezTestCase.assertVariableAllClose�s_�������"�'�*�*�*�)��I�r�y�t�$�8�8�8���/�/�"�)�R�Y�/�/�/�x�/�/rc�p�|�t|���|����dSrL)�assertFalser	rorps   r�assertVariableNotEqualzTestCase.assertVariableNotEqual�s0������R���/�/��3�3�4�4�4�4�4rc�&�t||��dSrLr�r[�a1�a2s   r�assertArrayEqualzTestCase.assertArrayEqual�s���2�r�"�"�"�"�"rc�2�||ks||kr||ksJ�dSdSrL�r�s   rryzTestCase.assertEqual�s(���R�x�x�B�"�H�H��r����2�x�x��rc�H�|�|��sJ||f���dSrL)ro�r[�d1�d2s   r�assertDatasetEqualzTestCase.assertDatasetEqual�s+���y�y��}�}�&�&�r�2�h�&�&�}�&�&rc�H�|�|��sJ||f���dSrL)rur�s   r�assertDatasetIdenticalzTestCase.assertDatasetIdentical�s.���|�|�B���)�)�"�b��)�)��)�)rc�2�|�t|t���t|t�����|�|j|j��|D]5}|j|}|j|}|�||||����6dS)N)�keyrN)ry�sortedrdr_�coords�	variablesr})r[r�r�rOrP�krqrrs        r�assertDatasetAllClosezTestCase.assertDatasetAllClose�s���������,�,�,�f�R�S�.A�.A�.A�B�B�B����b�i���3�3�3��	F�	F�A���a��B���a��B��'�'��B�T��'�E�E�E�E�	F�	Frc���|�t|j��t|j����|jD]2}|j|}|j|}|�||���3dSrL)ryr�r�rs)r[r�r�r�rqrrs      r�assertCoordinatesEqualzTestCase.assertCoordinatesEqual�sv�������	�*�*�F�2�9�,=�,=�>�>�>���	-�	-�A���1��B���1��B��$�$�R��,�,�,�,�	-�	-rc�^�|�||��|�||��dSrL)rsr��r[�ar1�ar2s   r�assertDataArrayEqualzTestCase.assertDataArrayEqual�s4��� � ��c�*�*�*��#�#�C��-�-�-�-�-rc��|�|j|j��|�|���|�����dSrL)ry�namer��_to_temp_datasetr�s   r�assertDataArrayIdenticalz!TestCase.assertDataArrayIdentical�s[�������3�8�,�,�,��#�#�C�$8�$8�$:�$:�$'�$8�$8�$:�$:�	<�	<�	<�	<�	<rc�d�|�||||���|�||��dSrx)r}r�)r[r�r�rOrPs     r�assertDataArrayAllClosez TestCase.assertDataArrayAllClose�s;���#�#�C��4�d�#�C�C�C��#�#�C��-�-�-�-�-rrL�r?r@)�__name__�
__module__�__qualname__r
r_rrmrsrvr}r�r�ryr�r�r�r�r�r�r�r�rrrXrX�s!������
�=�	=�	=�	=�	=��?�?��^�?�4�4�4�7�7�7�0�0�0�0�5�5�5�#�#�#�3�3�3�'�'�'�
*�*�*�
F�F�F�F�-�-�-�.�.�.�<�<�<�
.�.�.�.�.�.rrXc��eZdZdS)�UnexpectedDataAccessN)r�r�r�r�rrr�r��s�������Drr�c��eZdZd�Zd�ZdS)�InaccessibleArrayc��||_dSrL)�array)r[r�s  r�__init__zInaccessibleArray.__init__�s
����
�
�
rc� �td���)NzTried accessing data)r��r[r�s  r�__getitem__zInaccessibleArray.__getitem__�s��"�#9�:�:�:rN)r�r�r�r�r�r�rrr�r��s2���������;�;�;�;�;rr�c��eZdZd�ZdS)�
ReturnItemc��|SrLr�r�s  rr�zReturnItem.__getitem__�s���
rN)r�r�r�r�r�rrr�r��s#����������rr�c�^�t|dtj|��j��}|�|}|S)z_Given an ndarray, return the base object which holds its memory, or the
    object itself.
    �base)�getattrr9�asarrayr�)r�r�s  r�source_ndarrayr��s1���5�&�"�*�U�"3�"3�"8�9�9�D��|����Krr�)8r.rg�
contextlibr�numpyr9�
numpy.testingr�xarray.corerrr�xarray.core.variabler	�xarray.core.pycompatr
�	unittest2r�ImportError�scipyr
�pydap.client�pydapr�netCDF4r�h5netcdfr�Nior �
dask.array�dask�set_optionsrr%�
matplotlibr(rrrrr!r#r&r)r3r>rVrX�	Exceptionr��NDArrayMixinr��objectr�r�r�rr�<module>r�s��
�
�
�
�����%�%�%�%�%�%�����,�,�,�,�,�,�+�+�+�+�+�+�+�+�+�+�,�,�,�,�,�,�$�$�$�$�$�$�� � � � � �������O�O�O�O�O�������L�L�L��I�I�������I�I�I�����������I�I�������I�I�I�������N�N�N��K�K�������K�K�K�������O�O�O��L�L�������L�L�L�������J�J�J��I�I�������I�I�I�����������K�K�K��D����"�"�"�"��H�H�������H�H�H�����������N�N�������N�N�N�����H�H�H�O�O�O�L�L�L�N�N�N�H�H�H�C�C�C�
F�F�F�R�R�R�
L�
L�
L����G�G�G�G�K.�K.�K.�K.�K.�x� �K.�K.�K.�\	�	�	�	�	�9�	�	�	�;�;�;�;�;��*�;�;�;����������
����s��5�	A�A�A�A�A�A!�!A+�*A+�/A6�6B�?B�B�B�B�B � B*�)B*�. C�C�C�C$�$C.�-C.