Your IP : 216.73.216.213
(��^a. � @ s� d d l m Z m Z m Z d d l Z d d l Z d d l Z d d l Z d d l m Z d d l
Z
d d l Z d d l Z d d l
m Z d d l m Z d d l Z d d l j Z d d l m Z m Z d d l m Z m Z m Z m Z d d l m Z d d l m Z m Z m! Z! m" Z" m# Z# m$ Z$ m% Z% m& Z& m' Z' m( Z( m) Z) m* Z* Gd
d � d e � Z+ e j, d d � \ Z- Z. e j/ d d
k Z0 d d d � Z1 Gd d � d e2 � Z3 Gd d � d e3 e � Z4 Gd d � d e3 e � Z5 Gd d � d e � Z6 Gd d � d e � Z7 Gd d � d e � Z8 Gd d � d e � Z9 Gd d � d e � Z: d d! � Z; d" d# � Z< d$ d% � Z= d& d' � Z> d( d) � Z? e@ d* k r�e � d S)+� )�division�absolute_import�print_functionN)�NamedTemporaryFile)�BytesIO)�datetime)�ConverterError�ConversionWarning)�asbytes�bytes�unicode�Path)� assert_equal) �TestCase�run_module_suite� assert_warns�assert_�assert_raises_regex�
assert_raises�assert_allclose�assert_array_equal�temppath�dec�IS_PYPY�suppress_warningsc @ s= e Z d Z d Z d d d � Z d d � Z d d � Z d S)
�TextIOz�Helper IO class.
Writes encode strings to bytes if needed, reads return bytes.
This makes it easier to emulate files opened in binary mode
without needing to explicitly convert strings to bytes in
setting up the test data.
� c C s t j | t | � � d S)N)r �__init__r
)�self�s� r ��/builddir/build/BUILDROOT/alt-python35-numpy-1.13.3-2.el8.x86_64/opt/alt/python35/lib64/python3.5/site-packages/numpy/lib/tests/test_io.pyr # s zTextIO.__init__c C s t j | t | � � d S)N)r �writer
)r r r r r! r" & s z TextIO.writec C s! t j | d d � | D� � d S)Nc S s g | ] } t | � � q Sr )r
)�.0r r r r! �
<listcomp>* s z%TextIO.writelines.<locals>.<listcomp>)r �
writelines)r �linesr r r! r% ) s zTextIO.writelinesN)�__name__�
__module__� __qualname__�__doc__r r" r% r r r r! r s r � � c C s` t j d d k r< t t j | j d � | � d d � � St t j | | � d d � � Sd S)zY
This function is available in the datetime module only from Python >=
2.5.
r � �latin1N)�sys� version_infor �time�strptime�decode)r �fmtr r r! r2 1 s )r2 c @ s� e Z d Z d d � Z d d � Z d d � Z d d � Z d d
� Z e j j
j e j
d k d � d
d � � Z d d � Z e
j d d � � Z d S)�
RoundtripTestc O s< | j d i � } | j d i � } | j d d � } | rW t d d � } | j } n t � } | } z~ | } | | | | � | j � | j d � t j d k r� t | t � r� | j � t
j | | � }
| | _ |
| _
Wd t | t � s7| j � d t � k r7t |
t
j j j � s7t j | j � Xd S)
a�
save_func : callable
Function used to save arrays to file.
file_on_disk : bool
If true, store the file on disk, instead of in a
string buffer.
save_kwds : dict
Parameters passed to `save_func`.
load_kwds : dict
Parameters passed to `numpy.load`.
args : tuple of arrays
Arrays stored to file.
� save_kwds� load_kwds� file_on_diskF�deleter �win32N� arr_reloaded)�getr �namer �flush�seekr/ �platform�
isinstance�close�np�load�arrr; �locals�lib�npyio�NpzFile�os�remove) r Z save_func�args�kwargsr6 r7 r8 � target_fileZ load_filerE r; r r r! � roundtrip> s.
zRoundtripTest.roundtripc C s� | j | � | j | d d �| j t j | � � | j t j | � d d �| j d d k r� | j t j | � d d � � | j t j | � d d � d d �d S)Nr8 Tr � )rO rC �asfortranarray�shape)r �ar r r! �check_roundtripsn s
z RoundtripTest.check_roundtripsc C s� t j g t � } | j | � t j d d g d d g g t � } | j | � t j d d g d d g g t � } | j | � t j d
d g d d
g g d t j �} | j | � t j d d g d d g g d t j �} | j | � d S)NrP r+ r- � y @y @y @y @�dtypey �? @y @ @y @ @y @ @y �? @y @ @y @ @y @ @)rC �array�floatrT �int�csingle�cdouble)r rS r r r! �
test_arrayx s
$
$
*
*zRoundtripTest.test_arrayc C sT t j g t � } | j | � t j d d g d d g g t � } | j | � d S)NrP r+ r- rU )rC rW �objectrT )r rS r r r! �test_array_object� s
$z RoundtripTest.test_array_objectc C s/ t j d d d d g t � } | j | � d S)NrP r+ r- rU )rC rW rY rO )r rS r r r! �test_1D� s zRoundtripTest.test_1Dr: z
Fail on Win32c C s� t j d d g d d g g � } | j | d d d d d i �t j d d g d d g g � } | j | d d d d d i �d S)
NrP g @rU g333333 @r8 Tr7 � mmap_mode�r)rC rW rO rQ )r rS r r r! � test_mmap� s ! !zRoundtripTest.test_mmapc C s2 t j d d
g d d d g �} | j | � d S)
NrP r+ r- rU rV �x�i4�y)rP r+ )r- rU )rc rd )re rd )rC rW rT )r rS r r r! � test_record� s !zRoundtripTest.test_recordc C sl d d � t d � D� } t j d d | �} t j d d � �% t j d d t � | j | � Wd QRXd S)
Nc S s$ g | ] } d | d t f � q S)z%d�d )rX )r# �ir r r! r$ � s z1RoundtripTest.test_format_2_0.<locals>.<listcomp>i� i� rV �recordT�alwaysr )�rangerC �ones�warnings�catch_warnings�filterwarnings� UserWarningrT )r �dtrS r r r! �test_format_2_0� s
z RoundtripTest.test_format_2_0N)r' r( r) rO rT r\ r^ r_ rC �testingr �knownfailureifr/ r@ rb rf �slowrr r r r r! r5 = s 0
- r5 c @ s e Z d Z d d � Z d S)� TestSaveLoadc O st t j | t j | | � t | j d | j � t | j d j | j j � t | j d j j | j j j � d S)Nr )
r5 rO rC �saver rE r; rV �flags�fnc)r rL rM r r r! rO � s zTestSaveLoad.roundtripN)r' r( r) rO r r r r! rv � s rv c @ s� e Z d Z d d � Z e j j j e d � e j j j d d � � � Z
d d � Z d d � Z d
d � Z
d d
� Z d d � Z e j j j e d � d d � � Z d d � Z d S)�
TestSavezLoadc O s� t j | t j | | � zn xg t | j � D]V \ } } | j d | } t | | � t | j | j � t | j j
| j j
� q, WWd | j j r� | j j j � t
j | j j j � Xd S)Nzarr_%d)r5 rO rC �savez� enumeraterE r; r rV rx ry �fidrB rJ rK r= )r rL rM �nrE Zreloadedr r r! rO � s
! zTestSavezLoad.roundtripz Works only with 64bit systemsc C s} d } t j | d t j �} t d d d d � �C } t j | d | �~ t j | � } | d } | j � ~ Wd QRXd S) NrP � i�� rV �prefixZnumpy_test_big_arrays_�suffixz.npzrS l l � )rC �empty�uint8r r{ rD rB )r �LrS �tmpZnpfiler r r! �test_big_arrays� s
z TestSavezLoad.test_big_arraysc C s\ t j d d g d d g g t � } t j d d
g d d g g t � } | j | | � d S)
NrP r+ r- rU y @y @y @y (@y �? @y @ @y @ �y @ (@)rC rW rX �complexrO )r rS �br r r! �test_multiple_arrays� s $$z"TestSavezLoad.test_multiple_arraysc C s� t j d d g d d g g t � } t j d d
g d d g g t � } t � } t j | d | d
| �| j d � t j | � } t | | d � t | | d
� d S)NrP r+ r- rU y @y @y @y (@�file_a�file_br y �? @y @ @y @ �y @ (@) rC rW rX r� r r{ r? rD r )r rS r� �c�lr r r! �test_named_arrays� s $$
z TestSavezLoad.test_named_arraysc C s� t j d d g d d g g t � } t j d d
g d d g g t � } t � } t j | d | d
| �| j d � t j | � } t t t
| j � � d d
g � t | | j j � t | | j j
� d S)NrP r+ r- rU y @y @y @y (@r� r� r y �? @y @ @y @ �y @ (@)rC rW rX r� r r{ r? rD r �sorted�dir�fr� r� )r rS r� r� r� r r r! � test_BagObj� s $$
"zTestSavezLoad.test_BagObjc s� d d � � g � � � f d d � t d � D� } x | D] } | j � q; Wx | D] } | j � qV W� r| t � � � d S)Nc S s~ t d d � �h } t j j d d � } y t j | d | �Wn2 t k
rr } z | j | � WYd d } ~ Xn XWd QRXd S)Nr� z.npzi� rE )r rC �random�randnr{ �OSError�append)Z
error_listr� rE �errr r r! �writer� s z9TestSavezLoad.test_savez_filename_clashes.<locals>.writerc s+ g | ]! } t j d � d � f � � q S)�targetrL )� threading�Thread)r# �j)�errorsr� r r! r$ � s z=TestSavezLoad.test_savez_filename_clashes.<locals>.<listcomp>r- )rk �start�join�AssertionError)r �threads�tr )r� r� r! �test_savez_filename_clashes� s
z)TestSavezLoad.test_savez_filename_clashesc C s� t d d � �� } t | d � � } t j | d d �Wd QRXt | d d � �\ } | j d � t | j � t j | � d t | j � | j d � t | j � Wd QRXWd QRXd S) Nr� z.npz�wb�dataz LOVELY LOAD�rbi' r )r �openrC r{ r? r �closedrD )r r� �fpr r r! �test_not_closing_opened_fid� s
z)TestSavezLoad.test_not_closing_opened_fidz context manager required on PyPyc C s� t d d � �� } t j | d d �t � �� } | j t � xj t d d � D]Y } y t j | � d WqN t k
r� } z d | } t | � � WYd d } ~ XqN XqN WWd QRXWd QRXd S)Nr� z.npzr� z LOVELY LOADrP i z#Failed to load data from a file: %s)
r rC r{ r �filter�Warningrk rD � Exceptionr� )r r� �suprh �e�msgr r r! �test_closing_fid s
z TestSavezLoad.test_closing_fidc C sn d } t d d d | � �L } t j | d d �t j | � } | j j } | j � t | j � Wd QRXd S)NZ&numpy_test_closing_zipfile_after_load_r� z.npzr� �labz place holder) r rC r{ rD �zipr� rB r r� )r r� r� r� r� r r r! � test_closing_zipfile_after_load% s
z-TestSavezLoad.test_closing_zipfile_after_loadN)r' r( r) rO rC rs r �skipif�IS_64BITru r� r� r� r� r� r� r r� r� r r r r! rz � s
$rz 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 S)� TestSaveTxtc C s t j d d g d d g g t � } d } t � } t j | | d | �| j d � t | j � t | d | d d
� t | d | d d � g � t j d d g d d g g t � } t � } t j | | d d
�| j d � t | j � d d g � d S)NrP r+ r- rU z%.18er4 r � �
z%ds 1 2
s 3 4
)rP r+ )r- rU )
rC rW rX r �savetxtr? r � readlinesr
rY )r rS r4 r� r r r! r\ 4 s $
$
zTestSaveTxt.test_arrayc C ss t j d d d d g t � } t � } t j | | d d �| j d � | j � } t | d d d
d g � d S) NrP r+ r- rU r4 z%dr s 1
s 2
s 3
s 4
)rC rW rY r r� r? r� r )r rS r� r& r r r! r_ D s
zTestSaveTxt.test_1Dc C sj t j d d g d d d g �} t � } t j | | d d
�| j d � t | j � d d
g � d S)NrP r+ r- rU rV rc rd re r4 z%dr s 1 2
s 3 4
)rP r+ )r- rU )rc rd )re rd )rC rW r r� r? r r� )r rS r� r r r! rf L s
!
zTestSaveTxt.test_recordc C sp t j d d g d d g g � } t � } t j | | d d d d �| j d � t | j � d
d g � d S) Ng �?g @g @g @� delimiter�,r4 z%dr s 1,2
s 3,4
)rC rW r r� r? r r� )r rS r� r r r! �test_delimiterS s
!
zTestSaveTxt.test_delimiterc C s% t j d d g � } t � } t j | | d d d g �| j d � t | j � d d
g � t � } t j | | d d �| j d � | j � } t | d d
g � t � } t j | | d d d d �| j d � | j � } t | d d
g � t � } t t t j | | d d �d S)NrP r+ r- rU r4 z%02dz%3.1fr s 01 2.0
s 03 4.0
z %02d : %3.1fs 01 : 2.0
s 03 : 4.0
r� r� �c )rP r+ )r- rU ) rC rW r r� r? r r� r �
ValueError)r rS r� r& r r r! � test_formatZ s"
zTestSaveTxt.test_formatc C s� t � } t j d d g d t j �} d } t j | | d d d | �| j d
� t | j � t d | d � � t � } t j | | d d d
| �| j d
� t | j � t d | d � � t � } d } t j | | d d d | d | �| j d
� t | j � t | | d d � � t � } d } t j | | d d d
| d | �| j d
� t | j � t d | | d � � d S)NrP r+ r- rU rV zTest header / footerr4 z%1d�headerr z# z
1 2
3 4
�footerz
1 2
3 4
# r� z% �commentsz1 2
3 4
)rP r+ )r- rU ) r rC rW rY r� r? r �readr
)r r� rS �test_header_footerZ
commentstrr r r! r� t s4
z TestSaveTxt.test_header_footerc C sX t � �H } t j d d g � } t j | | � t j | � } t | | � Wd QRXd S)NrP r+ r- rU )rP r+ )r- rU )r rC rW r� �loadtxtr )r r= rS r� r r r! �test_file_roundtrip� s
z TestSaveTxt.test_file_roundtripc C sD d } d } t j | | f d t j �} t j } t j } | d | | d d � <t � } t j | | d d �| j d � | j � } t | d d g � t � } t j | | d d | �| j d � | j � } t | d d g � t � } t j | | d d
g | �| j d � | j � } t | d d g � d S)
Nr+ rV y �?r4 z %+.3er s8 ( +3.142e+00+ +2.718e+00j) ( +3.142e+00+ +2.718e+00j)
z %+.3es1 +3.142e+00 +2.718e+00 +3.142e+00 +2.718e+00
z (%.3e%+.3ej)s. (3.142e+00+2.718e+00j) (3.142e+00+2.718e+00j)
z %+.3e %+.3e)
rC �zeros�
complex128�pir� r r� r? r� r )r �ncols�nrowsrS �re�imr� r& r r r! �test_complex_arrays� s<
z TestSaveTxt.test_complex_arraysc C sd Gd d � d t � } | � } t j d d g � } t j | | � t j | � } t | | � d S) Nc @ s e Z d Z d d � Z d S)z4TestSaveTxt.test_custom_writer.<locals>.CustomWriterc S s | j | j d � � d S)Ns
)�extend�split)r �textr r r! r" � s z:TestSaveTxt.test_custom_writer.<locals>.CustomWriter.writeN)r' r( r) r" r r r r! � CustomWriter� s r� rP r+ r- rU )rP r+ )r- rU )�listrC rW r� r� r )r r� �wrS r� r r r! �test_custom_writer� s z TestSaveTxt.test_custom_writerN) r' r( r) r\ r_ rf r� r� r� r� r� r� r r r r! r� 3 s " &r� c @ sl 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 d1 d2 � Z d3 d4 � Z d5 d6 � Z d7 d8 � Z d9 d: � Z d; S)<� TestLoadTxtc C s� t � } | j d � | j d � t j | d d t j f d t j f g �} t j d d g d d d g �} t | | � t � } | j d � | j d � d d d d i } t j d d g d | �} t j | d | �} t | | � d S)!Nz1 2
3 4r rV rc re rP r+ r- rU rd zM 64.0 75.0
F 25.0 60.0�names�gender�age�weight�formats�S1�f4�M� P@� �R@�F� 9@� N@)rP r+ )r- rU )rc rd )re rd )r� r� r� )r� rd r� )r� r� r� )r� r� r� )r r" r? rC r� �int32rW r )r r� rc rS �dZ mydescriptorr� re r r r! rf � s
-!
zTestLoadTxt.test_recordc C s� t � } | j d � | j d � t j | d t j �} t j d d g d d g g t � } t | | � | j d � t j | d t �} t j d d g d d g g t � } t | | � d S)Nz1 2
3 4r rV rP r+ r- rU ) r r" r? rC r� rY rW r rX )r r� rc rS r r r! r\ � s
$
$zTestLoadTxt.test_arrayc C s� t � } | j d � | j d � t j | d t �} t j d d d d g t � } t | | � t � } | j d � | j d � t j | d t d d
�} t j d d d d g t � } t | | � d S) Nz1
2
3
4
r rV rP r+ r- rU z1,2,3,4
r� r� )r r" r? rC r� rY rW r )r r� rc rS r r r! r_ � s
zTestLoadTxt.test_1Dc
C s� t � } | j d � | j d � t j | d t d d d d d d � i �} t j d
d d d d
g t � } t | | � d S)Nz 1,2,3,,5
r rV r� r� �
convertersr- c S s t | p d � S)Ni� i���)rY )r r r r! �<lambda> s z*TestLoadTxt.test_missing.<locals>.<lambda>rP r+ i� � i���)r r" r? rC r� rY rW r )r r� rc rS r r r! � test_missing s
!zTestLoadTxt.test_missingc C s� t � } | j d � | j d � t j | d t d d d d d d � i d
d �} t j d d g d d g g t � } t | | � d S)Nz1,2,3,,5
6,7,8,9,10
r rV r� r� r� r- c S s t | p d � S)Ni� i���)rY )r r r r! r� s z:TestLoadTxt.test_converters_with_usecols.<locals>.<lambda>�usecolsrP r+ i� � � )rP r- i���)r r" r? rC r� rY rW r )r r� rc rS r r r! � test_converters_with_usecols s
$z(TestLoadTxt.test_converters_with_usecolsc C ss t � } | j d � | j d � t j | d t d d d d �} t j d d d
d g t � } t | | � d S) Nz# comment
1,2,3,5
r rV r� r� r� �#rP r+ r- r� )r r" r? rC r� rY rW r )r r� rc rS r r r! �test_comments_unicode s
z!TestLoadTxt.test_comments_unicodec C ss t � } | j d � | j d � t j | d t d d d d �} t j d d d
d g t � } t | | � d S) Nz# comment
1,2,3,5
r rV r� r� r� � #rP r+ r- r� )r r" r? rC r� rY rW r )r r� rc rS r r r! �test_comments_byte s
z TestLoadTxt.test_comments_bytec
C s� t � } | j d � | j d � t j | d t d d d d d d g �} t j d
d d g d
d d g g t � } t | | � d S)Nz,# comment
1,2,3
@ comment2
4,5,6 // comment3r rV r� r� r� r� �@z//rP r+ r- rU r� � )r r" r? rC r� rY rW r )r r� rc rS r r r! �test_comments_multiple% s
*z"TestLoadTxt.test_comments_multiplec
C s� t � } | j d � | j d � t j | d t d d d d �} t j d d d
d g t � } t | | � t � } | j d � | j d � t t t j | d t d d d d �d S)
Nz/* comment
1,2,3,5
r rV r� r� r� z/*rP r+ r- r� z*/ comment
1,2,3,5
)
r r" r? rC r� rY rW r r r� )r r� rc rS r r r! �test_comments_multi_chars. s
z%TestLoadTxt.test_comments_multi_charsc C s� t � } | j d � | j d � t j | d t d d d d �} t j d d d d
g t � } t | | � t � } | j d � | j d � t j | d t d d d d �} t j d d d d
g t � } t | | � d S) Nzcomment
1,2,3,5
r rV r� r� �skiprowsrP r+ r- r� z# comment
1,2,3,5
)r r" r? rC r� rY rW r )r r� rc rS r r r! �
test_skiprows> s
zTestLoadTxt.test_skiprowsc
C s� t j d d g d d g g t � } t � } t j | | � | j d � t j | d t d d �} t | | d d � d f � t j d d d g d d d g g t � } t � } t j | | � | j d � t j | d t d d �} t | | d d � d d � f � | j d � t j | d t d t j d d g � �} t | | d d � d d � f � x� t t j t j
t j t j t j
t j t j t j g D]W } | d � } | j d � t j | d t d | �} t | | d d � d f � q�WGd d
� d
t � } | � } | j d � t j | d t d | �} t | | d d � d f � | j d � t j | d t d | f �} t | | d d � d f � d } t | � } d d
g } d d g }
t j | d d d t t | |
� � �} t | d d d g � t | d
d d g � | j d � d } t t d t | � t j | d | �t t d t | � t j | d d | d g �d S)NrP r+ r- rU r rV r� r� c @ s e Z d Z d d � Z d S)z*TestLoadTxt.test_usecols.<locals>.CrazyIntc S s d S)NrP r )r r r r! � __index__n s z4TestLoadTxt.test_usecols.<locals>.CrazyInt.__index__N)r' r( r) r� r r r r! �CrazyIntm s r� z<JOE 70.1 25.3
BOB 60.5 27.9
�stid�temp�S4�f8s JOEs BOBg�����L9@gfffff�;@g �?z^usecols must be.*%s)rP )rP r+ )r r+ )rC rW rX r r� r? r� r rY �int8�int16r� �int64r� �uint16�uint32�uint64r] r r� r� r r � TypeError�type)
r rS r� rc �int_typeZto_readr� Z crazy_intr� r� �dtypesrE Z bogus_idxr r r! � test_usecolsO s^ $
*
#
*#
!
*
zTestLoadTxt.test_usecolsc C s� t � } | j d � | j d � t j d t f d d t f d t f g f g � } t j | d | d d �} t j d d g | � } t | | � d S)Nz1,2,3.0
4,5,6.0
r rc re r� r rV r� r� rP r+ � @rU r� � @�r+ r )rP r
�r� r )rU r )
r r" r? rC rV rY rX r� rW r )r r� rq rc rS r r r! �test_fancy_dtype� s
3z TestLoadTxt.test_fancy_dtypec C s� t d � } t j d d t f d t f d t d f g � } t j | d | �} t j d
d d d
d d g d d d g g f g d | �} t | | � d S)Nzaaaa 1.0 8.0 1 2 3 4 5 6r= r� rc re �blockr+ r- rV �aaaag �?g @rP rU r� r� )r= r� )r+ r- )r rC rV rX rY r� rW r )r r� rq rc rS r r r! �test_shaped_dtype� s 3 z TestLoadTxt.test_shaped_dtypec C s� t d � } t j d d t f d t f d t d f g � } t j | d | �} t j d
d d d
d d g d d d g g d d d g d d d g g g f g d | �} t | | � d S)Nz*aaaa 1.0 8.0 1 2 3 4 5 6 7 8 9 10 11 12r= r� rc re r r+ r- rV r g �?g @rP rU r� r� r� � r� �
� � )r= r� )r+ r+ r- )r rC rV rX rY r� rW r )r r� rq rc rS r r r! �test_3d_shaped_dtype� s B z TestLoadTxt.test_3d_shaped_dtypec C sg d d g } xT t t j f D]C } t j d d g d | �} t j | d | �} t | | � q Wd S)NZstr1Zstr2rV )�strrC �bytes_rW r� r )r r� rq rS rc r r r! �test_str_dtype� s
zTestLoadTxt.test_str_dtypec C s� t � �} } | j d d � t � } t j | � } t | j d � t j | d t j �} t | j d � t | j t j k � Wd QRXd S)N�messagezloadtxt: Empty input file:r rV )r )r )
r r� r rC r� r rR r r rV )r r� r� rc r r r! �test_empty_file� s zTestLoadTxt.test_empty_filec C s� t � } | j d d g � | j d � t j | d d d d d d � i �} t | d d
g � | j d � t j | d d d d d d � i �} t | d d
g � d S)Nz1 21
z3 42
r r� rP r� c S s
t | d � S)N� )rY )r r r r! r� � s z3TestLoadTxt.test_unused_converter.<locals>.<lambda>� �* c S s
t | d � S)Nr )rY )r r r r! r� � s �! �B )rP )rP )r r% r? rC r� r )r r� r� r r r! �test_unused_converter� s
z!TestLoadTxt.test_unused_converterc C s� d } d t f d t j f g } d d � } d | i } t j t | � d d d | d
| �} t j d t d d d � f d t d
d d � f g d | �} t | | � d S)Nz0 1; 2001-01-01
2; 2002-01-31 �idx�codec S s t | j � d � S)Nz%Y-%m-%d)r2 �strip)r r r r! r� � s z4TestLoadTxt.test_dtype_with_object.<locals>.<lambda>rP r� �;rV r� i� r+ i� r )rY rC r] r� r rW r r )r r� �ndtype�funcr� �test�controlr r r! �test_dtype_with_object� s 0 z"TestLoadTxt.test_dtype_with_objectc C sV d } t � } | j d | � | j d � t j | d t j �} t | | � d S)N� +&�| � -(�| z%s %sr rV )r, r- )r r" r? rC r� r r )r �tgtr� �resr r r! �test_uint64_type� s
z TestLoadTxt.test_uint64_typec C sV d } t � } | j d | � | j d � t j | d t j �} t | | � d S)N� ���� z%s %sr rV ������� )r2 r1 )r r" r? rC r� r r )r r. r� r/ r r r! �test_int64_type� s
zTestLoadTxt.test_int64_typec C s� t j d d d � j t j � } t j | | f � j t j � } d j t t j | � � } t � } | j
| � xS t j t j g D]? } | j d � t j | d | �} t
| | d d | �q� Wd S) Nr r� r� r rV �err_msgz%si����)rC �logspace�astype�float32�hstackrX r� �map�hexr r" r? r� r )r r. �inpr� rq r/ r r r! �test_from_float_hex� s !"
z TestLoadTxt.test_from_float_hexc C sn t d d � t d d � f } t � } | j d | � | j d � t j | d t j �} t | | � d S)NrP z%s %sr rV ���)r� r r" r? rC r� r )r r. r� r/ r r r! �test_from_complex� s
z TestLoadTxt.test_from_complexc C sk t � �< } t | d � � } | j d � Wd QRXt j | � } Wd QRXt | d d g d d g g � d S)Nr� z
1 21
3 42
rP r r- r )r r� r" rC r� r )r r= r� r� r r r! �test_universal_newline s
z"TestLoadTxt.test_universal_newlinec C s} t � } | j d � | j d � d d d d i } t j | d | d
d �} t j d d d g � } t | d | � d S)Nz 1 2 3 start
4 5 6
7 8 9.5 r r� rc re �z�commentr� �<i4�<f4�|S8rV r� � s start s � )rc re r@ rA )rB rB rC rD )r r" r? rC r� rW r )r r� rq rc rS r r r! �test_empty_field_after_tab
s
z&TestLoadTxt.test_empty_field_after_tabc C s� t d � } d d d d i } t j | d
| d d �\ } } } t | j j d k � t | j j d k � t | j j d k � t | t j d
d g � � t | t j d d g � � t | t j d d g � � d S)NzM 21 72
F 35 58r� rS r� r� r� �|S1�<i4�<f4rV �unpackT� M� Fr �# g R@g M@)rS r� r� )rH rI rJ )r rC r� r rV r r rW )r �txtrq rS r� r� r r r! �test_structure_unpack s $ z!TestLoadTxt.test_structure_unpackc C s� t � } | j d � | j d � t t t j | d d �| j d � t t t j | d d �| j d � t j | d t d d d d �} t j d d
d g d d d
g g � } t | | � t � } | j d � | j d � t j | d t d d d d
�} t
| j d k � | j d � t j | d t d d d d �} t
| j d k � | j d � t j | d t d d d d �} t
| j d k � t � } | j d � | j d � t j | d t d d d d
�} t
| j d k � | j d � t j | d t d d d d �} t
| j d k � | j d � t j | d t d d d d �} t
| j d k � t � �d } | j
d d � t � } t
t j | d d
�j d k � t
t j | d d �j d k � Wd QRXd S)Nz 1,2,3
4,5,6r �ndminr- g �?rV r� r� rP r+ rU r� r� z0,1,2z0
1
2r zloadtxt: Empty input file:)rP r- )r- )r- )r- rP )r- )r- )r rP )r )r r" r? r r� rC r� rY rW r r rR r r� )r r� rc rS r� r� r� r� r r r! �test_ndmin_keyword sJ
!'
!
!
!
!
!
! "z TestLoadTxt.test_ndmin_keywordc C s8 d d � } t j | � � } t | t j d � � d S)Nc s s$ x t d � D] } d | Vq
Wd S)Nr z%d)rk )rh r r r! �countK s z0TestLoadTxt.test_generator_source.<locals>.countr )rC r� r �arange)r rS r/ r r r! �test_generator_sourceJ s z!TestLoadTxt.test_generator_sourcec C s= t � } | j d � | j d � t t d t j | � d S)Nz1 2 3
4 5 6
2 3r �3)r r" r? r r� rC r� )r r� r r r! �
test_bad_lineR s
zTestLoadTxt.test_bad_linec C sj t � } | j d � | j d � t j d t f d d t f g � } t j | d d d | d
d �d S) Nz100,foo,200
300,None,400r rc rS �S10re r� r� rV r� )rS rX )r r" r? rC rV rY r� )r r� rq r r r! �test_none_as_stringZ s
$z TestLoadTxt.test_none_as_stringN) r' r( r) rf r\ r_ r� r� r� r� r� r� r� r
r r r r r r" r+ r0 r3 r< r> r? rG rP rR rU rW rY r r r r! r� � s:
E
+ r� c @ s4 e Z d Z d d � Z d d � Z d d � Z d S)�
Testfromregexc C s| t � } | j d � | j d � d t j f d g } t j | d | � } t j d d d g d
| �} t | | � d S)Nz 1.312 foo
1.534 bar
4.444 quxr �num�val�S3z([0-9.]+)\s+(...)�ˡE���?�foo�%��C��?�bar�-����@�quxrV )r\ r] )r^ r_ )r` ra )rb rc )r r" r? rC �float64� fromregexrW r )r r� rq rc rS r r r! rf e s
zTestfromregex.test_recordc C s| t � } | j d � | j d � d t j f d g } t j | d | � } t j d d d g d
| �} t | | � d S)Nz1312 foo
1534 bar
4444 quxr r[ r\ r] z
(\d+)\s+(...)� r_ � ra �\ rc rV )r\ r] )rf r_ )rg ra )rh rc )r r" r? rC r� re rW r )r r� rq rc rS r r r! �
test_record_2p s
zTestfromregex.test_record_2c C sy t � } | j d � | j d � d t j f g } t j | d | � } t j d d
d g d | �} t | | � d S) Nz1312 foo
1534 bar
4444 quxr r[ z (\d+)\s+...� � �\ rV )rj )rk )rl )r r" r? rC rd re rW r )r r� rq rc rS r r r! �
test_record_3{ s
zTestfromregex.test_record_3N)r' r( r) rf ri rm r r r r! rZ c s rZ 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 d1 d2 � Z d3 d4 � Z d5 d6 � Z d7 d8 � Z d9 d: � Z d; d< � Z d= d> � Z! d? d@ � Z" dA dB � Z# dC dD � Z$ dE dF � Z% dG dH � Z& dI dJ � Z' dK dL � Z( dM dN � Z) dO dP � Z* dQ dR � Z+ dS dT � Z, dU dV � Z- dW dX � Z. dY dZ � Z/ d[ d\ � Z0 d] d^ � Z1 d_ d` � Z2 da db � Z3 dc dd � Z4 de df � Z5 dg dh � Z6 di dj � Z7 dk dl � Z8 dm dn � Z9 do dp � Z: dq dr � Z; ds dt � Z< du dv � Z= dw dx � Z> dy dz � Z? d{ d| � Z@ d} d~ � ZA d d� � ZB d� d� � ZC d� d� � ZD d� S)�� TestFromTxtc C s� t d � } t j | d d t j f d t j f g �} t j d d g d d d g �} t | | � t d
� } d d d d i } t j d d g d | �} t j | d | �} t | | � d S) Nz1 2
3 4rV rc re rP r+ r- rU rd zM 64.0 75.0
F 25.0 60.0r� r� r� r� r� r� r� r� � P@� �R@r� � 9@� N@)rP r+ )r- rU )rc rd )re rd )r� r� r� )r� rd r� )r� ro rp )r� rq rr )r rC � ndfromtxtr� rW r )r r� r) r* �
descriptorr r r! rf � s -!
zTestFromTxt.test_recordc C s� t d � } t j d d g d d g g d t �} t j | d t �} t | | � | j d � t j d d g d d g g d t �} t j | d t �} t | | � d S)Nz1 2
3 4rP r+ r- rU rV r ) r rC rW rY rs r r? rX r� )r r� r* r) r r r! r\ � s '
'zTestFromTxt.test_arrayc C s� t j d d d d g t � } t d � } t j | d t �} t | | � t d � } t j | d t d d �} t | | � d S)
NrP r+ r- rU z1
2
3
4
rV z1,2,3,4
r� r� )rC rW rY r rs r )r r* r� r) r r r! r_ � s
zTestFromTxt.test_1Dc C s� t j d d d d g t � } t d � } t j | d t d d d d
�} t | | � t d � } t j | d t d d d d
�} t | | � d S) NrP r+ r- r� z# comment
1,2,3,5
rV r� r� r� r� z1,2,3,5# comment
)rC rW rY r rs r )r r* r� r) r r r! �
test_comments� s !
!zTestFromTxt.test_commentsc C s� t j d d d d g t � } t d t d d � } t d � } t j | d d | �} t | | � t d
� } t j | d d | �} t | | � d S) NrP r+ r- r� rV r� r� zcomment
1,2,3,5
� skip_headerz# comment
1,2,3,5
r� )rC rW rY �dictr rs r r� )r r* rM r� r) r r r! r� � s
zTestFromTxt.test_skiprowsc C s� d d � t d d � D� } | j d � | j d d � t d � D� � d | d <t d d
d d d
d d d � } t j t d j | � � | � } t j d d � t d � D� d d d � d D� �} t | | � d S)Nc S s g | ] } d | � q S)z# %ir )r# rh r r r! r$ � s z0TestFromTxt.test_skip_footer.<locals>.<listcomp>rP r� zA, B, Cc S s# g | ] } d | | | f � q S)z
%i,%3.1f,%03sr )r# rh r r r! r$ � s �3 z99,99r� r� r� Trv r� � skip_footerr r� c S s+ g | ]! } d | d | d | f � q S)z%fr )r# rh r r r! r$ � s �) rV c S s g | ] } | t f � q Sr )rX )r# �_r r r! r$ � s �ABCr= )
rk r� r� rw rC �
genfromtxtr r� rW r )r r� rM r) �ctrlr r r! �test_skip_footer� s
!! z TestFromTxt.test_skip_footerc C s� t � ��} | j t � d } t t t j t | � d d �t j t | � d d d d �} t | t j d d g d d g d d g d d g g � � t j t | � d d
�} t | t j d d g d d g d d g d d g g � � d } t j t | � d d d d �} t | t j d d g d d g d d g d d g g � � t j t | � d d
d d �} t | t j d d g d d g d d g g � � Wd QRXd S)
Nz 1 1
2 2
3 3
4 4
5
6
7
ry rP �
invalid_raiseFg �?g @g @g @r- z 1 1
2
3 3
4 4
5
6 6
7 7
g @)
r r� r r r� rC r} r r rW )r r� ZbasestrrS r r r! � test_skip_footer_with_invalid� s"
:::z)TestFromTxt.test_skip_footer_with_invalidc C s� t d � } t j | d d d d �} d t j d d g � d t j d d
g � d t j d d
g � i } t | d | d � t | d | d � t | d | d � d S)Nz)gender age weight
M 64.0 75.0
F 25.0 60.0rV r� Tr� rL rM r� g P@g 9@r� g �R@g N@)r rC rs rW r )r r� r) r* r r r! � test_header� s zTestFromTxt.test_headerc C s� t d � } t j | d d �} t j d d g � t j d d g � t j d d g � t j d d g � t j d
d g � g } t | j j d d d d d g � x/ t | � D]! \ } } t | d | | � q� Wd S)Nz*A 64 75.0 3+4j True
BCD 25 60.0 5+6j FalserV � As BCD�@ � g �R@g N@r- y @r� y @TF�f0�f1�f2�f3r� zf%iy @ @y @ @)r rC rs rW r rV r� r| )r r� r) r* rh r~ r r r! �test_auto_dtype� s "zTestFromTxt.test_auto_dtypec C s_ t d � } t j | d d �} t j d d d d g d d d d
g g � } t | | � d S) Nz1 2 3 4
5 6 7 8
rV rP r+ r- rU r� r� r� r )r rC rs rW r )r r� r) r* r r r! �test_auto_dtype_uniform s -z#TestFromTxt.test_auto_dtype_uniformc C s� t d � } t j d t f d d t f d t f g f g � } t j | d | d d �} t j d d g d | �} t | | � d S)Nz1,2,3.0
4,5,6.0
rc re r� r rV r� r� rP r+ � @rU r� � @�r+ r� )rP r� �r� r� )rU r� )r rC rV rY rX rs rW r )r r� �
fancydtyper) r* r r r! r
s
3z TestFromTxt.test_fancy_dtypec C su d d d d i } t d � } d } t j | d
| d | �} | | d <t j d d g d
| �} t | | � d S)Nr� �grS r� r� r� rd r� s M 64.0 75.0
F 25.0 60.0r� r� r� rV r� � P@� �R@r� � 9@� N@)r� rS r� )r� rd r� )r� r� r� )r� r� r� )r� r� r� )r rC rs rW r )r rt r� r� r) r* r r r! �test_names_overwrite s
z TestFromTxt.test_names_overwritec C s� t d � } t j | d d d d �} t j d d d g d d d t f d t f g �} t | | � t d � } t j | d d d d �} t | | � d S)NzN
#gender age weight
M 21 72.100000
F 35 58.330000
M 33 21.99
r� TrV r� r �fffffR@r� rN �
ףp=*M@r �=
ףp�5@r� �|S1r� r� sO
# gender age weight
M 21 72.100000
F 35 58.330000
M 33 21.99
)r� r r� )r� rN r� )r� r r� )r� r� )r rC r} rW rY rX r )r r� r) r~ r r r! �test_commented_header! s
z!TestFromTxt.test_commented_headerc C sh t d � } t j | d d
d d d d �} t j d d d d t f d t f g �} t | | � d S)NzA B C D
aaaa 121 45 9.1r� �A�C�Dr� TrV r �- �333333"@�|S4)r� r� r� )r r� r� )r� r� )r rC rs rW rY rX r )r r� r) r* r r r! �test_autonames_and_usecols8 s z&TestFromTxt.test_autonames_and_usecolsc C st t d � } t j | d t d d d d d d � i d d �} t j d d g d
d g g t � } t | | � d S)Nz1,2,3,,5
6,7,8,9,10
rV r� r� r� r- c S s t | p d � S)Ni� i���)rY )r r r r! r� E s z:TestFromTxt.test_converters_with_usecols.<locals>.<lambda>r� rP r+ i� r� r� )rP r- i���)r rC rs rY rW r )r r� r) r* r r r! r� A s $z(TestFromTxt.test_converters_with_usecolsc C sz t d � } t j | d d d d d d d d d
d � i �} t j d d d d t f d t f g �} t | | � d S)NzA B C D
aaaa 121 45 9.1r� r� r� r� r� TrV r� c S s d t | � S)Nr+ )rY )r r r r! r� N s zDTestFromTxt.test_converters_with_usecols_and_names.<locals>.<lambda>r �Z �333333"@�|S4)r� r� r� )r r� r� )r� r� )r rC rs rW rY rX r )r r� r) r* r r r! �&test_converters_with_usecols_and_namesJ s z2TestFromTxt.test_converters_with_usecols_and_namesc
C s� d d d � i } t d � } t j | d d d d d d d g d
| �} t j t d d d
� d f d d t j f d t f g �} t | | � d S)N�datec S s
t | d � S)Nz%Y-%m-%d %H:%M:%SZ)r2 )r r r r! r� V s z9TestFromTxt.test_converters_cornercases.<locals>.<lambda>z 2009-02-03 12:00:00Z, 72214.0r� r� rV r� r� r� i� r+ r- g `��@)r rC rs rW r �object_rX r )r � converterr� r) r* r r r! �test_converters_cornercasesS s z'TestFromTxt.test_converters_cornercasesc
C s� d d d � i } t d � } t j | d d d d d d d g d
| �} t j t d d d
� d f d d d t f g �} t | | � d S)Nr� c S s t j t | d � � S)Nz%Y-%m-%d %H:%M:%SZ)rC �
datetime64r2 )r r r r! r� a s z:TestFromTxt.test_converters_cornercases2.<locals>.<lambda>z 2009-02-03 12:00:00Z, 72214.0r� r� rV r� r� r� i� r+ r- g `��@�datetime64[us])r� r� )r rC rs rW r rX r )r r� r� r) r* r r r! � test_converters_cornercases2^ s z(TestFromTxt.test_converters_cornercases2c C s� t d � } t j | d d
d d d d � i �} t | d d g � | j d � t j | d d d d d
d � i �} t | d d g � d S)Nz 1 21
3 42
r� rP r� r c S s
t | d � S)Nr )rY )r r r r! r� m s z3TestFromTxt.test_unused_converter.<locals>.<lambda>r r c S s
t | d � S)Nr )rY )r r r r! r� r s r r! )rP )rP )r rC rs r r? )r r� r) r r r! r" i s
z!TestFromTxt.test_unused_converterc C se d d � } d d � } t d � } t d d | d | i d d d
d � } t t t j | | � d S) Nc S sJ t d | j � k r% | j � d pF d | j � k rC | j � pF d � S)N� rrP g r= )rX �lowerr� r% )rc r r r! r� v s %z4TestFromTxt.test_invalid_converter.<locals>.<lambda>c S sJ t d | j � k r% | j � d pF d | j � k rC | j � pF d � S)N� %r g )rX r� r� r% )rc r r r! r� x s %zgD01N01,10/1/2003 ,1 %,R 75,400,600
L24U05,12/5/2003, 2 %,1,300, 150.5
D02N03,10/10/2004,R 1,,7,145.55r� r+ r- r� r� rV )r rw r r rC r} )r Z
strip_randZ strip_perr rM r r r! �test_invalid_converteru s z"TestFromTxt.test_invalid_converterc C sk t d � } d d � } t j | d d d d | i �} t j d d g d
d g g � } t | | � d S) Nz q1,2
q3,4c S s t | d d � � S)NrP )rX )r r r r! r� � s z;TestFromTxt.test_tricky_converter_bug1666.<locals>.<lambda>r� r� r� r g �?g @g @g @)r rC r} rW r )r r Zcnvr) r* r r r! � test_tricky_converter_bug1666� s
!!z)TestFromTxt.test_tricky_converter_bug1666c C s� d } t j t | � d d d t d d t i �} t j d g d d d t f d
t f g �} t | | � t j t | � d d d t d d t i �} t j d d d g � } t | | � d S)Nz 2009; 23; 46r� r& rV r� r �2009� 7@�. r� �|S4r� r� g d�@)r� r� r� )r� r� )rC rs r rX r rW r )r �dstrr) r* r r r! �test_dtype_with_converters� s
z&TestFromTxt.test_dtype_with_convertersc sn d } d d d d d d d d i � d d d d g } d t d t d t d � f d d � i } t j t | � d | d d d d d | �} t j j d d d d g d d d d g d d d! d g g d | �} t | | � d" d# d$ g } t j t | � d | d d d d% d d d | �} t j j d d d g d d d g d d d g g d | �} t | | � d S)&Nz!1,5,-1,1:1
2,8,-1,1:n
3,3,-2,m:n
z1:1r z1:nrP zm:1r+ zm:nr- �e1rd �e2�e3�i2r~ �i1c s � | j � S)N)r3 )ra )�dmapr r! r� � s zDTestFromTxt.test_dtype_with_converters_and_usecols.<locals>.<lambda>rV r� r� r� r� r� r r� )r� rd )r� rd )r� r� )r~ r� r= r= ���)r� rd )r� rd )r~ r� )r rP r- )rY rC �
recfromcsvr �recrW r )r r� �dtyp�convr) r* r )r� r! �&test_dtype_with_converters_and_usecols� s * E
<z2TestFromTxt.test_dtype_with_converters_and_usecolsc C s& d } d t f d t j f g } d d � } d | i } t j t | � d d d | d
| �} t j d t d d d � f d t d
d d � f g d | �} t | | � d d t f d t j f g f g } y+ t j t | � d d d | d
| �} Wn t k
rYn Xd } t | � � d S)Nz0 1; 2001-01-01
2; 2002-01-31 r# r$ c S s t | j � d � S)Nz%Y-%m-%d)r2 r% )r r r r! r� � s z4TestFromTxt.test_dtype_with_object.<locals>.<lambda>rP r� r& rV r� i� r+ i� r Znestz3Nested dtype involving objects should be supported.)
rY rC r] r} r rW r r �NotImplementedErrorr� )r r� r' r( r� r) r* �errmsgr r r! r+ � s$ 0
$
z"TestFromTxt.test_dtype_with_objectc
C st t d � } t j | d d d d d t d d d d t i �} t j d g d d d t f g �} t | | � d S)Nz skip,skip,2001-01-01,1.0,skipr� r� r� rV r� r+ r- r� �
2001-01-01� �?r �|S10)r+ r- )r� r� )r r� )r rC r} rX r rW r )r r� r) r* r r r! �'test_userconverters_with_explicit_dtype� s z3TestFromTxt.test_userconverters_with_explicit_dtypec C s_ t d � } t j | � } t j d d d d d g d d d d
d g g � } t | | � d S) Nz 1 2 3 4 5
6 7 8 9 10g �?g @g @g @g @g @g @g @g "@g $@)r rC rs rW r )r r� r) r* r r r! �test_spacedelimiter� s
z TestFromTxt.test_spacedelimiterc C se d } t j t | � d d �} t j d d d g d d d g d d
d g g � } t | | � d S) Nz 1 2 3
4 5 67
890123 4r� r- rP r+ rU r� �C iz �{ )rC r} r rW r )r r� r) r* r r r! �test_integer_delimiter� s 3z"TestFromTxt.test_integer_delimiterc
C sk t d � } t j | d t d d d d d d � i �} t j d d
d d
d g t � } t | | � d S)Nz 1,2,3,,5
rV r� r� r� r- c S s t | p d � S)Ni� i���)rY )r r r r! r� � s z*TestFromTxt.test_missing.<locals>.<lambda>rP r+ i� r� i���)r rC rs rY rW r )r r� r) r* r r r! r� � s
!zTestFromTxt.test_missingc C s� d } t j t | � d d d d �} t j d t j d t j f d t j d f g � } t j d d
d g d
t �} t | j | � t | j | � d S)Nz1 2 3
2
1 3r� rE �usemaskTrP r+ r- r rV )rP r+ r- )r r r )rP r rP )r rP r ) rC r} r rW �nan�boolr r� �mask)r rO r) Zctrl_dZctrl_mr r r! �test_missing_with_tabs� s 3 z"TestFromTxt.test_missing_with_tabsc C sn t j d d g d d g g t � } t � } t j | | � | j d � t j | d t d d �} t | | d d � d f � t j d d d g d d d g g t � } t � } t j | | � | j d � t j | d t d d
�} t | | d d � d d � f � | j d � t j | d t d t j d d g � �} t | | d d � d d � f � d S) NrP r+ r- rU r rV r� r� )rP )rP r+ )rC rW rX r r� r? rs r )r r* r� r) r r r! r
� s $
*
#
*zTestFromTxt.test_usecolsc C s` d } t j t | � d d d d �} t j d d g d
d d � d
D� �} t | | � d S)Nz 1 2 3
4 5 6r� za, b, cr� za, crP r- rU r� rV c S s g | ] } | t f � q Sr )rX )r# r{ r r r! r$ s z3TestFromTxt.test_usecols_as_css.<locals>.<listcomp>�ac)rP r- )rU r� )rC r} r rW r )r r� r) r~ r r r! �test_usecols_as_css� s
(z TestFromTxt.test_usecols_as_cssc C s� t d � } d d g } d d g } t j | d d d t t | | � � �} t | d d
d g � t | d d d
g � d S)NzJOE 70.1 25.3
BOB 60.5 27.9r� r� r� r� r� r r+ rV s JOEs BOBg�����L9@gfffff�;@)r r+ )r rC rs r� r� r )r r� r� r r) r r r! �"test_usecols_with_structured_dtype s $z.TestFromTxt.test_usecols_with_structured_dtypec C s; t j t d � d d �} t | t j d d g � � d S)Ns 1 2 3
4 5 6r� r g �?g @)rC r} r r rW )r r) r r r! �test_usecols_with_integer
s z%TestFromTxt.test_usecols_with_integerc C s� t j d
d g d d t f d t f g �} d } t d d
� } t j t | � d d | �} t | | � t j t | � d d | �} t | | � d S)NrP r- rU r� rV rS r� z 1 2 3
4 5 6r� za, b, cr� r )rP r- )rU r� r= )r r= )rS r� )rC rW rX rw r} r r )r r~ r� rM r) r r r! � test_usecols_with_named_columns s -
z+TestFromTxt.test_usecols_with_named_columnsc C sU t � �E } | j d d � t � } t j | � } t | t j g � � Wd QRXd S)Nr z genfromtxt: Empty input file:)r r� r rC r} r rW )r r� r� r) r r r! r s
zTestFromTxt.test_empty_filec C s� t d � } t j d t f d d t f d t f g f g � } t j | d | d d �} t j d d g d | �} t | | � d S)Nz1,2,3.0
4,5,6.0
rc re r� r rV r� r� rP r+ � @rU r� � @�r+ r� )rP r� �r� r� )rU r� ) r rC rV rY rX � mafromtxt�marW r )r r� r� r) r* r r r! �test_fancy_dtype_alt% s
3z TestFromTxt.test_fancy_dtype_altc C s� t d � } t j d d t f d t f d t d f g � } t j | d | �} t j d
d d d
d d g d d d g g f g d | �} t | | � d S)Nzaaaa 1.0 8.0 1 2 3 4 5 6r= r� rc re r r+ r- rV r g �?g @rP rU r� r� )r= r� )r+ r- )r rC rV rX rY rs rW r )r r� rq rc rS r r r! r - s 3 z TestFromTxt.test_shaped_dtypec C s t d � } t d d d d d d � } t j | d d | �} t j d d g d d d g d d t j f d t j f g �} t | | � t | j | j � | j d � t j | | � } t j d d g d d d g d d t j
f d t j
f g �} t | | � t | j | j � d S)Nz
A,B
0,1
2,N/Ar� r� �missing_valueszN/Ar� TrV r rP r+ r� Fr� �B)r rP r= )r+ r= )FF)FT)r rP r= )r+ r= )FF)FT) r rw rC r� r� rW rY r r� r? rX )r r� rM r) r* r r r! �test_withmissing6 s !
!
z TestFromTxt.test_withmissingc C s� d } t d d d d d d � } d t f d t f d t f g } t j t | � d
d | �} t j d d d d! g d d" d# d$ d% g d | �} t | | � | | d <t j t | � d
d d& d d' d d( i | �} t j d) d+ d. d0 g d d1 d2 d3 d4 g d | �} t | | � t j t | � d
d d5 d d6 d d7 i | �} t j d8 d: d= d? g d d@ dA dB dC g d | �} t | | � d S)DNz4A, B, C
0, 0., 0j
1, N/A, 1j
-9, 2.2, N/A
3, -99, 3jrV r� r� r� Tr� r� r� r� zN/Ar � � rP i� � �?r� 皙����@y 8�@r- r� � @r� r+ )r r� r� ����)rP r� r� ���� � 8��)r� r� r� ���)r- r� r� )r r r )r rP r )r r rP )r r r i����i����y � 8��)r r� r� ����)rP r� r� ���� � 8��)r� r� r� ���)r- r� r� )r r r )r rP r )rP r rP )r rP r i����i����y � 8��)r r� r� ����)rP r� r� ���� � 8��)r� r� r� ���)r- r� r� )r r r )r rP r )rP r rP )r rP r )
rw rY rX r� rC r� r r� rW r )r r� Z
basekwargs�mdtyper) r* r r r! �test_user_missing_valuesH s4 !
z$TestFromTxt.test_user_missing_valuesc C s� t j d d! g d d t f d t f g �} d } t d d
d t d d d
d d d d d d i d d d d d d d" i � } t j t | � | � } t j d# d% g d d d � d D� �} t | | � t j t | � d d' | �} t j d( d* g d d d � d D� �} t | | � d } t j t | � d d
d t d
d d d �} t j d d d d g d d d d g g � } t | | � t j t | � d d
d t d
d d d+ �} t j d d d, d g d d- d d g g � } t | | � d S).Nr r- rU i� rV rS r� zN/A, 2, 3
4, ,???r� r� r� za,b,cr� zN/Ar� r+ z???�filling_valuesc S s g | ] } | t f � q Sr )rY )r# r{ r r r! r$ r s z8TestFromTxt.test_user_filling_values.<locals>.<listcomp>�abcr� rP c S s g | ] } | t f � q Sr )rY )r# r{ r r r! r$ v s r� z1,2,*,4
5,*,7,8
�*r� r� r )r r- ����)rU r� i���)r r+ r- ����)rU r r� r= )r r= )r r- ����)rU r� r= r= r= )rC rW rY rw r} r r )r r~ r� rM r) �data2r r r! �test_user_filling_valuesg s. -
(
-
-z$TestFromTxt.test_user_filling_valuesc
C s� t d � } t j | d d d d d d d d �} t j d d g d
d d g d d t j f d t j f g �} t | | � t | j | j � d S)NzA,B
0,1.5
2,-999.00rV r� r� r� z-999.0r� Tr � �?r+ g �?r� Fr� r� )r r� � �)r+ r� )FF)FT) r rC r� r� rW rY rX r r� )r r� r) r* r r r! �test_withmissing_float� s !
z"TestFromTxt.test_withmissing_floatc C s� t d � } t j | d d d d d d �} t j d d d g d
d d g g d
d d d g d d d g g �} t | | � d S)Nz 1 2 3
4 5 6
rV r� z2,5r� TrP r+ r- rU r� r� r� r )r rC r} r� rW r )r r� r) r* r r r! � test_with_masked_column_uniform� s
Ez+TestFromTxt.test_with_masked_column_uniformc C s� t d � } t j | d d d d d d �} t j d d g d
d d g d d t f d t f d t f g �} t | | � d S)NzTrue 2 3
False 5 6
rV r� z2,5r� TrP r+ r- r r� r� r� r� r� r� )rP r+ r- )r r� r� )r rP r )r rP r )r rC r} r� rW r� rY r )r r� r) r* r r r! � test_with_masked_column_various� s $z+TestFromTxt.test_with_masked_column_variousc s d g d } x" t d � D] } d | d | <q W| j d d � t d j | � � � t d d
d d d d
� � i } i � � f d d � } t t | d | �| d } t t | � d � t | t j
d d d d � d D� �� � j d � t t
t j � d d
d d
�d S)Nz
1, 1, 1, 1, 1�2 r� z 2, 2, 2, 2 2r r z
a, b, c, d, er� r� r� rV r� Tc s t j � d d � �| d <d S)Nr� F�mtest)rC rs )�_ret)rM �mdatar r! r� � s z)TestFromTxt.test_invalid_raise.<locals>.fr r r� c S s g | ] } | t f � q Sr )rY )r# r{ r r r! r$ � s z2TestFromTxt.test_invalid_raise.<locals>.<listcomp>�abcde)rk �insertr r� rw r r r �lenrC rl r? r r� rs )r r� rh �retr� r r )rM r r! �test_invalid_raise� s
)
z TestFromTxt.test_invalid_raisec sw d g d } x" t d � D] } d | d | <q W| j d d � t d j | � � � t d d
d d d d
d d � � i } i � � f d d � } t t | d | �| d } t t | � d � t | t j
d d d d � d D� �� � j d � t j � d d � �} t t | � d � t j
d d d d � d D� �} d | d d � t d � D� <t | | � d S) Nz
1, 1, 1, 1, 1r r� z 2, 2, 2, 2 2r r z
a, b, c, d, er� r� r� rV r� Tr� Fc s t j � d d � �| d <d S)Nr� r rU r )r rU )rC rs )r )rM r r r! r� � s z6TestFromTxt.test_invalid_raise_with_usecols.<locals>.fr r r� c S s g | ] } | t f � q Sr )rY )r# r{ r r r! r$ � s z?TestFromTxt.test_invalid_raise_with_usecols.<locals>.<listcomp>Zaer� rP c S s g | ] } | t f � q Sr )rY )r# r{ r r r! r$ � s �abr+ c S s g | ] } d | � q S)r r )r# r{ r r r! r$ � s )r rP )r+ r+ )
rk r r r� rw r r r r rC rl r? rs )r r� rh r
r� r r* r )rM r r! � test_invalid_raise_with_usecols� s&
)
" z+TestFromTxt.test_invalid_raise_with_usecolsc C sv d g d } t d j | � � } d d d � i } t d d d | d
d d � d
D� � } t t t j | | � d S)Nz1, 1, 1, 1, -1.1r r� rU c S s d | S)Nz(%s)r )rc r r r! r� � s z5TestFromTxt.test_inconsistent_dtype.<locals>.<lambda>r� r� r� rV c S s g | ] } | t f � q Sr )rY )r# r{ r r r! r$ � s z7TestFromTxt.test_inconsistent_dtype.<locals>.<listcomp>r )r r� rw r r� rC r} )r r� r r� rM r r r! �test_inconsistent_dtype� s
z#TestFromTxt.test_inconsistent_dtypec C st d } t j t | � d d d d d d �} t j d d g d d
t f d t f d t f g �} t | | � d S)Nz0, 1, 2.3
4, 5, 6.7r� r� rV �
defaultfmtzf%02ir rP �ffffff@rU r� �������@�f00�f01�f02)r rP r )rU r� r )rC rs r rW rY rX r )r r� r r~ r r r! �test_default_field_format� s $z%TestFromTxt.test_default_field_formatc C sk d } t j t | � d d d t d d �} t j d d d g d
d d g g d t �} t | | � d S)
Nz0, 1, 2.3
4, 5, 6.7r� r� rV r zf%02ig g �?gffffff@g @g @g������@)rC rs r rX rW r )r r� r r~ r r r! �test_single_dtype_wo_names� s
-z&TestFromTxt.test_single_dtype_wo_namesc C sf d } t j t | � d d d t d d �} t j d d g d d
d � d D� �} t | | � d S)Nz0, 1, 2.3
4, 5, 6.7r� r� rV r� za, b, c� � �?�ffffff@� @� @�������@c S s g | ] } | t f � q Sr )rX )r# r{ r r r! r$ � s zBTestFromTxt.test_single_dtype_w_explicit_names.<locals>.<listcomp>r� )r r r )r r r )rC rs r rX rW r )r r� r r~ r r r! �"test_single_dtype_w_explicit_names� s z.TestFromTxt.test_single_dtype_w_explicit_namesc C sf d } t j t | � d d d t d d �} t j d d g d d
d � d D� �} t | | � d S)Nza, b, c
0, 1, 2.3
4, 5, 6.7r� r� rV r� T� � �?�ffffff@� @� @�������@c S s g | ] } | t f � q Sr )rX )r# r{ r r r! r$ � s zBTestFromTxt.test_single_dtype_w_implicit_names.<locals>.<listcomp>r� )r r r )r! r" r# )rC rs r rX rW r )r r� r r~ r r r! �"test_single_dtype_w_implicit_names� s z.TestFromTxt.test_single_dtype_w_implicit_namesc C s} d } t j t | � d d d t t t f d d �} t j d d g d d
t f d t f d t f g �} t | | � d S)Nz0, 1, 2.3
4, 5, 6.7r� r� rV r zf_%02ir � �?�ffffff@rU � @�������@Zf_00Zf_01Zf_02)r r% r& )rU r' r( )rC rs r rY rX rW r )r r� r r~ r r r! �test_easy_structured_dtype� s $z&TestFromTxt.test_easy_structured_dtypec C s� d } t d d d d � } t j t | � | � } t j d g d d d
t f d g �} t | | � t j t | � d
d | �} t j d g d d d
t f d g �} t | | � d S)Nz01/01/2003 , 1.3, abcder� r� rV � 01/01/2003 ��������?� abcder� �|S12r� r� �|S8� autostripT�
01/01/2003r �|S10�|S5)r* r+ r, )r� r- )r� r. )r0 r+ r )r� r1 )r� r2 )rw rC rs r rW rX r )r r� rM r r~ r r r! �test_autostrip s
zTestFromTxt.test_autostripc C sZ d } t j t | � d d d d d d �} d t f d t f d t f g } t j d d | �} t | | � t j t | � d d d d d d d
d d d �} d t f d t f d t f g } t j d d | �} t | | � t j t | � d d d d d d d d �} d t f d t f d t f g } t j d d | �} t | | � d S)NzA.A, B (B), C:C
1, 2, 3.14r� r� r� TrV �AA�B_B�CCrP r+ � ��Q� @�
replace_spacer � deletecharszA.AzB (B)zC:CzB_(B))rP r+ r7 )rP r+ r7 )rP r+ r7 )rC r} r rY rX rW r )r rO r) �
ctrl_dtyper~ r r r! �test_replace_space s$ !
!
!z TestFromTxt.test_replace_spacec C sZ d } t j t | � d d d d d t �} d t f d t f d t f g } t j d d | �} t | | � t j t | � d d d d d t d
d d d �} d t f d t f d t f g } t j d d | �} t | | � t j t | � d d d d d t d d �} d t f d t f d t f g } t j d d | �} t | | � d S)NzA.A, B (B), C:C
1, 2, 3r� r� r� TrV r4 r5 r6 rP r+ r- r8 r r9 zA.AzB (B)zC:CzB_(B))rP r+ r- )rP r+ r- )rP r+ r- )rC r} r rY rW r )r rO r) r: r~ r r r! � test_replace_space_known_dtype- s$ !
!
!z*TestFromTxt.test_replace_space_known_dtypec C s� d } t d d d d � } t j d d g d d
d � d D� �} t j t | � d d | �} t | | � t j d d g d d d � d D� �} t j t | � | � } d S)NzA,,C
0,1,2
3,4,5r� r� r� Tr rP r+ r- rU r� rV c S s g | ] } | t f � q Sr )rY )r# r{ r r r! r$ K s z5TestFromTxt.test_incomplete_names.<locals>.<listcomp>r� r� r� c S s g | ] } | t f � q Sr )rX )r# r{ r r r! r$ P s )r rP r+ )r- rU r� )r� r� r� )r rP r+ )r- rU r� )r� r� r� )rw rC rW rs r r )r r� rM r~ r) r r r! �test_incomplete_namesE s
z!TestFromTxt.test_incomplete_namesc C sw d } t j t | � d t t t f d d �} t j d
d g d d t f d t f d t f g �} t | | � d S)Nz 1 2 3
4 5 6rV r� rS rP r+ r- rU r� r� r� r� )rP r+ r- )rU r� r� )rC r} r rY rX rW r )r r� r) r~ r r r! �test_names_auto_completionS s $z&TestFromTxt.test_names_auto_completionc
C s� d } d } t j t | � d t t t f d d d d d d
�} t | j j | � t j t | � d t t t f d d d d d d
�} t | j j | � t j t | � d t d d d d d d
�} t | j j | � d S)Nz'A,B,C,D,E
0,1,2,3,4
0,1,2,3,4
0,1,2,3,4r� r� �ErV r� r� r� r r+ rU r� T)r� r� r? )r r+ rU )r� r� r? )r� r� r? )rC r} r rY r rV r� )r r� Z
ctrl_namesr) r r r! � test_names_with_usecols_bug1636\ s z+TestFromTxt.test_names_with_usecols_bug1636c C s� d } t d d d d d d � } t j d d g d d t f d t f d t f g �} t j t | � | � } t | | � t d d d d d d � } t j d d g d d t f d t f d t f g �} t j t | � | � } t | | � d S)Nz- A B C
0 1 2.3
45 67 9.r� r� rU r� TrV r rP �ffffff@r� r� � "@r� r� r� )r� r� rU )r rP rA )r� r� rB )r rP rA )r� r� rB )rw rC rW rY rX rs r r )r r� rM r~ r) r r r! �test_fixed_width_nameso s $
$z"TestFromTxt.test_fixed_width_namesc C s� d } t d d d d d d
� } t j d d d g d d d
g d d d g g d t �} t j t | � | � } t | | � d S)Ns 1, 2, 3
1, , 5
0, 6,
r� r� rV r� i� rP r+ r- r� r r� i���i���i���)rw rC rW rY rs r r )r r� rM r~ r) r r r! �test_filling_values~ s
9z TestFromTxt.test_filling_valuesc C st t j t d � d d d d d d �} t | d d � t j t d � d d d d d d �} t | d d � d S)
Nz test1,testNonetherestofthedatarV r� r� r� rP s testNonetherestofthedataz test1, testNonetherestofthedatas testNonetherestofthedata)rC r} r r )r r) r r r! �test_comments_is_none� s z!TestFromTxt.test_comments_is_nonec C s5 t d � } t d d d d d d � } t j | | � } t j d d g d d
t j f d t j f g �} | j t | t j � � t | | � t d � } t j | d d d d | �} t
j d d g d d d g d d
t j f d t j f g �} t | | � t | j | j � t | j d d
g � d S)Nz A,B
0,1
2,3r� r� r� zN/Ar� Tr rP r+ r- rV r� r� z
A,B
0,1
2,N/Ar� r� F)r rP )r+ r- )r rP r= )r+ r= )FF)FT)
r rw rC �
recfromtxtrW rY �
assertTruerA �recarrayr r� r� r� )r r� rM r) r* r r r! �test_recfromtxt� s !
!
zTestFromTxt.test_recfromtxtc C s; t d � } t d d d d d d � } t j | d d | �} t j d d g d d t j f d
t j f g �} | j t | t j � � t | | � t d � } t j | d d d d | �} t
j d d g d d d g d d t j f d
t j f g �} t | | � t | j | j � t | j d d
g � t d � } t j | d d �} t j d d g d d t j f d t j f g �} | j t | t j � � t | | � t d � } d t j f d t j
f g } t j | d d d | �} t j d d g d | �} | j t | t j � � t | | � d S) Nz A,B
0,1
2,3r� zN/Ar� T�case_sensitiverV r rP r+ r- r� r� z
A,B
0,1
2,N/Ar� r� FrS r� )r rP )r+ r- )r rP r= )r+ r= )FF)FT)r rP )r+ r- )r rP )r+ r- )r rw rC r� rW rY rG rA rH r r� r� r� rX )r r� rM r) r* rV r r r! �test_recfromcsv� s8 !
!
!
zTestFromTxt.test_recfromcsvc C s� d } t | � } t j | d d �} t j | � } t | d d g d d g d d g g � t | d d
g d d g g � t t t j t | � d d
�d } t j t | � d d �} t j d d g d d g g � } t | | � t t t j t | � d d d d �t t t j t | � d d �t � �� } | j t � t j t | � d d d d �} t j d d g d d g d d g d d g g � } t | | � t j t | � d d d d �} t j d d g d d g d d g d d g g � } t | | � Wd QRXd } t | � } t j | d d d d d d �} t j d d d! g d d" d# g �} t | | � t j | d d d | j
�} t j d$ d% g d d& d' g �} t | | � d S)(Nz1 2
3 4
5 6
7 8
9 10
�max_rowsr- rP r+ rU r� r� r� r r� r r z 1 1
2 2
0
3 3
4 4
5
6
7
� �?� @ry r� F� @� @z"a b
#c d
1 1
2 2
#0
3 3
4 4
5 5
rv r� TrV r� �<f8r� � @)rM rM )rN rN )rO rO )r� rQ )r� rQ )rP rP )rR rR )r� rQ )r� rQ ) r rC r} r r r� rW r r� r rV )r r� rO �a1�a2r) r* r� r r r! �
test_max_rows� s@ ( !
!3
!3 !
zTestFromTxt.test_max_rowsc C s� t j d � j d � } t j d d k r4 d } n d
} xq | D]i } d | d } t � �< } t | d
� � } | j | � Wd QRXt j | � } Wd QRXt | | � qA Wd S)Nr� r+ r- r r� �
�
z0 1 2z3 4 5r� )r+ r- )r� rV )r� rV rW )
rC rT �reshaper/ r0 r r� r" r} r )r r. �linesep�sepr� r= r� r/ r r r! �test_gft_using_filename� s
z#TestFromTxt.test_gft_using_filenamec C s8 d d � } t j | � � } t | t j d � � d S)Nc s s* x# t d � D] } t d | � Vq
Wd S)Nr z%d)rk r
)rh r r r! rS s z3TestFromTxt.test_gft_using_generator.<locals>.countr )rC r} r rT )r rS r/ r r r! �test_gft_using_generator s z$TestFromTxt.test_gft_using_generatorc C s� t d � } t j | d d �} t | j j d d d g � t | j d t j k � t | j d t j k � t | j d t j k � t
| d d � t | d d � t | d d � d S) Nz%73786976294838206464 17179869184 1024rV r� r� r� g Dl i ) r rC rs r rV r� r rX r �integerr )r r� r) r r r! �test_auto_dtype_largeint s z$TestFromTxt.test_auto_dtype_largeintN)Er' r( r) rf r\ r_ ru 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 r� r� r� r r r r r
r r r r r$ r) r3 r; r< r= r> r@ rC rD rE rI rK rU r[ r\ r^ r r r r! rn � s�
" 6 rn c @ s� e Z d Z e j j j e d k d � d d � � Z e j j j e d k d � d d � � Z e j j j e d k d � d d � � Z
e j j j e d k d � d d
� � Z e j j j e d k d � d d � � Z e j j j e d k d � d
d � � Z
e j j j e d k d � d d � � Z e j j j e d k d � d d � � Z e j j j e d k d � d d � � Z d S)�
TestPathUsageNzNo pathlib.Pathc C sv t d d � �` } t | � } t j d d g d d g g � } t j | | � t j | � } t | | � Wd QRXd S)Nr� z.txtg�������?r+ r- rU )r r
rC rW r� r� r )r �pathrS rc r r r! � test_loadtxt1 s !zTestPathUsage.test_loadtxtc C sy t d d � �c } t | � } t j d d g d d g g t � } t j | | � t j | � } t | | � Wd QRXd S)Nr� z.npyrP r+ r- rU )r r
rC rW rY rw rD r )r r` rS r� r r r! �test_save_load: s $z TestPathUsage.test_save_loadc C sf t d d � �P } t | � } t j | d d �t j | � � } t | d d � Wd QRXWd QRXd S)Nr� z.npzr� z place holder)r r
rC r{ rD r )r r` r� r r r! �test_savez_loadD s
z TestPathUsage.test_savez_loadc C sf t d d � �P } t | � } t j | d d �t j | � } t | d d � | j � Wd QRXd S)Nr� z.npzr� z place holder)r r
rC �savez_compressedrD r rB )r r` r� r r r! �test_savez_compressed_loadM s z(TestPathUsage.test_savez_compressed_loadc C sj t d d � �T } t | � } t j d d g � } t j | | � t j | � } t | | � Wd QRXd S) Nr� z.txtrP r+ r- rU )rP r+ )r- rU )r r
rC rW r� r} r )r r` rS r� r r r! �test_genfromtxtW s z TestPathUsage.test_genfromtxtc C s� t d d � �� } t | � } | j d � � } | j d � Wd QRXt j d d g d d g g d t �} t j | d t �} t | | � Wd QRXd S)
Nr� z.txtr� z1 2
3 4rP r+ r- rU rV ) r r
r� r" rC rW rY rs r )r r` r� r* r) r r r! �test_ndfromtxt` s 'z TestPathUsage.test_ndfromtxtc C s� t d d � �p } t | � } | j d � � } | j d � Wd QRXt j | d d �} t j d
d g � } t | | � Wd QRXd S)Nr� z.txtr� z1,2,3.0
4,5,6.0
r� r� � �?� @� @� @� @� @)rh ri rj )rk rl rm ) r r
r� r" rC r� r� rW r )r r` r� r) r* r r r! �test_mafromtxtl s z TestPathUsage.test_mafromtxtc C s� t d d � �� } t | � } | j d � � } | j d � Wd QRXt d d d d d d
� } t j | | � } t j d d g d d t j f d t j f g �} | j t
| t j � � t | | � Wd QRXd S)Nr� z.txtr� z A,B
0,1
2,3r� r� r� zN/Ar� Tr rP r+ r- rV r� r� )r rP )r+ r- )
r r
r� r" rw rC rF rW rY rG rA rH r )r r` r� rM r) r* r r r! rI x s !z TestPathUsage.test_recfromtxtc C s� t d d � �� } t | � } | j d � � } | j d � Wd QRXt d d d d d d � } t j | d
d | �} t j d d g d
d t j f d t j f g �} | j t
| t j � � t | | � Wd QRXd S)Nr� z.txtr� z A,B
0,1
2,3r� zN/Ar� TrJ rV r rP r+ r- r� r� )r rP )r+ r- )
r r
r� r" rw rC r� rW rY rG rA rH r )r r` r� rM r) r* r r r! rK � s !z TestPathUsage.test_recfromcsv)r' r( r) rC rs r r� r
ra rb rc re rf rg rn rI rK r r r r! r_ / s * *
* *
* * * *r_ c C s� t j j d � } t � } t j d | d d � } t j | | � | j � | j d � t j d | d d � } t t j | � | � d S)Nr� �fileobj�moder� r ra )r� r� )
rC r� r �gzip�GzipFilerw rB r? r rD )rS r r� r r r! �test_gzip_load� s
rs c C s� t � } t j d | d d � } | j d � | j � | j d � t d d � �B } t | d � � } | j | j � � Wd QRXt j
| � } Wd QRX| j � t | d d
d g � d S) Nro rp r� s 1 2 3
r r� z.gzr� rP r+ r- ) r rq rr r" rB r? r r� r� rC r� r )r r� r= r� r/ r r r! �test_gzip_loadtxt� s
rt c C s� t � } t j d | d d � } | j d � | j � | j d � t j d | d d � } t t j | � d d d g � d S)
Nro rp r� s 1 2 3
r ra rP r+ r- ) r rq rr r" rB r? r rC r� )r r� r r r! � test_gzip_loadtxt_from_string� s
ru c C s< t � } t j d � } t j d � } t j | d | d | �| j d � t j | � } t d | k � t d | k � t d | j � k � t d | j � k � x: | j � D], \ } } t | d k � t | j
d � q� Wt t | j � � d k � x | D] } t | d
k � qWt d | j � k � d S) Nr- rc re r r+ )r- r- )r- r- )rc re )r- r- )rc re ) r rC r� r{ r? rD r �keys�itemsr rR r )r rc re r@ r� rS r r r! �test_npzfile_dict� s"
rx c C s� t � } t j | d d d g � | j d � t t j � � t j � z' t j � t j | � t j � } Wd t j
� Xt | d � d S)NrP r+ r- r ) r rC r{ r? r �gc� isenabled�disable�collectrD �enabler )r� Zn_objects_in_cyclesr r r! �test_load_refcount� s
r~ �__main__)A�
__future__r r r r/ rq rJ r� �tempfiler r1 rm ry �ior r �numpyrC �numpy.mar� Znumpy.lib._iotoolsr r � numpy.compatr
r r r
Znumpy.ma.testutilsr Z
numpy.testingr r r r r r r r r r r r r r0 ZMAJVERZMINVER�maxsizer� r2 r] r5 rv rz r� r� rZ rn r_ rs rt ru rx r~ r' r r r r! �<module> sN "R k��� �&� � � �f