Your IP : 216.73.216.213
��Yf�E � @ s� d Z d d l Z d d l Z d d l Z d d l Z d d l Z d d l Z d d l Z d d l Z d d l
Z
Gd d � d � Z Gd d � d e � Z Gd d � d e � Z
Gd d
� d
� Z Gd d � d � Z Gd
d � d � Z Gd d � d e j e e � Z Gd d � d e j e e
� Z Gd d � d e j e e � Z Gd d � d e j e e
� Z Gd d � d e j e e � Z Gd d � d e j e e
� Z Gd d � d e j � Z e d k r�e j � d S) z�Unit tests for the memoryview
Some tests are in test_bytes. Many tests that require _testbuffer.ndarray
are in test_buffer.
� Nc @ sZ e Z d Z d Z e d d � � Z e 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 d0 d1 � Z d2 d3 � Z d4 d5 � Z d6 S)7�AbstractMemoryTestss abcdefc C s | j S)N)� source_bytes)�self� r �9/opt/alt/python35/lib64/python3.5/test/test_memoryview.py�_source s zAbstractMemoryTests._sourcec C s t d | j | j g � S)N)�filter�ro_type�rw_type)r r r r �_types s zAbstractMemoryTests._typesc s� | | j � } t j | � } | j | � � | j � d t d � � | j � d t � | j � d t d � � | j � d t d � � | j � d t d � � | j t � f d d � � | j t � f d d � � | j t � f d
d � � | j t � f d d � � | j t
� f d d � � | j t
� f d
d � � | j t
� f d d � � d � | j t j | � | � d S)Nr � a� � f� � c s � d S)Nr r r )�mr r �<lambda>' s z=AbstractMemoryTests.check_getitem_with_type.<locals>.<lambda>c s � d S)N� i����r r )r r r r ( s c s � t j S)N)�sys�maxsizer )r r r r ) s c s � t j S)N)r r r )r r r r * s c s � d S)Nr r )r r r r , s c s � d S)Ng r r )r r r r - s c s � d S)N�ar r )r r r r . s ���i����) r r � getrefcount�_view� assertEqual�ord�assertIsInstance�int� assertRaises�
IndexError� TypeError)r �tp�b� oldrefcountr )r r �check_getitem_with_type s" z+AbstractMemoryTests.check_getitem_with_typec C s% x | j D] } | j | � q
Wd S)N)r r$ )r r! r r r � test_getitem2 s z AbstractMemoryTests.test_getitemc sk xd | j D]Y } | | j � } | j | � � | j t � � � f d d � t t � � � D� � q
Wd S)Nc s g | ] } � | � q Sr r )�.0�i)r r r �
<listcomp>: s z1AbstractMemoryTests.test_iter.<locals>.<listcomp>)r r r r �list�range�len)r r! r" r )r r � test_iter6 s z AbstractMemoryTests.test_iterc s� | j s | j d � | j | j � } t j | � } | j | � � � f d d � } | j t | d � | j t | d � | j t | t d � � d � | j t j | � | � d S)Nzno read-only type to testc s | � d <d S)Nr r )�value)r r r �setitemB s z:AbstractMemoryTests.test_setitem_readonly.<locals>.setitemr �A )
r �skipTestr r r r r r �
memoryviewr )r r" r# r. r )r r �test_setitem_readonly< s
z)AbstractMemoryTests.test_setitem_readonlyc s� | j s | j d � | j � | j | j � } t j | � } | j | � � t d � � d <| j � | d � � d � � d d � <| j � | d � � d � � d d � <| j � | d
� � d � � d d � <| j � | d
� � d � � d d � <| j � | d � � d
d � � d d � <| j � | d � � d � � d d � <� d d � � d
d � <| j � | d � � � f d d � } | j t | d d � | j t | d d � | j t | t j
d � | j t | t j
d � | j t | d d � | j t | d d � | j t | t d d d � d f d � | j t | d t d d d � f d � | j t | d d � | j t | d d � t d d d � t d d d
� f } | j t
| | d � � j d k r�t n t } | j | | d d � | j | | d d � | j t | t d d � d � | j t | t d d
� d � d � | j t j | � | � d S) Nzno writable type to test� 1r s 1bcdef� 0r s 0bcdefs 12� s 012def� s abcdef� r
s cdedefs ababcfc s � | � � | <d S)Nr )�keyr- )r r! r r r. c s z:AbstractMemoryTests.test_setitem_writable.<locals>.setitemr r r g r �cs abi����)r )r )r
r0 r r r r r �_check_contentsr r r r �slice�NotImplementedError�format�
ValueErrorr )r r" r# r. Zslices�excr )r r! r �test_setitem_writableJ sR
(($ z)AbstractMemoryTests.test_setitem_writablec C sx xq | j D]f } | | j � } | j | � } | j t � � | d =Wd QRX| j t � � | d d � =Wd QRXq
Wd S)Nr � )r r r r r )r r! r" r r r r � test_delitem~ s z AbstractMemoryTests.test_delitemc s~ xw � j D]l } � j | � j � � } | j � } d j � f d d � d D� � } � j | | � � j | t � q
Wd S)Nr6 c 3 s' | ] } � j t | g � � Vq d S)N)� getitem_type�bytes)r& r9 )r r r � <genexpr>� s z3AbstractMemoryTests.test_tobytes.<locals>.<genexpr>s abcdef)r r r �tobytes�joinr r rD )r r! r r" Zexpectedr )r r � test_tobytes� s z AbstractMemoryTests.test_tobytesc C sR xK | j D]@ } | j | | j � � } | j � } | j | t d � � q
Wd S)Ns abcdef)r r r �tolistr r) )r r! r �lr r r � test_tolist� s z AbstractMemoryTests.test_tolistc s- x&| j D]} | j | | j � � � x� | j D]� } | j � | d � k � | j � | d � k � | j � | d � k � | j � | d � k � | j � | d � k � | j � | d � k � q2 W| j � � k � | j � � d d � k � | j � d d � � d d � k � | j � d d � � k � | j � d k � | j � d k � | j d � k � | j d � k � x� � d f D]� � | j t � � f d d � � | j t � � f d
d � � | j t � � f d d � � | j t � � f d d � � q�Wq
Wd S)
Ns abcdefs abcdes abcde1r r r
Zabcdefc s
� � k S)Nr r )r9 r r r r � s z2AbstractMemoryTests.test_compare.<locals>.<lambda>c s
� � k S)Nr r )r9 r r r r � s c s
� � k S)Nr r )r9 r r r r � s c s
� � k S)Nr r )r9 r r r r � s )r r r �
assertTrueZ assertFalser r )r r! Ztp_compr )r9 r r � test_compare� s, ' z AbstractMemoryTests.test_comparec C s� | j | | j � � } | j | j | j � | j | j | j � | j | j d � | j | j d � | j t | � d � | j | j | j f � | j | j f � | S)Nr r )r )
r r r r= �itemsize�ndim�shaper+ �strides�
suboffsets)r r! r r r r �check_attributes_with_type� s z.AbstractMemoryTests.check_attributes_with_typec C s? | j s | j d � | j | j � } | j | j d � d S)Nzno read-only type to testT)r r0 rS r �readonly)r r r r r �test_attributes_readonly� s
z,AbstractMemoryTests.test_attributes_readonlyc C s? | j s | j d � | j | j � } | j | j d � d S)Nzno writable type to testF)r
r0 rS r rT )r r r r r �test_attributes_writable� s
z,AbstractMemoryTests.test_attributes_writablec C s� x� | j D]� } | | j � } t j | � } | j | � } t j | � } t | d � } | j | | | j d � � | j t j | � | � d } | j t j | � | � q
Wd S)Nzutf-8) r r r r r �strr: �encoder )r r! r" r# r Zoldviewrefcount�sr r r �test_getbuffer� s z"AbstractMemoryTests.test_getbufferc C sQ xJ| j D]?} t | t � s" q
Gd d � d � } Gd d � d | � } Gd d � d � } | | d � � } | j | � } | � } | | _ | | _ t j | � } d } } } t j � | j
| � d k | � � | | d � � } | � } | | _ | | _ t j | � } d } } t j � | j
| � d k | � � q
Wd S)Nc @ s e Z d Z d d � Z d S)z+AbstractMemoryTests.test_gc.<locals>.MyViewc S s t | � | _ d S)N)r1 r )r �baser r r �__init__� s z4AbstractMemoryTests.test_gc.<locals>.MyView.__init__N)�__name__�
__module__� __qualname__r\ r r r r �MyView� s r` c @ s e Z d Z d S)z-AbstractMemoryTests.test_gc.<locals>.MySourceN)r] r^ r_ r r r r �MySource� s ra c @ s e Z d Z d S)z-AbstractMemoryTests.test_gc.<locals>.MyObjectN)r] r^ r_ r r r r �MyObject� s rb s abc) r �
isinstance�typer r �o�weakref�ref�gcZcollectrL �x) r r! r` ra rb r" r re �wrr r r �test_gc� s.
zAbstractMemoryTests.test_gcc g C s� | j t d � } | � t | � Wd QRX| � | j � Wd QRX| � | j � Wd QRX| �
| d Wd QRX| � d | d <Wd QRX| � t | � Wd QRX| � | j Wd QRX| � | j Wd QRX| � | j Wd QRX| � | j Wd QRX| � | j
Wd QRX| � | j Wd QRX| � | � Wd QRXWd QRX| j d t
| � � | j d t | � � | j | | � | j | t | | j � � � | j | | | j � � d S)NZreleasedr � xzreleased memory)ZassertRaisesRegexr> rD rF rI r+ r= rN rO rT rP rQ ZassertInrW �reprr ZassertNotEqualr1 r )r r r! Zcheckr r r �_check_released� sB z#AbstractMemoryTests._check_releasedc C s� x� | j D]{ } | | j � } | j | � } | � } | j | | � Wd QRX| j | | � | j | � } | � | j � Wd QRXq
Wd S)N)r r r �assertIsrn �release)r r! r" r �cmr r r �test_contextmanager s z'AbstractMemoryTests.test_contextmanagerc C sj xc | j D]X } | | j � } | j | � } | j � | j | | � | j � | j | | � q
Wd S)N)r r r rp rn )r r! r" r r r r � test_release# s
z AbstractMemoryTests.test_releasec C si | j } | d k r" | j d � | | j � } | j | � } t j d � } | j t | j | � d S)Nzno read-only type to tests ZZZZ) r r0 r r �io�BytesIOr r �readinto)r r! r" r r' r r r �test_writable_readonly- s
z*AbstractMemoryTests.test_writable_readonlyc C s | j t | j i � d S)N)r r r )r r r r �test_getbuf_fail8 s z$AbstractMemoryTests.test_getbuf_failc C s� | j } | d k r" | j d � | | j � } | j | � } | j t | � t d � � | j � | j t | � t d � � | j | � } | j � | j t t | � d S)Nzno read-only type to tests abcdef) r r0 r r r �hashrp r r> )r r! r" r r r r � test_hash; s
z AbstractMemoryTests.test_hashc C sW | j } | d k r" | j d � | | j � } | j | � } | j t t | � d S)Nzno writable type to test)r
r0 r r r r> ry )r r! r" r r r r �test_hash_writableM s
z&AbstractMemoryTests.test_hash_writablec s� x� | j D]� } | | j � } | j | � } g � | � f d d � } t j | | � } | j | � | � ~ t j j � | j | � d � | j � d | � q
Wd S)Nc s � j | � d S)N)�append)rj r" )�Lr r �callback\ s z2AbstractMemoryTests.test_weakref.<locals>.callbackr ) r r r rf rg ro �test�supportZ
gc_collect)r r! r" r r~ rj r )r} r � test_weakrefV s
z AbstractMemoryTests.test_weakrefc C s� x� | j D]� } | | j � } | j | � } t t | j � � � } | j t t | � � | � | j t t | � � t | d d d � � � q
Wd S)Nr r )r r r r) �reversedrI r )r r! r" r Zaslistr r r �
test_reversede s z!AbstractMemoryTests.test_reversedc C s t j d d d d d g � } t | � } | j d � } | j d � } | d d � } t | � } ~ | j | d d � | j | d d � | j | j d � | j | j d � | j d � } | j | d d � | j | d d � | j | j d � | j | j d � d S)N�H� �Br r7 �I)�arrayr1 �castr r= )r r ri r r" r9 �d�_r r r �test_issue22668m s z#AbstractMemoryTests.test_issue22668N) r] r^ r_ r �propertyr r r$ r% r, r2 r@ rB rH rK rM rS rU rV rZ rk rn rr rs rw rx rz r{ r� r� r� r r r r r s6 4
&
r c @ s. e Z d Z e Z e Z e Z d Z d Z d S)�BaseBytesMemoryTestsr r� N)
r] r^ r_ rD r � bytearrayr
rC rN r= r r r r r� � s
r� c @ s| e Z d Z d Z d d � Z d d � Z e j d � j Z d Z e j
d � d d � � Z e j
d � d
d � � Z d S) �BaseArrayMemoryTestsNc C s t j d t | � � S)Nr' )r� r) )r r" r r r r � s z BaseArrayMemoryTests.<lambda>c C s t j d t | � � j � S)Nr' )r� r) rF )r r" r r r r � s r' z/XXX test should be adapted for non-byte buffersc C s d S)Nr )r r r r rZ � s z#BaseArrayMemoryTests.test_getbufferz:XXX NotImplementedError: tolist() only supports byte viewsc C s d S)Nr )r r r r rK � s z BaseArrayMemoryTests.test_tolist)
r] r^ r_ r r
rC r� rN r= �unittest�skiprZ rK r r r r r� � s r� c @ s( e Z d Z d d � Z d d � Z d S)�BaseMemoryviewTestsc C s
t | � S)N)r1 )r �objr r r r � s zBaseMemoryviewTests._viewc C s | j | | | � � d S)N)r )r r! r� �contentsr r r r: � s z#BaseMemoryviewTests._check_contentsN)r] r^ r_ r r: r r r r r� � s r� c @ s: e Z d Z d Z d d � Z d d � Z d d � Z d S) �BaseMemorySliceTestss XabcdefYc C s t | � } | d d � S)Nr r )r1 )r r� r r r r r � s zBaseMemorySliceTests._viewc C s$ | j | d d � | | � � d S)Nr r )r )r r! r� r� r r r r: � s z$BaseMemorySliceTests._check_contentsc C sc x\ | j D]Q } t | | j � � } t j | � } | d d � | j t j | � | � q
Wd S)Nr r7 )r r1 r r r r )r r! r r# r r r � test_refs� s
z BaseMemorySliceTests.test_refsN)r] r^ r_ r r r: r� r r r r r� � s r� c @ s. e Z d Z d Z d d � Z d d � Z d S)�BaseMemorySliceSliceTestss XabcdefYc C s$ t | � } | d d � d d � S)Nr r )r1 )r r� r r r r r � s z BaseMemorySliceSliceTests._viewc C s$ | j | d d � | | � � d S)Nr r )r )r r! r� r� r r r r: � s z)BaseMemorySliceSliceTests._check_contentsN)r] r^ r_ r r r: r r r r r� � s r� c @ s e Z d Z d d � Z d S)�BytesMemoryviewTestc C s� x� | j D]� } | | j � } | j t | � � | j t d | � � | j t t � | j t t | | � | j t t d | �| j t t | d d �q
Wd S)N�objectZargumentT)r r rL r1 r r )r r! �obr r r �test_constructor� s z$BytesMemoryviewTest.test_constructorN)r] r^ r_ r� r r r r r� � s r� c @ s e Z d Z d d � Z d S)�ArrayMemoryviewTestc C sf t j d t d � � } t | � } t j d t d d d � � } | | d d � <| j | | � d S)Nr' �
� r r r )r� r* r1 r )r r r Znew_ar r r �test_array_assign� s
z%ArrayMemoryviewTest.test_array_assignN)r] r^ r_ r� r r r r r� � s r� c @ s e Z d Z d S)�BytesMemorySliceTestN)r] r^ r_ r r r r r� � s r� c @ s e Z d Z d S)�ArrayMemorySliceTestN)r] r^ r_ r r r r r� � s r� c @ s e Z d Z d S)�BytesMemorySliceSliceTestN)r] r^ r_ r r r r r� � s r� c @ s e Z d Z d S)�ArrayMemorySliceSliceTestN)r] r^ r_ r r r r r� � s r� c @ s@ e Z d Z d d � Z d d � Z d d � Z d d � Z d S)
� OtherTestc C sA t j j d � } t | j d � � } | j � } t | � j d � } | d d � | d d � <| d d � | d d � <| j | j d � x� d D]� } | j | � �� | j � } t | � j | � } t | � j | � d d � | d d � <t | � j | � d d � | d d � <| j | j d � Wd QRXq� Wd S)N�ctypesg333333�?r� r7 ZBbc)
r r� �
import_modulerD Zc_doubler1 r� r r- ZsubTest)r r� Zp6r� r r= r r r �test_ctypes_cast� s
))zOtherTest.test_ctypes_castc C sG d d } t | � } | d d d � } | j | j � d d � d S)Nr4 i@
r Z30r )r1 r �hex)r ri �m1�m2r r r �test_memoryview_hex s
z OtherTest.test_memoryview_hexc C s4 t d � } | j t � � t j | � Wd QRXd S)Ns abc)r1 r r �copy)r r r r r � test_copy s zOtherTest.test_copyc C sU t d � } xB t t j d � D]- } | j t � � t j | | � Wd QRXq Wd S)Ns abcr )r1 r* �pickleZHIGHEST_PROTOCOLr r �dumps)r r �protor r r � test_pickle s zOtherTest.test_pickleN)r] r^ r_ r� r� r� r� r r r r r� � s r� �__main__)�__doc__r� Z test.supportr r rh rf r� rt r� r� r r� r� r� r� r� ZTestCaser� r� r� r� r� r� r� r] �mainr r r r �<module> s> � v
'