o
    e1L                     @   s   d Z ddlZddlZddlZddlmZmZ G dd deZG dd deZ	G d	d
 d
eZ
G dd deZeejjdkdG dd deZG dd deZeejjdkdG dd deZG dd deZG dd deZdd ZG dd deZdS )a  
    Tests the h5py.Dataset.__getitem__ method.

    This module does not specifically test type conversion.  The "type" axis
    therefore only tests objects which interact with the slicing system in
    unreliable ways; for example, compound and array types.

    See test_dataset_getitem_types for type-conversion tests.

    Tests are organized into TestCases by dataset shape and type.  Test
    methods vary by slicing arg type.

    1. Dataset shape:
        Empty
        Scalar
        1D
        3D

    2. Type:
        Float
        Compound
        Array

    3. Slicing arg types:
        Ellipsis
        Empty tuple
        Regular slice
        MultiBlockSlice
        Indexing
        Index list
        Boolean mask
        Field names
    N   )utTestCasec                   @   t   e Zd Zdd Zdd Zdd Zdd Zd	d
 Zdd Zdd Z	dd Z
dd Zdd Zdd Zdd Zdd ZdS )	TestEmptyc                 C   sh   t |  tjtjj}tjj }|	d tj
| jjd||}t|| _ttd| _d S )N
      xZS10)r   setUph5pyh5screateNULLh5tZC_S1copyZset_sizeh5dfidDatasetdsetEmptynpdtype	empty_obj)selfZsidtidZdsid r   OD:\Projects\ConvertPro\env\Lib\site-packages\h5py/tests/test_dataset_getitem.pyr	   6   s   

zTestEmpty.setUpc                 C      |  | jjd dS  Verify number of dimensions r   NassertEqualr   ndimr   r   r   r   	test_ndim?      zTestEmpty.test_ndimc                 C      |  | jjd dS  Verify shape Nr!   r   shaper#   r   r   r   
test_shapeC   r%   zTestEmpty.test_shapec                 C   r&   r'   r!   r   sizer#   r   r   r   	test_sizeG   r%   zTestEmpty.test_sizec                 C   r   ) Verify nbytes r   N)r!   r   nbytesr#   r   r   r   test_nbytesK   r%   zTestEmpty.test_nbytesc                 C      |  | jd | j d S N.r!   r   r   r#   r   r   r   test_ellipsisO      zTestEmpty.test_ellipsisc                 C   r2   Nr   r4   r#   r   r   r   
test_tupleR   r6   zTestEmpty.test_tuplec                 C   >   |  t | jdd  W d   dS 1 sw   Y  dS z slice -> ValueError r      NassertRaises
ValueErrorr   r#   r   r   r   
test_sliceU      "zTestEmpty.test_slicec                 C   >   |  t | jt   W d   dS 1 sw   Y  dS z MultiBlockSlice -> ValueError Nr=   r>   r   r
   ZMultiBlockSlicer#   r   r   r   test_multi_block_sliceZ   r@   z TestEmpty.test_multi_block_slicec                 C   :   |  t | jd  W d   dS 1 sw   Y  dS z index -> ValueError r   Nr<   r#   r   r   r   
