Файловый менеджер - Редактировать - /home/digitalm/venv/lib/python3.7/site-packages/pandas/tests/io/sas/__pycache__/test_sas7bdat.cpython-37.pyc
Назад
B �5�g 2 � @ s d dl m Z d dlZd dlZd dlmZ d dlZd dlZd dl Z d dl m Z m Z d dl m mZ d dlZd dlmZ e j�d�G dd� d��Zdd � Zd d � Zd d � Zdd� Z dd� Z dd� Z dd� Z dd� Z dd� Z!dd� Z"d d � Z#d d � Z$d d!� Z%d"d#� Z&dS )$� )�datetimeN)�Path)�EmptyDataError�PerformanceWarningz*ignore:can't resolve package:ImportWarningc @ s� e Zd Zejdd�dd� �Zejjdd� �Zejjdd� �Z ejjd d � �Z ejjd d � �Z e � d �ejjdd� ��Zejjdd� �Zdd� ZdS )� TestSAS7BDATT)Zautousec C s |ddd�| _ g | _ttdd��dgg| _x�dD ]�}tj�| j d|� d��}t� |�}t d dd�}tj |d d d �}|| |d < tj |d d d �}|| |d < x\t|j d �D ]J}|j d d �|f } | jtjkr�|j d d �|f �tj�|j d d �|f< q�W | j�|� q0W d S )N�io�sas�data� � )r � Ztest_sas7bdat_z.csvi� ZColumn4�d)�unitZColumn12)�dirpathr �list�range�test_ix�os�path�join�pd�read_csvr Z to_timedelta�shape�ilocZdtype�npZint64�astype�float64�append) �self�datapath�j�fname�df�epoch�t1�t2�k�col� r( �W/home/digitalm-up/venv/lib/python3.7/site-packages/pandas/tests/io/sas/test_sas7bdat.py� setup_method s ,zTestSAS7BDAT.setup_methodc C sb x\dD ]T}| j | }xD| j| D ]6}tj�| jd|� d��}tj|dd�}t� ||� q W qW d S )N)r r �testz .sas7bdatzutf-8)�encoding) r r r r r r r �read_sas�tm�assert_frame_equal)r r �df0r& r! r" r( r( r) �test_from_file* s zTestSAS7BDAT.test_from_filec C s� x�dD ]�}| j | }x�| j| D ]z}tj�| jd|� d��}t|d��}|�� }W d Q R X t� |�}t j |dddd��}|�� } W d Q R X t j | |d d � q W qW d S ) N)r r r+ z .sas7bdat�rb�sas7bdatTzutf-8)�format�iteratorr, F)� check_exact)r r r r r r �open�readr �BytesIOr r- r. r/ ) r r r0 r& r! �fZbyts�buf�rdrr" r( r( r) �test_from_buffer3 s z TestSAS7BDAT.test_from_bufferc C s� x�dD ]�}| j | }x�| j| D ]�}tj�| jd|� d��}tj|ddd��V}|�d�}t � ||j dd�d d �f � |�d �}t � ||j dd �d d �f � W d Q R X q W qW d S ) N)r r r+ z .sas7bdatTzutf-8)r5 r, r r � � ) r r r r r r r r- r8 r. r/ r )r r r0 r& r! r< r" r( r( r) �test_from_iteratorB s z TestSAS7BDAT.test_from_iteratorc C sf x`dD ]X}| j | }xH| j| D ]:}ttj�| jd|� d���}tj|dd�}t � ||� q W qW d S )N)r r r+ z .sas7bdatzutf-8)r, ) r r r r r r r r r- r. r/ )r r r0 r& r! r" r( r( r) �test_path_pathlibN s z TestSAS7BDAT.test_path_pathlibzpy.pathc C sr ddl m} x`dD ]X}| j| }xH| j| D ]:}|tj�| jd|� d���}tj |dd�}t � ||� q,W qW d S )Nr )�local)r r r+ z .sas7bdatzutf-8)r, ) Zpy.pathrB r r r r r r r r- r. r/ )r Z LocalPathr r0 r& r! r" r( r( r) �test_path_localpathW s z TestSAS7BDAT.test_path_localpathc C s� x�dD ]�}x�| j | D ]t}xndD ]f}tj�| jd|� d��}tj||dd��&}d}x|D ]}||jd 7 }qXW W d Q R X ||jks t �q W qW qW d S )N)r r )r> r? � � r+ z .sas7bdatzutf-8)� chunksizer, r ) r r r r r r r- r � row_count�AssertionError)r r r& rF r! r< �y�xr( r( r) �test_iterator_loopc s z TestSAS7BDAT.test_iterator_loopc C s� | j d d }tj�| jd|� d��}tj|dddd��}|�|jd �}W d Q R X tj|ddd ��}|�|jd �}W d Q R X t � ||� d S ) Nr r+ z .sas7bdatr3 Tzutf-8)r4 r5 r, � )r5 r, ) r r r r r r r- r8 rG r. r/ )r r& r! r< Zd1Zd2r( r( r) �test_iterator_read_too_muchr s z(TestSAS7BDAT.test_iterator_read_too_muchN)�__name__� __module__� __qualname__�pytestZfixturer* �markZslowr1 r= r@ rA �tdZ skip_if_norC rK rM r( r( r( r) r s r c C s� | dddd�}t �|�}t j|dd�}x<|jD ]2}y|| j�d�||< W q. tk r^ Y q.X q.W t�||� ddlm } ||d d �}|� � }|� � x*t |j|j�D ]\}} || �� ks�t �q�W d S ) Nr r r ztest1.sas7bdatzutf-8)r, r )�SAS7BDATReaderF)Zconvert_header_text)r r- �columns�str�decode�AttributeErrorr. r/ Zpandas.io.sas.sas7bdatrT r8 �close�ziprH ) r r! Zdf1Zdf2r' rT r< Zdf3rJ rI r( r( r) �test_encoding_options� s r[ c C sj | dddd�}t j|dd�}| dddd�}t j|dgd �}d d d d g}|| �tj�||< t�||� d S )Nr r r zproductsales.sas7bdatzutf-8)r, zproductsales.csvZMONTH)� parse_datesZACTUALZPREDICTZQUARTERZYEAR)r r- r r r r r. r/ )r r! r" r0 Zvnr( r( r) �test_productsales� s r] c C sL | dddd�}t �|�}| dddd�}t �|�}|�tj�}t�||� d S )Nr r r ztest_12659.sas7bdatztest_12659.csv)r r- r r r r r. r/ )r r! r" r0 r( r( r) � test_12659� s r^ c C sP | dddd�}t �|�}| dddd�}t �|�}|�tj�}tj||dd� d S )Nr r r zairline.sas7bdatz airline.csvF)r6 )r r- r r r r r. r/ )r r! r" r0 r( r( r) � test_airline� s r_ c C sv | dddd�}t �|�}| dddd�}t j|dddd d gd �}|jd d �d f j�d �|jd d �d f< t�||� d S )Nr r r zdatetime.sas7bdatz datetime.csvZDate1ZDate2�DateTimeZ DateTimeHiZTaiw)r\ r> �us)r r- r r �dt�roundr. r/ )r r! r" r0 r( r( r) �test_date_time� s (rd c C sh | dddd�}t j|dd�}|d }|d �� }tj||dd � |d }|d �� }tj||dd � d S ) Nr r r z cars.sas7bdatzlatin-1)r, ZWGTT)r6 ZCYL)r r- rc r. Zassert_series_equal)r r! r" �result�expectedr( r( r) � test_compact_numerical_values� s rg c C sj | dddd�}d }|st }t�|�� tj|dd�}W d Q R X | dddd�}tj|dd�}t�||� d S )Nr r r zmany_columns.sas7bdatzlatin-1)r, zmany_columns.csv)r r. Zassert_produces_warningr r- r r/ )r Zusing_array_managerr! Zexpected_warningr" r0 r( r( r) �test_many_columns� s rh c C s0 | dddd�}t j|dd�}t|�dks,t�d S )Nr r r zload_log.sas7bdatzlatin-1)r, i1 )r r- �lenrH )r r! r" r( r( r) � test_inconsistent_number_of_rows� s rj c C s6 | dddd�}t jtdd�� t�|� W d Q R X d S )Nr r r zzero_variables.sas7bdatz No columns to parse from file)�match)rQ �raisesr r r- )r r! r( r( r) �test_zero_variables� s rm c C s: | dddd�}d}t jt|d�� t�|� W d Q R X d S )Nr r r zcorrupt.sas7bdatz4'SAS7BDATReader' object has no attribute 'row_count')rk )rQ rl rX r r- )r r! �msgr( r( r) �test_corrupt_read� s ro c C sj t | t�r*| jtt| jd�d �d d�S t | t�rbtjj | d�}|jtt|jd�d �d d�S | S d S )N���i� )� microsecond)Ztimestr) � isinstancer �replace�intrc rq rV �dateutil�parser�parse)�tsZ_tsr( r( r) �round_datetime_to_ms� s ry c C s | dddd�}t j|dd�}|�dd� �}y|d j�d �|d < W nJ t jjjjk rh |�t �}Y n& t k r� |d � t �|d < Y nX t j d d gd dgt ddddddd�t ddddddd�gddgt ddd�t ddd�gd�dd d d d gd �}t�||� d S ) Nr r r zmax_sas_date.sas7bdatz iso-8859-1)r, c S s t | t�r| �� S | S )N)rr rV �lstrip)rJ r( r( r) �<lambda> � z#test_max_sas_date.<locals>.<lambda>�dt_as_dtra �max�normalg���`�MBg�� ��Ai' � � � �; iX> i� � r g �qgFAg �@�@)�text� dt_as_floatr} � date_as_float� date_as_dater� r� r� r� )rU )r r- �applymaprb rc �_libs�tslibs� np_datetime�OutOfBoundsDatetimery rX �apply� DataFramer r. r/ )r r! r" rf r( r( r) �test_max_sas_date� s$ r� c C sZ dddddg}| dddd �}g }x�t j|d d d �D ]�}|�d d� �}y|d j�d�|d< W nJ t jjjjk r� |�t �}Y n& t k r� |d � t �|d< Y nX |j ddd� |� |� q2W t jdgdgtddddddd�gdgtddd�gd�|d �t jd gd gt�d �gd gt�d!�gd�|d �g}x$t||�D ]\}}t�||� �q<W d S )"Nr� r� r} r� r� r r r zmax_sas_date.sas7bdatz iso-8859-1r )r, rF c S s t | t�r| �� S | S )N)rr rV rz )rJ r( r( r) r{ * r| z,test_max_sas_date_iterator.<locals>.<lambda>ra T)ZinplaceZdropr~ g���`�MBi' r� r� r� r� iX> g �qgFA)r� r� r} r� r� )rU r g�� ��Az2019-08-01 23:59:59.999g �@�@z 2019-08-01)r r- r� rb rc r� r� r� r� ry rX r� Z reset_indexr r� r r Z datetime64rZ r. r/ )r Z col_orderr! �resultsr" rf re r( r( r) �test_max_sas_date_iterator s: r� c C s` | dddd�}t j|dd�}t �tddd �t jgtddd d d d d �t jgd ��}t�||� d S )Nr r r zdates_null.sas7bdatzutf-8)r, i' r� r� r� r� iQ> )ZdatecolZ datetimecol)r r- r� r ZNaTr. r/ )r r! r" rf r( r( r) �test_null_dateN s r� )'r r r �pathlibr Zdateutil.parserru �numpyr rQ Z pandas.errorsr r Z pandas.util._test_decorators�utilZ_test_decoratorsrS Zpandasr Zpandas._testingZ_testingr. rR �filterwarningsr r[ r] r^ r_ rd rg rh rj rm ro ry r� r� r� r( r( r( r) �<module> s2 k #.
| ver. 1.4 |
Github
|
.
| PHP 8.3.23 | Генерация страницы: 0.03 |
proxy
|
phpinfo
|
Настройка