Файловый менеджер - Редактировать - /opt/imh-python/lib/python3.9/site-packages/influxdb/tests/__pycache__/dataframe_client_test.cpython-39.pyc
Ðазад
a p�h�� � @ s� d Z ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlZddl Z ddl Z ddl Z ddl m Z dd lmZmZ d d lmZ es�ddlZdd lmZ dd lmZ ddlZeG dd� de j��Z dS )zUnit tests for misc module.� )�absolute_import)�division)�print_function)�unicode_literals)� timedeltaN)�raises)� skip_if_pypy� using_pypy� )�_mocked_session)�assert_frame_equal)�DataFrameClientc @ s� e Zd ZdZdd� Zdd� Zdd� Zdd � Zd d � Zd d � Z dd� Z dd� Z dd� Z dd� Z dd� Zdd� Zdd� Zd d � Zd d � Zd d!� Zd"d#� Zee�d$d%� �Zee�d&d'� �Zd(d)� Zd*d+� Zee�d,d-� �Z d.d/� Z d0d1� Z d2d3� Z ee�d4d5� �Z d6d7� Z!d8d9� Z"d:d;� Z#d<d=� Z$ee�d>d?� �Z%d@dA� Z&ee�dBdC� �Z'dDdE� Z(dFdG� Z)dHdI� Z*dJdK� Z+dLdM� Z,dNdO� Z-dPdQ� Z.dRdS� Z/dTdU� Z0dVdW� Z1dXdY� Z2dZd[� Z3d\d]� Z4d^d_� Z5d`da� Z6dbS )c�TestDataFrameClientz%Set up a test DataFrameClient object.c C s t �dt� dS )z)Instantiate a TestDataFrameClient object.�errorN)�warnings� simplefilter� FutureWarning)�self� r ��/root/rpmbuild/BUILDROOT/imh-python39-modules-3.9.7-92.el8.x86_64/opt/imh-python/lib/python3.9/site-packages/influxdb/tests/dataframe_client_test.py�setUp s zTestDataFrameClient.setUpc C s� t �d�}t jg d�g d�g||tdd� gg d�d�}d}t�� �h}|jtjd d d � td d �}|� |d� | � |j j |� |j |ddd� | � |j j |� W d � n1 s�0 Y dS )�8Test write points from df in TestDataFrameClient object.�1970-01-01 00:00+00:00��1r � �?��2� � @r ��hours�� column_one� column_two� column_three��data�index�columns�t foo column_one="1",column_two=1i,column_three=1.0 0 foo column_one="2",column_two=2i,column_three=2.0 3600000000000 �http://localhost:8086/write�� �� status_code�db��database�fooN��tags� �pd� Timestamp� DataFramer � requests_mock�Mocker� register_uri�POSTr � write_points� assertEqual� last_request�body�r �now� dataframe�expected�m�clir r r � test_write_points_from_dataframe% s" �� � z4TestDataFrameClient.test_write_points_from_dataframec C s� t �d�}t jg d�g d�g||tdd� gg d�d�}d}t�� �H}|jtjd d d � td d �}|� |d� | � |j j |� W d � n1 s�0 Y dS )z7write_points should escape white space in measurements.r r r r r r" r& s� meas\ with\ space column_one="1",column_two=1i,column_three=1.0 0 meas\ with\ space column_one="2",column_two=2i,column_three=2.0 3600000000000 r+ r, r- r/ r0 zmeas with spaceNr5 rA r r r �7test_dataframe_write_points_with_whitespace_measurement? s �� � zKTestDataFrameClient.test_dataframe_write_points_with_whitespace_measurementc C s� t �d�}t jg d�g d�g||tdd� gg d�d�}d}t�� �H}|jtjd d d � td d �}|� |d� | � |j j |� W d � n1 s�0 Y dS )z7write_points should escape white space in column names.r r r r r )z column onez column twoz column threer& sz foo column\ one="1",column\ two=1i,column\ three=1.0 0 foo column\ one="2",column\ two=2i,column\ three=2.0 3600000000000 r+ r, r- r/ r0 r2 Nr5 rA r r r �;test_dataframe_write_points_with_whitespace_in_column_namesU s �� � zOTestDataFrameClient.test_dataframe_write_points_with_whitespace_in_column_namesc C s� t �d�}t jg d�g d�g||tdd� gg d�d�}d}t�� �h}|jtjd d d � td d �}|� |d� | � |j j |� |j |ddd� | � |j j |� W d � n1 s�0 Y dS )r r )r Nr �r r r r r r" r& sg foo column_one="1",column_three=1.0 0 foo column_one="2",column_two=2.0,column_three=2.0 3600000000000 r+ r, r- r/ r0 r2 Nr3 r5 rA r r r �*test_write_points_from_dataframe_with_nonei s" �� � z>TestDataFrameClient.test_write_points_from_dataframe_with_nonec C s� t �d�}t jg d�g d�g||tdd� gg d�d�}d}t�� �h}|jtjd d d � td d �}|� |d� | � |j j |� |j |ddd� | � |j j |� W d � n1 s�0 Y dS )r r �NNNrJ r r r" r& sA foo column_one="2",column_two=2.0,column_three=2.0 3600000000000 r+ r, r- r/ r0 r2 Nr3 r5 rA r r r �2test_write_points_from_dataframe_with_line_of_none� s" �� � zFTestDataFrameClient.test_write_points_from_dataframe_with_line_of_nonec C s� t �d�}t jg d�g d�g||tdd� gg d�d�}d}t�� �h}|jtjdd d � td d �}|� |d � | � |j j |� |j |d dd� | � |j j |� W d � n1 s�0 Y dS )r r rL r r r" r& � r+ r, r- r/ r0 r2 Nr3 r5 rA r r r �.test_write_points_from_dataframe_with_all_none� s" �� � zBTestDataFrameClient.test_write_points_from_dataframe_with_all_nonec C s� t �d�}t jg d�g d�g||tdd� gg d�d�}t�� �B}|jtjdd d � td d �}| � |j |d dd�� W d � n1 s�0 Y dS )zATest write points in batch from df in TestDataFrameClient object.r r r r r r" r& r+ r, r- r/ r0 r2 )Z batch_sizeN) r6 r7 r8 r r9 r: r; r< r Z assertTruer= �r rB rC rE rF r r r �+test_write_points_from_dataframe_in_batches� s � � z?TestDataFrameClient.test_write_points_from_dataframe_in_batchesc C s� t �d�}t jg d�g d�g||tdd� gg d�d�}d}t�� �v}|jtjd d d � td d �}|j |dddgd� | � |j j |� |j |dddgdd� | � |j j |� W d � n1 s�0 Y dS )z>Test write points from df w/tag in TestDataFrameClient object.r ��bluer r r r ��redr r r r r r ��tag_one�tag_twor# r$ r% r& s� foo,tag_one=blue,tag_two=1 column_one="1",column_two=1i,column_three=1.0 0 foo,tag_one=red,tag_two=0 column_one="2",column_two=2i,column_three=2.0 3600000000000 r+ r, r- r/ r0 r2 rW rX �� tag_columnsN�rZ r4 r5 rA r r r �1test_write_points_from_dataframe_with_tag_columns� s. ��� � ��zETestDataFrameClient.test_write_points_from_dataframe_with_tag_columnsc C s� t �d�}t jg d�g d�g||tdd� gg d�d�}d}t�� �V}|jtjd d d � td d �}|j |dddgddid� | � |j j |� W d � n1 s�0 Y dS )z>Test write points from df w/tag + cols in TestDataFrameClient.r rR rT r r rV r& s� foo,global_tag=value,tag_one=blue,tag_two=1 column_one="1",column_two=1i,column_three=1.0 0 foo,global_tag=value,tag_one=red,tag_two=0 column_one="2",column_two=2i,column_three=2.0 3600000000000 r+ r, r- r/ r0 r2 rW rX Z global_tag�valuer[ Nr5 rA r r r �>test_write_points_from_dataframe_with_tag_cols_and_global_tags� s( ��� � �zRTestDataFrameClient.test_write_points_from_dataframe_with_tag_cols_and_global_tagsc C s t �d�}t jg d�g d�g||tdd� gg d�d�}d}d }d }d }t�� ��}|jtjd d d� tdd�}|j |dddgdgd� | � |j j |� |j |dddgd� | � |j j |� |j |dg d�d� | � |j j |� |� |d� | � |j j |� W d � n1 �s0 Y dS )z?Test default write points from df w/tag in TestDataFrameClient.r )rS r r r r Zhot)rU r r r r Zcoldr r )rW rX r# r$ r% � tag_threer& sk foo,tag_one=blue column_one="1",column_two=1i 0 foo,tag_one=red column_one="2",column_two=2i 3600000000000 s� foo,tag_one=blue,tag_two=1 column_one="1",column_two=1i,column_three=1.0,tag_three="hot" 0 foo,tag_one=red,tag_two=0 column_one="2",column_two=2i,column_three=2.0,tag_three="cold" 3600000000000 s� foo,tag_one=blue,tag_three=hot,tag_two=1 column_one="1",column_two=1i,column_three=1.0 0 foo,tag_one=red,tag_three=cold,tag_two=0 column_one="2",column_two=2i,column_three=2.0 3600000000000 s� foo tag_one="blue",tag_two=1i,column_one="1",column_two=1i,column_three=1.0,tag_three="hot" 0 foo tag_one="red",tag_two=0i,column_one="2",column_two=2i,column_three=2.0,tag_three="cold" 3600000000000 r+ r, r- r/ r0 r2 r# r$ rW �� field_columnsrZ rX rY r" )ra Nr5 ) r rB rC Zexpected_tags_and_fieldsZexpected_tags_no_fieldsZexpected_fields_no_tagsZexpected_no_tags_no_fieldsrE rF r r r �;test_write_points_from_dataframe_with_tag_cols_and_defaults sH ��� � � � � ��� zOTestDataFrameClient.test_write_points_from_dataframe_with_tag_cols_and_defaultsc C s� t �d�}t jg d�g d�g d�g d�g d�gt j|ddd �g d �d �}d }t�� �V}|jtjd dd� tdd�}|j |dddgddgd� | � |j j |� W d � n1 s�0 Y dS )z?Test write points from df w/escaped tag in TestDataFrameClient.r )z blue oranger r zhot=cold)z red,greenr r z cold\fire)�somer r � )rc r r N)rd r r N�H� )�freq�periods)rW r# r$ r_ r& sL foo,tag_one=blue\ orange,tag_three=hot\=cold column_one="1",column_two=1i 0 foo,tag_one=red\,green,tag_three=cold\\fire column_one="2",column_two=2i 3600000000000 foo,tag_one=some column_one="2",column_two=2i 7200000000000 foo,tag_one=some column_one="2",column_two=2i 10800000000000 foo column_one="2",column_two=2i 14400000000000 r+ r, r- r/ r0 r2 r# r$ rW r_ r` N) r6 r7 r8 Z period_ranger9 r: r; r< r r= r> r? r@ )r rB rC Zexpected_escaped_tagsrE rF r r r �1test_write_points_from_dataframe_with_tag_escapedG s0 �� � � �zETestDataFrameClient.test_write_points_from_dataframe_with_tag_escapedc C s� t �d�}t jg d�g d�g||tdd� gd�}d}t�� �N}|jtjdd d � td d �}|� |d ddi� | � |j j |� W d � n1 s�0 Y dS )z,Test write points from df with numeric cols.r r r r r �r' r( sR foo,hello=there 0="1",1=1i,2=1.0 0 foo,hello=there 0="2",1=2i,2=2.0 3600000000000 r+ r, r- r/ r0 r2 �hello�thereNr5 rA r r r �:test_write_points_from_dataframe_with_numeric_column_namesr s �� � zNTestDataFrameClient.test_write_points_from_dataframe_with_numeric_column_namesc C s� t jtg d�g d�g d�g d�g d�g d�g d�d�t jt �d �d d d �d �}d}t�� �f}|jtjddd� t dd�}g d�}|j |j dd�|f dg d�d� | � |j j|� W d � n1 s�0 Y dS )zAwrite_points detect erroneous leading comma for null first field.)r NN� � )r NNN� )� gffffff@NN� )�oneNN�eightN)�twoNNNN)�three�fourNNN)zAll columns filledzFirst two of three emptyz All emptyzLast two of three emptyzEmpty tags with values)�first�second�third� first_tag� second_tag� third_tag�commentz 2018-01-01Z1Drf )�startrg rh �r( s� foo,first_tag=one,second_tag=two,third_tag=three comment="All columns filled",first=1.0,second=2.0,third=3.0 1514764800000000000 foo,third_tag=four comment="First two of three empty",third=4.1 1514851200000000000 foo comment="All empty" 1514937600000000000 foo,first_tag=eight comment="Last two of three empty",first=8.0 1515024000000000000 foo comment="Empty tags with values",first=9.0,second=10.0,third=11.0 1515110400000000000 r+ r, r- r/ r0 )r{ r| r} r~ rx ry rz Nr2 )r{ r| r} rY )r6 r8 �dictZ date_range� to_datetimer9 r: r; r< r r= �locr> r? r@ )r rC rD rE rF Zcolnamesr r r �9test_write_points_from_dataframe_with_leading_none_column� s: ���� � �zMTestDataFrameClient.test_write_points_from_dataframe_with_leading_none_columnc C s6 t �d�}t jg d�g d�g||tdd� gd�}tj�tj�dkrJd}nd }d }d }t� � ��}|j tj d d d � t dd�}|� |dddi� t|� t|jj� | �|jj|� t dd�}|j |dddidd� | �|jj|� t dd�}|j |dddidd� | �|jj|� W d � n1 �s(0 Y dS )z1Test write points from df with numeric precision.r )r r g@ �q ��?)r r g@ �q �@r r rj z1.13.3sf foo,hello=there 0="1",1=1i,2=1.11111111111 0 foo,hello=there 0="2",1=2i,2=2.22222222222 3600000000000 sj foo,hello=there 0="1",1=1i,2=1.1111111111111 0 foo,hello=there 0="2",1=2i,2=2.2222222222222 3600000000000 sX foo,hello=there 0="1",1=1i,2=1.1111 0 foo,hello=there 0="2",1=2i,2=2.2222 3600000000000 r+ r, r- r/ r0 r2 rk rl � )Znumeric_precision�fullN)r6 r7 r8 r �np�libZ NumpyVersion� __version__r9 r: r; r< r r= �printr? r@ r> )r rB rC Zexpected_default_precisionZ expected_specified_precisionZexpected_full_precisionrE rF r r r �7test_write_points_from_dataframe_with_numeric_precision� sF ������ � � �zKTestDataFrameClient.test_write_points_from_dataframe_with_numeric_precisionc C s� t jg d�g d�gt �d�t �d�gg d�d�}d}t�� �H}|jtjdd d � td d �}|�|d � | � |j j |� W d � n1 s�0 Y dS )z,Test write points from df with period index.r r z 1970-01-01z 1970-01-02r" r& su foo column_one="1",column_two=1i,column_three=1.0 0 foo column_one="2",column_two=2i,column_three=2.0 86400000000000 r+ r, r- r/ r0 r2 N) r6 r8 ZPeriodr9 r: r; r< r r= r> r? r@ )r rC rD rE rF r r r �2test_write_points_from_dataframe_with_period_index� s ��� � zFTestDataFrameClient.test_write_points_from_dataframe_with_period_indexc C s� t �d�}t jg d�g d�g||tdd� gg d�d�}t�� ��t}|jtjdd d � td d �}d }|j ||dd� | � |j j d dg� | � d|j j � |j ||dd� | � |j j d dg� | � d|j j � |j ||dd� | � |j j d dg� | � d|j j � |j ||dd� | � |j j d dg� | � d|j j � |j ||dd� | � |j j d dg� | � d|j j � |j ||dd� | � |j j d dg� | � d|j j � W d � n1 �s�0 Y d S ) z.Test write points from df with time precision.r r r r r r" r& r+ r, r- r/ r0 r2 �h�Ztime_precision� precisionsh foo column_one="1",column_two=1i,column_three=1.0 0 foo column_one="2",column_two=2i,column_three=2.0 1 rE si foo column_one="1",column_two=1i,column_three=1.0 0 foo column_one="2",column_two=2i,column_three=2.0 60 �ssk foo column_one="1",column_two=1i,column_three=1.0 0 foo column_one="2",column_two=2i,column_three=2.0 3600 �mssn foo column_one="1",column_two=1i,column_three=1.0 0 foo column_one="2",column_two=2i,column_three=2.0 3600000 �usq foo column_one="1",column_two=1i,column_three=1.0 0 foo column_one="2",column_two=2i,column_three=2.0 3600000000 �nr* N)r6 r7 r8 r r9 r: r; r< r r= r> r? �qsr@ )r rB rC rE rF � measurementr r r �4test_write_points_from_dataframe_with_time_precision s` � � ������zHTestDataFrameClient.test_write_points_from_dataframe_with_time_precisionc C sr t jg d�g d�gg d�d�}t�� �8}|jtjddd� tdd �}|�|d � W d � n1 sd0 Y d S ) z4Test failed write points from df without time index.r r r" )r' r) �"http://localhost:8086/db/db/seriesr, r- r/ r0 r2 N)r6 r8 r9 r: r; r< r r= )r rC rE rF r r r �9test_write_points_from_dataframe_fails_without_time_indexX s � � zMTestDataFrameClient.test_write_points_from_dataframe_fails_without_time_indexc C s~ t �d�}t jddg||tdd� gd�}t�� �8}|jtjddd � td d �}|� |d � W d � n1 sp0 Y d S )z-Test failed write points from df with series.r r r r r rj r� r, r- r/ r0 r2 N) r6 r7 ZSeriesr r9 r: r; r< r r= rP r r r �2test_write_points_from_dataframe_fails_with_seriesg s � � zFTestDataFrameClient.test_write_points_from_dataframe_fails_with_seriesc C sj t dd�}t�� �D}|jtjddd� |�d� | �|jjd d d � W d � n1 s\0 Y d S ) z3Test create database for TestInfluxDBClient object.r/ r0 �http://localhost:8086/query�{"results":[{}]}��text�new_db�qr zcreate database "new_db"N� r r9 r: r; r< �create_databaser> r? r� �r rF rE r r r �test_create_databasev s � �z(TestDataFrameClient.test_create_databasec C sj t dd�}t�� �D}|jtjddd� |�d� | �|jjd d d � W d � n1 s\0 Y d S ) z<Test create db w/numeric name for TestInfluxDBClient object.r/ r0 r� r� r� �123r� r zcreate database "123"Nr� r� r r r �"test_create_numeric_named_database� s � �z6TestDataFrameClient.test_create_numeric_named_databasec C sD t dd�}t|dd�� |�d� W d � n1 s60 Y dS )z8Test create database fail for TestInfluxDBClient object.r/ r0 �post� r� N)r r r� �r rF r r r �test_create_database_fails� s z.TestDataFrameClient.test_create_database_failsc C sj t dd�}t�� �D}|jtjddd� |�d� | �|jjd d d � W d � n1 s\0 Y d S ) z1Test drop database for TestInfluxDBClient object.r/ r0 r� r� r� r� r� r zdrop database "new_db"N� r r9 r: r; r< Z drop_databaser> r? r� r� r r r �test_drop_database� s � �z&TestDataFrameClient.test_drop_databasec C sj t dd�}t�� �D}|jtjddd� |�d� | �|jjd d d � W d � n1 s\0 Y d S ) z4Test drop measurement for TestInfluxDBClient object.r/ r0 r� r� r� Znew_measurementr� r z"drop measurement "new_measurement"N) r r9 r: r; r< Zdrop_measurementr> r? r� r� r r r �test_drop_measurement� s � �z)TestDataFrameClient.test_drop_measurementc C sj t dd�}t�� �D}|jtjddd� |�d� | �|jjd d d � W d � n1 s\0 Y d S ) z3Test drop numeric db for TestInfluxDBClient object.r/ r0 r� r� r� r� r� r zdrop database "123"Nr� r� r r r � test_drop_numeric_named_database� s � �z4TestDataFrameClient.test_drop_numeric_named_databasec C sF t dddd�}t|dd�� |�� W d � n1 s80 Y dS )z8Test get list of dbs fail for TestInfluxDBClient object.�host� �username�password�getr� N)r r �get_list_databaser� r r r � test_get_list_database_fails� s z0TestDataFrameClient.test_get_list_database_failsc C s� t dd�}ddddgdgdggd �gigi}t|d d t�|���, | �|�� ddiddig� W d � n1 sr0 Y d S ) z<Test get list of measurements for TestInfluxDBClient object.r/ r0 �results�series� measurements�name�cpuZdisk�r� r) �valuesr� �� N)r r �json�dumps�assertListEqualZget_list_measurements)r rF r' r r r �test_get_list_measurements� s ���� �z.TestDataFrameClient.test_get_list_measurementsc C sx t dd�}d}t�� �N}|jtjd|d� |jdddd dd � | �|jjd d d � W d � n1 sj0 Y dS )z=Test create default ret policy for TestInfluxDBClient object.r/ r0 r� r� r� �somename�1dr� T)�defaultr1 r� r z^create retention policy "somename" on "db" duration 1d replication 4 shard duration 0s defaultN� r r9 r: r; r< Zcreate_retention_policyr> r? r� �r rF Zexample_responserE r r r �$test_create_retention_policy_default� s � ��z8TestDataFrameClient.test_create_retention_policy_defaultc C sv t dd�}d}t�� �L}|jtjd|d� |jddddd� | �|jjd d d � W d � n1 sh0 Y d S ) z;Test create retention policy for TestInfluxDBClient object.r/ r0 r� r� r� r� r� r� r� r zVcreate retention policy "somename" on "db" duration 1d replication 4 shard duration 0sNr� r� r r r � test_create_retention_policy� s ���z0TestDataFrameClient.test_create_retention_policyc C s� t dd�}d}t�� ��}|jtjd|d� |jdddd� | �|jjd d d � |jddd d � | �|jjd d d� |jdddd� | �|jjd d d� |jdddd� | �|jjd d d� W d � n1 s�0 Y dS )z:Test alter retention policy for TestInfluxDBClient object.r/ r0 r� r� r� r� Z4d)�durationr� r z5alter retention policy "somename" on "db" duration 4dr� )Z replicationz7alter retention policy "somename" on "db" replication 4Z1h)Zshard_durationz;alter retention policy "somename" on "db" shard duration 1hT)r� z1alter retention policy "somename" on "db" defaultN) r r9 r: r; r< �alter_retention_policyr> r? r� r� r r r �test_alter_retention_policy sH ���������z/TestDataFrameClient.test_alter_retention_policyc C sJ t dddd�}t|dd�� |�dd� W d � n1 s<0 Y d S ) z<Test invalid alter ret policy for TestInfluxDBClient object.r� r� r� r� r� i� r� r/ N)r r r� r� r r r �#test_alter_retention_policy_invalid6 s z7TestDataFrameClient.test_alter_retention_policy_invalidc C sp t dd�}d}t�� �F}|jtjd|d� |�dd� | �|jjd d d � W d � n1 sb0 Y d S ) z9Test drop retention policy for TestInfluxDBClient object.r/ r0 r� r� r� r� r� r z(drop retention policy "somename" on "db"N) r r9 r: r; r< �drop_retention_policyr> r? r� r� r r r �test_drop_retention_policy= s � �z.TestDataFrameClient.test_drop_retention_policyc C sJ t dddd�}t|dd�� |�dd� W d � n1 s<0 Y d S ) z:Test failed drop ret policy for TestInfluxDBClient object.r� r� r� r� �deleter� r� r/ N)r r r� r� r r r � test_drop_retention_policy_failsN s z4TestDataFrameClient.test_drop_retention_policy_failsc C sh t dd�}d}t�� �>}|jtjd|d� | �|�d�dddd �g� W d � n1 sZ0 Y d S ) z:Test get retention policies for TestInfluxDBClient object.r/ r0 zq{"results": [{"series": [{"values": [["fsfdsdf", "24h0m0s", 2]], "columns": ["name", "duration", "replicaN"]}]}]}r� r� Z24h0m0sZfsfdsdfr )r� r� ZreplicaNN)r r9 r: r; �GETr� Zget_list_retention_policiesr� r r r � test_get_list_retention_policiesU s � ���z4TestDataFrameClient.test_get_list_retention_policiesc C s> dddddiddgdd ggd �ddd iddgdd gdd gdd ggd �gigi}t jd ggdgt �dg�d �}|jjd u r�|j�d�|_t jd gd gd ggdgt �g d��d �}|jjd u r�|j�d�|_||d�}tddddd�}t|dd|��8 |�d�}|D ]}t || || � �q W d � n1 �s00 Y d S )z2Test query into df for TestDataFrameClient object.r� r� �network� directionrd �timer] �2009-11-10T23:00:00Z�~[ )r� r4 r) r� �in�r) r( N�UTC)r� r� r� ))r� ))r� rd )r� ))r� r� r� r� r� r� r/ r� r� z-select value from network group by direction;) r6 r8 r� r( �tzinfo� tz_localizer r �queryr )r r' �pd1�pd2rD rF �result�kr r r �test_query_into_dataframeh sL ������ � � � z-TestDataFrameClient.test_query_into_dataframec C sd dddddgddgddgd d ggd �gidddd gd dggd �gigi}t jdgdgd ggdgt �g d��d�}|jjdu r�|j�d�|_t jdggd gt �dg�d�}|jjdu r�|j�d�|_d|id|ig}tddddd�}d}dd i}t|d d |��R |j||d �}t ||�D ](\} } | D ]} t | | | | � �q"�qW d � n1 �sV0 Y dS ) �7Test multiquery into df for TestDataFrameClient object.r� r� �cpu_load_shortr� r] � 2015-01-29T21:55:43.702900257Z皙�����?r� �2015-06-11T20:46:02Z�{�G�z�?r� �count�1970-01-01T00:00:00Zrq � ���@��"2015-01-29 21:55:43.702900257+0000r� z2015-06-11 20:46:02+0000r� Nr� �1970-01-01 00:00:00+00:00r� r� r� r� r/ �rSELECT value FROM cpu_load_short WHERE region=$region;SELECT count(value) FROM cpu_load_short WHERE region=$region�region�us-westr� r� �� bind_params) r6 r8 r� r( r� r� r r r� �zipr ) r r' r� r� rD rF �iqlr� r� �r�er� r r r � test_multiquery_into_dataframe� sT ���� ������ � � z2TestDataFrameClient.test_multiquery_into_dataframec C st dddg d�dddt jgddd t jgd d d t jggd �giddddgddggd �gigi}tjddt jgdd t jgd d t jggg d�t�g d��d�}|jjdu r�|j�d�|_tjddgdd gd d ggddgt�g d��d�}|jjdu r�|j�d�|_tjdggdgt�dg�d�}|jjdu �r4|j�d�|_d|id|ig}d|id|ig}tdd d d d �}d }d!d"i} d#D ]�} t |d$d%|��b |j || | d&�} | �r�|n|} t | | �D ](\} }|D ]}t || | | � �q��q�W d � n1 �s�0 Y �qtt |d$d%|��R |j || d'�} t | |�D ](\} }|D ]}t || | | � �q2�q&W d � n1 �sf0 Y dS )(r� r� r� r� )r� r] �value2�value3r� r� g��K7�A�?r� i�� r� r� gO��e�c�?r� r� r� r� rq r� )r] r� r� r� r� Nr� r] r� r� r� r� r� r� r/ r� r� r� )TFr� r� )r� �dropnar� ) r� �NaNr6 r8 r� r( r� r� r r r� r� r )r r' r� Z pd1_dropnar� Zexpected_dropna_trueZexpected_dropna_falserF r� r� r� r� rD r� r� r� r r r �%test_multiquery_into_dataframe_dropna� s� ������������� � � � ��� �>z9TestDataFrameClient.test_multiquery_into_dataframe_dropnac C s^ t ddddd�}t|dddi gi��& |�d �}| �|i � W d � n1 sP0 Y d S ) z<Test query with empty results in TestDataFrameClient object.r� r� r� r� r/ r� r� r� zselect column_one from foo;N)r r r� r> )r rF r� r r r � test_query_with_empty_result' s z0TestDataFrameClient.test_query_with_empty_resultc C s� ddddgdggdgd�gigi}t dd d d d �}t|d dt�|���, | �|�� ddiddig� W d � n1 sx0 Y dS )z9Test get list of databases in TestDataFrameClient object.r� r� Z databasesZnew_db_1Znew_db_2r� )r� r� r) r� r� r� r� r/ r� r� N)r r r� r� r� r� )r r' rF r r r �test_get_list_database. s ���� �z*TestDataFrameClient.test_get_list_databasec C s� t �d�}tddddd�}| �|�|�d� | �|j|dd �d � | �|j|d d �d � | �|j|d d �d� | �|j|dd �d� | �|j|dd �d� | �|j|dd �d� dS )z=Test convert datetime to epoch in TestDataFrameClient object.z 2013-01-01 00:00:00.000+00:00r� r� r� r� r/ g ��8�Ar� r� g �ArE g ��uAr� r� g ��6�sBr� g _>�HCr� g ���ղCN)r6 r7 r r> Z_datetime_to_epoch)r � timestamprF r r r �test_datetime_to_epoch@ s< � � � � � � �z*TestDataFrameClient.test_datetime_to_epochc C s( t �d�}| �|t � | �d|j� dS )z;Test data source name deconstructor in TestDataFrameClient.zinfluxdb://localhost:8086zhttp://localhost:8086N)r Zfrom_dsnZassertIsInstancer> Z_baseurl)r �clientr r r �test_dsn_constructorb s z(TestDataFrameClient.test_dsn_constructorc C s� t �d�}t jddtjgddtjgg||tdd� gg d�d�}d }t�� �n}|j tj d d d � t d d�}|j |ddd� | � |jj|� |j |dddd� | � |jj|� W d � n1 s�0 Y dS )z0Test write points from dataframe with Nan lines.r r r r r r r" r& �R foo column_one="1",column_two=1i 0 foo column_one="2",column_two=2i 3600000000000 r+ r, r- r/ r0 r2 �line��protocolN�r4 r �r6 r7 r8 r� �inf�nanr r9 r: r; r< r r= r> r? r@ rA r r r �.test_write_points_from_dataframe_with_nan_lineh s" �� � zBTestDataFrameClient.test_write_points_from_dataframe_with_nan_linec C s� t �d�}t jddtjgddtjgg||tdd� gg d�d�}d }t�� �n}|j tj d d d � t d d�}|j |ddd� | � |jj|� |j |dddd� | � |jj|� W d � n1 s�0 Y dS )z+Test write points from json with NaN lines.r r r r r r r" r& r r+ r, r- r/ r0 r2 r� r Nr r rA r r r �.test_write_points_from_dataframe_with_nan_json� s" �� � zBTestDataFrameClient.test_write_points_from_dataframe_with_nan_jsonc C s� t �d�}t jddddtjgddddtjgg||tdd � gg d �d �}d }t�� �z}|j tj d dd� t dd�}|j |ddddgd� | � |jj|� |j |dddddgd� | � |jj|� W d � n1 s�0 Y dS )z9Test write points from dataframe with NaN lines and tags.r rS r r rU r r r r rV r& � foo,tag_one=blue,tag_two=1 column_one="1",column_two=1i 0 foo,tag_one=red,tag_two=0 column_one="2",column_two=2i 3600000000000 r+ r, r- r/ r0 r2 r rW rX �r rZ N�r4 r rZ r rA r r r �7test_write_points_from_dataframe_with_tags_and_nan_line� s. ��� � � �zKTestDataFrameClient.test_write_points_from_dataframe_with_tags_and_nan_linec C s� t �d�}t jddddtjgddddtjgg||tdd � gg d �d �}d }t�� �z}|j tj d dd� t dd�}|j |ddddgd� | � |jj|� |j |dddddgd� | � |jj|� W d � n1 s�0 Y dS )z4Test write points from json with NaN lines and tags.r rS r r rU r r r r rV r& r r+ r, r- r/ r0 r2 r� rW rX r Nr r rA r r r �7test_write_points_from_dataframe_with_tags_and_nan_json� s. ��� � � �zKTestDataFrameClient.test_write_points_from_dataframe_with_tags_and_nan_jsonc C s� dddg d�g d�g d�g d�gd�gigi}t d d d d d �}d}ddi}t|dd|��L |j||dd gd�}|d }t|� | �dd gt|jj�� W d � n1 s�0 Y dS )z Test query with custom indexes.r� r� r� )r� r] r� )r r� �local)r r� r )rq r� r r� r� r� r� r� r/ r� r� r� r� r� r� )r� Zdata_frame_indexN)r r r� r� r� �listr( �names)r r' rF r� r� r� Z _data_framer r r �test_query_custom_index� s2 ������� �z+TestDataFrameClient.test_query_custom_indexc C s� dt d�ddgg d�g d�d�}tj�|�}g d�|d < tj|d d d �|d < |�d �}d }t�� �L}|jtj d dd� t dd�}|j |ddd� | � |j j|� W d � n1 s�0 Y dS )zTest nanosecond precision.皙�����?r �ffffff @皙����@)r g������@r r )Zone_oneZtwo_twoZ three_threeZ four_four)Z nanFloatsZ onlyFloats�strings)z#2019-10-04 06:27:19.850557111+00:00z#2019-10-04 06:27:19.850557184+00:00z#2019-10-04 06:27:42.251396864+00:00z#2019-10-04 06:27:42.251396974+00:00r� �ns)�units foo nanFloats=1.1,onlyFloats=1.1,strings="one_one" 1570170439850557111 foo onlyFloats=2.2,strings="two_two" 1570170439850557184 foo nanFloats=3.3,onlyFloats=3.3,strings="three_three" 1570170462251396864 foo nanFloats=4.4,onlyFloats=4.4,strings="four_four" 1570170462251396974 r+ r, r- r/ r0 r2 r� r� N)�floatr6 r8 � from_dictr� Z set_indexr9 r: r; r< r r= r> r? r@ )r Z for_df_dict�dfrD rE rF r r r �#test_dataframe_nanosecond_precision� s& � � � z7TestDataFrameClient.test_dataframe_nanosecond_precisionc C s t �d�}t �d�}t �||t �dd� t �dd��}t �dt|��}tjd|i|d�}t�� ��}|j tj dd d � t d d �}|j |d dd� |j j�d��d�}| �t|�d� t|�D ]<\} } | dkr�| �| d� q�| �| d| � dd| d��� q�W d � n1 �s0 Y dS )z1Test nanosecond precision within one microsecond.z 2019-10-04T06:27:19.850557000z 2019-10-04T06:27:19.850558000r r g r] r� r+ r, r- r/ r0 r2 r� r� zutf-8� i� i� rd z foo value=z.0 157017043985055iX Z04N)r� Z datetime64ZarangeZ timedelta64�lenr6 r8 r9 r: r; r< r r= r? r@ �decode�splitr> � enumerate) r r �endZ timestampsr� r rE rF �linesr( r r r r �3test_dataframe_nanosecond_precision_one_microsecond s6 � � �zGTestDataFrameClient.test_dataframe_nanosecond_precision_one_microsecondN)7�__name__� __module__� __qualname__�__doc__r rG rH rI rK rM rO rQ r\ r^ rb ri rm r� r� r� r� r � TypeErrorr� r� r� r� � Exceptionr� 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 sn G+A6A * ,7\" ""r ) r+ � __future__r r r r �datetimer r� Zunittestr r9 Z nose.toolsr Zinfluxdb.testsr r Z client_testr Zpandasr6 Zpandas.util.testingr Zinfluxdbr �numpyr� ZTestCaser r r r r �<module> s&
| ver. 1.1 | |
.
| PHP 8.3.30 | Ð“ÐµÐ½ÐµÑ€Ð°Ñ†Ð¸Ñ Ñтраницы: 0 |
proxy
|
phpinfo
|
ÐаÑтройка