test_index_      "zTestEmpty.test_indexc                 C   >   |  t | jg d  W d   dS 1 sw   Y  dS  index list -> ValueError r         Nr<   r#   r   r   r   test_indexlistd   r@   zTestEmpty.test_indexlistc                 C   H   t jddd}| t | j|  W d   dS 1 sw   Y  dS z mask -> ValueError Tboolr   Nr   arrayr=   r>   r   r   maskr   r   r   	test_maski      "zTestEmpty.test_maskc                 C   rE   )z field name -> ValueError fieldNr<   r#   r   r   r   test_fieldnameso   rH   zTestEmpty.test_fieldnamesN__name__
__module____qualname__r	   r$   r+   r.   r1   r5   r8   r?   rD   rG   rO   rX   r[   r   r   r   r   r   4   s    	r   c                   @   r   )TestScalarFloatc                 C   2   t |  tjddd| _| jjd| jd| _d S )N     @E@r   rS   xdatar   r	   r   rU   re   r   create_datasetr   r#   r   r   r   r	   w      
zTestScalarFloat.setUpc                 C   r   r   r    r#   r   r   r   r$   |   r%   zTestScalarFloat.test_ndimc                 C   r    Verify size r   Nr,   r#   r   r   r   r.      r%   zTestScalarFloat.test_sizec                 C      |  | jj| jjj dS r/   Nr!   r   r0   re   r   itemsizer#   r   r   r   r1         zTestScalarFloat.test_nbytesc                 C      |  | jjt  dS r'   r!   r   r*   tupler#   r   r   r   r+         zTestScalarFloat.test_shapec                 C      | j d }| || j dS z Ellipsis -> scalar ndarray .Nr   assertArrayEqualre   r   outr   r   r   r5         
zTestScalarFloat.test_ellipsisc                 C   s    | j d }| || j  dS )z () -> bare item r   N)r   rw   re   itemrx   r   r   r   r8      s   
zTestScalarFloat.test_tuplec                 C   r9   r:   r<   r#   r   r   r   r?      r@   zTestScalarFloat.test_slicec                 C   rA   rB   rC   r#   r   r   r   rD      r@   z&TestScalarFloat.test_multi_block_slicec                 C   rE   rF   r<   r#   r   r   r   rG      rH   zTestScalarFloat.test_indexc                 C   rI   rJ   r<   r#   r   r   r   rO      r@   zTestScalarFloat.test_indexlistc                 C   rP   rQ   rT   rV   r   r   r   rX      rY   zTestScalarFloat.test_maskc                 C   rE   z& field name -> ValueError (no fields) rZ   Nr<   r#   r   r   r   r[      rH   zTestScalarFloat.test_fieldnamesNr]   r^   r_   r	   r$   r.   r1   r+   r5   r8   r?   rD   rG   rO   rX   r[   r   r   r   r   r`   u   s    r`   c                   @   r   )TestScalarCompoundc                 C   s6   t |  tjdg dd| _| jjd| jd| _d S )N)rb   iZHello))ar   )bi)cz|S10rS   rc   rd   rf   r#   r   r   r   r	         
zTestScalarCompound.setUpc                 C   r   r   r    r#   r   r   r   r$      r%   zTestScalarCompound.test_ndimc                 C   rp   r'   rq   r#   r   r   r   r+      rs   zTestScalarCompound.test_shapec                 C   r   ri   r,   r#   r   r   r   r.      r%   zTestScalarCompound.test_sizec                 C   rk   rl   rm   r#   r   r   r   r1      ro   zTestScalarCompound.test_nbytesc                 C   s@   | j d }| |tj | |j| jj | |j| jj dS ru   )r   assertIsInstancer   Zndarrayr!   r*   re   r   rx   r   r   r   r5      s   
z TestScalarCompound.test_ellipsisc                 C   s.   | j d }| |tj | |j| jj dS )z () -> np.void instance r   N)r   r   r   Zvoidr!   r   re   rx   r   r   r   r8         
zTestScalarCompound.test_tuplec                 C   r9   r:   r<   r#   r   r   r   r?      r@   zTestScalarCompound.test_slicec                 C   rA   rB   rC   r#   r   r   r   rD      r@   z)TestScalarCompound.test_multi_block_slicec                 C   rE   rF   r<   r#   r   r   r   rG      rH   zTestScalarCompound.test_indexc                 C   rI   rJ   r<   r#   r   r   r   rO      r@   z!TestScalarCompound.test_indexlistc                 C   rP   )z mask -> ValueError  TrR   rS   NrT   rV   r   r   r   rX      rY   zTestScalarCompound.test_maskc                 C   s.   | j d }| |tj | || j d  dS )z field name -> bare value r   N)r   r   r   Zfloat32r!   rx   r   r   r   r[      r   z"TestScalarCompound.test_fieldnamesNr\   r   r   r   r   r~      s    r~   c                   @   r   )TestScalarArrayc                 C   sP   t |  td| _tjg ddd| _| jjdd| jd| _	| j| j	d< d S )Nz(3,2)f))g	@i)*   g33333X@)gQ	@r   r   rS   rc   r   .)
r   r	   r   r   dtrU   re   r   rg   r   r#   r   r   r   r	     s
   
