U
|
O ±d ã @ s® d dl Z d dlZd dlZd dlZd dlmZ d dlmZmZ d dl m
|
Z
|
m Z m Z m Z mZmZmZmZmZ d dl mZmZmZ d dlmZmZmZmZmZm Z G dd dZ dS ) é N)ÚPath)ÚNamedTemporaryFileÚ TemporaryFile) ÚmemmapÚsumÚaverageÚproductÚndarrayÚisscalarÚaddÚsubtractÚmultiply)ÚarangeÚallcloseÚasarray)Úassert_Ú assert_equalÚassert_array_equalÚsuppress_warningsÚIS_PYPYÚ break_cyclesc @ sÊ e Zd Zdd Zdd Zdd Zdd Zd d
|
Zd d Zd d Z dd Z
|
dd Z e j jejdkdddd Zdd Zdd Zd d Zd d Zd d! Zd"d# Zd$d% Zd&d' Zd(d) Zd*d+ Zd,d- Z d.S )/Ú
|
TestMemmapc C s: t dd| _d| _d| _td| jd| _| j | j¡ d S )NÚmmap)Úprefix)é é Úfloat32é ©Údtype)r ÚtmpfpÚshaper r ÚdataÚresize©Úself© r& úSd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\numpy/core/tests/test_memmap.pyÚ setup_method s
|
zTestMemmap.setup_methodc C s$ | j ¡ d | _tr t t d S )N)r Úcloser" r r r$ r& r& r' Úteardown_method s
|
|
zTestMemmap.teardown_methodc C sr t | j| jd| jd}| jd d
|d d
< ~t | j| jd| jd}tt| j| t| j| t|j j
|
d d S )Núw+©r Úmoder! ÚrF) r r r r! r" r r r r ÚflagsZ writeable)r% ÚfpZnewfpr& r& r' Útest_roundtrip s ÿ ÿ zTestMemmap.test_roundtripc C s8 |d }t || jd| jd}| jd d
|d d
< ~d S )Nr r+ r, )r r r! r" )r% Útmp_pathÚtmpnamer0 r& r& r' Útest_open_with_filename. s
|
ÿz"TestMemmap.test_open_with_filenamec C s* t }t|| j| jd}~W 5 Q R X d S ©N©r r! )r r r r! )r% Úfr0 r& r& r' Útest_unnamed_file5 s z TestMemmap.test_unnamed_filec C s> d}d}t | j| j|| j|d}t||j t||j ~d S )Né r+ )r r- r! Úoffset)r r r r! r r: r- )r% r: r- r0 r& r& r' Útest_attributes: s ÿ zTestMemmap.test_attributesc C sn |d }t || jd| jd}ttj |¡}| jd d
|d d
< t||j |d d
}t||j ~~d S ©Nr r+ r, r9 )
|
r r r! r ÚosÚpathÚabspathr" r Úfilename©r% r2 r3 r0 r? Úbr& r& r' Ú test_filenameC s
|
ÿ zTestMemmap.test_filenamec C s |d }t t|| jd| jd}tt| ¡ }| jd d
|d d
< t|t|j ¡ |d d
}t|t|j ¡ ~~d S r< ) r r r r! ÚstrÚresolver" r r@ rA r& r& r' Ú test_pathO s ÿ zTestMemmap.test_pathc C s* t | j| jd| jd}t|j| jj d S )Nr+ r, )r r r r! r r@ Úname©r% r0 r& r& r' Útest_filename_fileobj^ s ÿz TestMemmap.test_filename_fileobjZgnu0zKnown to fail on hurd)Úreasonc C sL t | j| jd| jd}| jd d
|d d
< t|d | jd | ¡ d S )Nr+ r, r )r r r r! r" r ÚflushrH r& r& r' Ú
|
test_flushc s ÿzTestMemmap.test_flushc C sb t | j| jd| jd}d|d< |dd
}t|d d ~t|d d d|d< t|d d d S )Nr+ r, é r r9 é )r r r r! r )r% Zfp_baseZfp_viewr& r& r' Útest_dell s ÿ zTestMemmap.test_delc C s< t | j| jd| jd}|d }t|t r8t|j|jk d S )Nr+ r, é
|
©r r r r! Ú
|
isinstancer Ú_mmap©r% r0 Útmpr& r& r' Ú test_arithmetic_drops_referencesz s ÿ
|
z+TestMemmap.test_arithmetic_drops_referencesc C s< t | j| jd| jd}|d }t|t r8t|j|jk d S )Nr+ r, ))r9 é )rW r rQ rT r& r& r' Ú test_indexing_drops_references s ÿ
|
z)TestMemmap.test_indexing_drops_referencesc C s: t | j| jd| jd}t|d d
d d
f j|jk d S )Nr+ r, rW )r r r r! r rS rH r& r& r' Ú test_slicing_keeps_references s ÿz(TestMemmap.test_slicing_keeps_referencesc C sZ t | j| j| jd}| ¡ }| ¡ }t|j|k t|j|k t|}t|j|k d S r5 )r r r r! Úviewr Úbaser )r% r0 Znew1Znew2Z new_arrayr& r& r' Ú test_view s zTestMemmap.test_viewc C s0 t | j| j| jd}| j|d d
< t x}| td¡ tt t
|
fD ]Z}||}t t | t |j | jd j k t ||ddj tk t ||ddj tk q@W 5 Q R X tttfD ]D}t ||| jj tk t || j|j tk t |||j tk q°|d7 }|j t kstt|d|d |j t ks,td S )Nr6 z&np.average currently does not preserve)r r r ©Zaxisr9 )Úout)r r r r! r" r ÚfilterÚ FutureWarningr r r r r
|
Ú __class__r r r r ÚAssertionError)r% r0 ÚsupZunary_opÚresultZ binary_opr& r& r' Útest_ufunc_return_ndarray s$ "z$TestMemmap.test_ufunc_return_ndarrayc C sZ t | j| j| jd}| j|d d
< t|dd
d d
f jt k t|ddg jtk d S )Nr6 r9 éÿÿÿÿr )r r r r! r" r ra r rH r& r& r' Ú test_getitem® s zTestMemmap.test_getitemc C s G dd dt }|| j| j| jd}| j|d d
< tt|ddj|k tt|j|k t|dd
d d
f j|k |ddg j|kstd S )Nc @ s e Zd ZdS )z7TestMemmap.test_memmap_subclass.<locals>.MemmapSubClassN)Ú__name__Ú
|
__module__Ú __qualname__r& r& r& r' ÚMemmapSubClass· s rk r6 r r] r9 rf ) r r r r! r" r r ra rb )r% rk r0 r& r& r' Útest_memmap_subclass¶ s z TestMemmap.test_memmap_subclassc C s8 dt j }t jd }t| j|d|d}t|j|k d S )NrM r9 r+ )r! r- r: )r ZALLOCATIONGRANULARITYr r r r: )r% Úsizer: r0 r& r& r' Ú4test_mmap_offset_greater_than_allocation_granularityÄ s
|
|
z?TestMemmap.test_mmap_offset_greater_than_allocation_granularityc C s* | j d¡ t| j dd}t|jd d S )Ns aaaaaaaaaaaaaaaaÚfloat64r )rW )r Úwriter r r! )r% Úmmr& r& r' Ú test_no_shapeÊ s zTestMemmap.test_no_shapec C sJ t jtdd t| jddd W 5 Q R X | j d¡ t| jddd d S )Nz
|
empty file)Úmatch)r r r+ )r! r- ó )ÚpytestZraisesÚ
|
ValueErrorr r rp r$ r& r& r' Útest_empty_arrayÏ s zTestMemmap.test_empty_arrayN) rh ri rj r( r* r1 r4 r8 r; rC rF rI ru ÚmarkZskipifÚsysÚplatformrL rO rV rX rY r\ re rg rl rn rr rw r& r& r& r' r s0 ÿ
|
r ) ry r= r ru Úpathlibr Útempfiler r Únumpyr r r r r r
|
r r r r r r Z numpy.testingr r r r r r r r& r& r& r' Ú<module> s ,
|