zTestScalarArray.setUpc                 C   s$   |  | jjd |  | jjd dS )r   rM   r   N)r!   re   r"   r   r#   r   r   r   r$   
  s   zTestScalarArray.test_ndimc                 C   r   ri   r,   r#   r   r   r   r.     r%   zTestScalarArray.test_sizec                 C   s   |  | jj| jjj dS rl   )r!   r   r0   r   rn   r#   r   r   r   r1     ro   zTestScalarArray.test_nbytesc                 C   s&   |  | jjd |  | jjt  dS )r(   )   rM   N)r!   re   r*   r   rr   r#   r   r   r   r+     s   zTestScalarArray.test_shapec                 C   rt   )z2 Ellipsis -> ndarray promoted to underlying shape .Nrv   rx   r   r   r   r5     rz   zTestScalarArray.test_ellipsisc                 C   rt   ) () -> same as ellipsis .Nrv   rx   r   r   r   r8   !  rz   zTestScalarArray.test_tuplec                 C   r9   r:   r<   r#   r   r   r   r?   &  r@   zTestScalarArray.test_slicec                 C   rA   rB   rC   r#   r   r   r   rD   +  r@   z&TestScalarArray.test_multi_block_slicec                 C   rE   rF   r<   r#   r   r   r   rG   0  rH   zTestScalarArray.test_indexc                 C   s:   |  t | jg   W d   dS 1 sw   Y  dS )rK   Nr<   r#   r   r   r   rO   5  rH   zTestScalarArray.test_indexlistc                 C   rP   rQ   rT   rV   r   r   r   rX   :  rY   zTestScalarArray.test_maskc                 C   rE   r|   r<   r#   r   r   r   r[   @  rH   zTestScalarArray.test_fieldnamesNr}   r   r   r   r   r     s    r   )r         zHDF5 1.8.7+ requiredc                   @   sd   e Zd Zdd Zdd Zdd Zdd Zd	d
 Zdd Zdd Z	dd Z
dd Zdd Zdd ZdS )Test1DZeroFloatc                 C   ra   )Nr   r   rS   rc   rd   r   r	   r   onesre   r   rg   r   r#   r   r   r   r	   I  rh   zTest1DZeroFloat.setUpc                 C   r   r   r   Nr    r#   r   r   r   r$   N  r%   zTest1DZeroFloat.test_ndimc                 C   r   )r(   r   Nr)   r#   r   r   r   r+   R  r%   zTest1DZeroFloat.test_shapec                 C      |  | j| jtjd  dS )z' Ellipsis -> ndarray of matching shape .NassertNumpyBehaviorr   re   r   s_r#   r   r   r   r5   V     zTest1DZeroFloat.test_ellipsisc                 C   r   )r   r   Nr   r#   r   r   r   r8   Z  r   zTest1DZeroFloat.test_tuplec                 C   s    |  | j| jtjdd  dS )z  slice -> ndarray of shape (0,) r   r;   Nr   r#   r   r   r   r?   ^  s    zTest1DZeroFloat.test_slicec                 C       |  | j| jtjdd  d S Nr   rN   r   r#   r   r   r   test_slice_stop_less_than_startb      z/Test1DZeroFloat.test_slice_stop_less_than_startc                 C   rE   )z index -> out of range r   Nr=   
IndexErrorr   r#   r   r   r   rG   e  rH   zTest1DZeroFloat.test_indexc                 C   s   |  | j| jtjg   dS )z index list Nr   r#   r   r   r   rO   j  r   zTest1DZeroFloat.test_indexlistc                 C   s.   t jddd}| j| j| jt j| dd dS )z# mask -> ndarray of matching shape r   rR   rS   TZskip_fast_readerN)r   r   r   r   re   r   rV   r   r   r   rX   n  s   
zTest1DZeroFloat.test_maskc                 C   rE   r|   r<   r#   r   r   r   r[   y  rH   zTest1DZeroFloat.test_fieldnamesN)r]   r^   r_   r	   r$   r+   r5   r8   r?   r   rG   rO   rX   r[   r   r   r   r   r   F  s    r   c                   @   s  e Zd Zdd Zdd Zdd Zdd Zd	d
 Zdd Zdd Z	dd Z
dd Zdd Zdd Zdd Zdd Zdd Zdd Zdd  Zd!d" Zd#d$ Zd%d& Zd'd( Zd)d* Zd+d, Zd-d. Zd/d0 Zd1d2 Zd3d4 Zd5d6 Zd7d8 Zd9d: Zd;d< Z d=d> Z!d?d@ Z"dAdB Z#dCS )DTest1DFloatc                 C   s4   t |  tdd| _| jjd| jd| _d S )N   r   rc   rd   )	r   r	   r   arangeZastypere   r   rg   r   r#   r   r   r   r	     s   
zTest1DFloat.setUpc                 C   r   r   r    r#   r   r   r   r$     r%   zTest1DFloat.test_ndimc                 C   r   )r(   )r   Nr)   r#   r   r   r   r+     r%   zTest1DFloat.test_shapec                 C      |  | j| jtjd  d S r3   r   r#   r   r   r   r5        zTest1DFloat.test_ellipsisc                 C   r   r7   r   r#   r   r   r   r8     r   zTest1DFloat.test_tuplec                 C   r   )Nr   r;   r   r#   r   r   r   test_slice_simple  r   zTest1DFloat.test_slice_simplec                 C   s    |  | j| jtjdd  d S )Nr;   r   r#   r   r   r   test_slice_zerosize  r   zTest1DFloat.test_slice_zerosizec                 C   "   |  | j| jtjddd  d S )Nr   r   r   r   r#   r   r   r   test_slice_strides     "zTest1DFloat.test_slice_stridesc                 C   r   )Nir   r   r#   r   r   r   test_slice_negindexes  r   z!Test1DFloat.test_slice_negindexesc                 C   r   r   r   r#   r   r   r   r     r   z+Test1DFloat.test_slice_stop_less_than_startc                 C   r   )Nd   i  r   r   r#   r   r   r   test_slice_outofrange  r   z!Test1DFloat.test_slice_outofrangec                 C   s@   |  t | jddd  W d   dS 1 sw   Y  dS )z we disallow negative steps Nr<   r#   r   r   r   test_slice_backwards  s   "z Test1DFloat.test_slice_backwardsc                 C   s"   |  | j| jtjd d d  d S )Nr   r   r#   r   r   r   test_slice_zerostride  r   z!Test1DFloat.test_slice_zerostridec                 C   r   )Nr   r   r#   r   r   r   test_index_simple  r   zTest1DFloat.test_index_simplec                 C   r   )Nr   r#   r   r   r   test_index_neg  r   zTest1DFloat.test_index_negc                 C   s:   |  t | jd   W d    d S 1 sw   Y  d S Nr=   	TypeErrorr   r#   r   r   r   test_index_none     "zTest1DFloat.test_index_nonec                 C   s:   |  t | ji   W d   dS 1 sw   Y  dS )z Illegal slicing argument Nr   r#   r   r   r   test_index_illegal  rH   zTest1DFloat.test_index_illegalc                 C   s:   |  t | jd  W d    d S 1 sw   Y  d S Nr   r   r#   r   r   r   test_index_outofrange  r   z!Test1DFloat.test_index_outofrangec                 C   s    |  | j| jtjg d  d S NrL   r   r#   r   r   r   test_indexlist_simple  r   z!Test1DFloat.test_indexlist_simplec              	   C   s&   |  | j| jtjtg d  d S r   r   r   re   r   r   rU   r#   r   r   r   test_indexlist_numpyarray  s   &z%Test1DFloat.test_indexlist_numpyarrayc                 C   s"   |  | j| jtjdgdf  d S Nr   .r   r#   r   r   r   $test_indexlist_single_index_ellipsis  r   z0Test1DFloat.test_indexlist_single_index_ellipsisc                 C   s(   |  | j| jtjtdgdf  d S r   r   r#   r   r   r   /test_indexlist_numpyarray_single_index_ellipsis  s   (z;Test1DFloat.test_indexlist_numpyarray_single_index_ellipsisc              	   C   s*   |  | j| jtjtg ddf  d S )NrL   .r   r#   r   r   r   "test_indexlist_numpyarray_ellipsis  s   *z.Test1DFloat.test_indexlist_numpyarray_ellipsisc                 C   s   |  | j| jtjg   d S r   r   r#   r   r   r   test_indexlist_empty  r   z Test1DFloat.test_indexlist_emptyc                 C   s<   |  t | jdg  W d    d S 1 sw   Y  d S r   r   r#   r   r   r   test_indexlist_outofrange  s   "z%Test1DFloat.test_indexlist_outofrangec                 C   rI   )z8 we require index list values to be strictly increasing )r   r   rM   Nr   r#   r   r   r   test_indexlist_nonmonotonic  r@   z'Test1DFloat.test_indexlist_nonmonotonicc                 C   sZ   |  | j| jtjg d  | t | jddg  W d    d S 1 s&w   Y  d S )N)r   rM   r   r   )r   r   re   r   r   r=   r   r#   r   r   r   !test_indexlist_monotonic_negative  s   "z-Test1DFloat.test_indexlist_monotonic_negativec                 C   rI   )z! we forbid repeated index values )r   r   rM   Nr   r#   r   r   r   test_indexlist_repeated  r@   z#Test1DFloat.test_indexlist_repeatedc                 C   &   | j | j| jtj| jdk dd d S )NiTr   r   r#   r   r   r   test_mask_true     
zTest1DFloat.test_mask_truec                 C   r   )Nr   Tr   r   r#   r   r   r   test_mask_false  r   zTest1DFloat.test_mask_falsec                 C   r   )NrN   Tr   r   r#   r   r   r   test_mask_partial  r   zTest1DFloat.test_mask_partialc                 C   sD   |  t | jtjddd  W d   dS 1 sw   Y  dS )z4 we require the boolean mask shape to match exactly )rM   rR   rS   N)r=   r   r   r   r   r#   r   r   r   test_mask_wrongsize  s   "zTest1DFloat.test_mask_wrongsizec                 C   rE   r|   r<   r#   r   r   r   r[   	  rH   zTest1DFloat.test_fieldnamesN)$r]   r^   r_   r	   r$   r+   r5   r8   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     sD    			r   c                   @   s,   e Zd Zdd Zdd Zdd Zdd Zd	S )
Test2DZeroFloatc                 C   ra   )Nr   r   r   rS   rc   rd   r   r#   r   r   r   r	     rh   zTest2DZeroFloat.setUpc                 C   r   r   rM   Nr    r#   r   r   r   r$     r%   zTest2DZeroFloat.test_ndimc                 C   r   )r(   r   Nr)   r#   r   r   r   r+     r%   zTest2DZeroFloat.test_shapec                 C   (   |  | j| jtjddg df  dS z see issue #473 N)r   r   rM   r   r#   r   r   r   rO        (zTest2DZeroFloat.test_indexlistN)r]   r^   r_   r	   r$   r+   rO   r   r   r   r   r     s
    r   c                   @   sD   e Zd Zdd Zdd Zdd Zdd Zd	d
 Zdd Zdd Z	dS )Test2DFloatc                 C   ra   )NrN   r   r   rS   rc   rd   r   r#   r   r   r   r	   &  rh   zTest2DFloat.setUpc                 C   r   r   r    r#   r   r   r   r$   +  r%   zTest2DFloat.test_ndimc                 C   r   )rj      Nr,   r#   r   r   r   r.   /  r%   zTest2DFloat.test_sizec                 C   s   |  | jjd| jjj  dS )r/   r   Nrm   r#   r   r   r   r1   3  s   zTest2DFloat.test_nbytesc                 C   r   )r(   r   Nr)   r#   r   r   r   r+   7  r%   zTest2DFloat.test_shapec                 C   r   r   r   r#   r   r   r   rO   ;  r   zTest2DFloat.test_indexlistc                 C   s<   |  | j| jtjd d g f  |  | j| jtjg   d S r   r   r#   r   r   r   test_index_emptylist?  s    z Test2DFloat.test_index_emptylistN)
r]   r^   r_   r	   r$   r.   r1   r+   rO   r   r   r   r   r   r   $  s    r   c                   @   s.   e Zd Zdd Zeejdk ddd ZdS )TestVeryLargeArrayc                 C   s    t |  | jjddd| _d S )Nrc   )i   i   )r*   )r   r	   r   rg   r   r#   r   r   r   r	   F  s   
zTestVeryLargeArray.setUp        z&Maximum integer size >= 2**31 requiredc                 C   s   |  | jjd d S )Nr   r,   r#   r   r   r   r.   J  r%   zTestVeryLargeArray.test_sizeN)	r]   r^   r_   r	   r   ZskipIfsysmaxsizer.   r   r   r   r   r   D  s    r   c              
   C   st   t jt jj}|d |t jj t t j| j	dt j
jt jd|}tj|d d tdtj d S )N)r      a)rN   r   )r
   Zh5pr   ZDATASET_CREATEZ	set_chunkZset_fill_timer   ZFILL_TIME_NEVERr   r   r   Z
IEEE_F64LEr   Zcreate_simpler   testingZassert_array_equalZzerosZfloat64)Zwritable_fileZdcplZdsr   r   r   test_read_no_fill_valueO  s   

$r   c                       s0   e Zd ZdZ fddZdd Zdd Z  ZS )TestBoolIndexz0
    Tests for indexing with Boolean arrays
    c                    s6   t    tddd| _| jjd| jd| _d S )N	   r   r   rc   rd   )	superr	   r   r   Zreshapearrr   rg   r   r#   	__class__r   r   r	   `  r   zTestBoolIndex.setUpc                 C   s,   t jg dd d f }| | j| j| d S )N)FTF)r   r   r   r   r   r   selr   r   r   test_select_first_axise  s   z$TestBoolIndex.test_select_first_axisc                 C   sP   t jg dd d f }| t | j|  W d    d S 1 s!w   Y  d S )N)FTFF)r   r   r=   r   r   r   r   r   r   test_wrong_sizei  s   "zTestBoolIndex.test_wrong_size)r]   r^   r_   __doc__r	   r   r   __classcell__r   r   r   r   r   \  s
    r   )r   r   numpyr   r
   commonr   r   r   r`   r~   r   Z
skipUnlessversionZhdf5_version_tupler   r   r   r   r   r   r   r   r   r   r   <module>   s&   	"ACIE8  