async def test_write_non_string_identifier_and_tags(client): point = dict(tags={1: 2}, fields={3: 4}) with pytest.warns(UserWarning): assert await client.write(point, measurement='my_measurement') resp = await client.query('SELECT * FROM my_measurement') logger.info(resp) assert len(resp['results'][0]['series'][0]['values']) == 1
async def test_decorator(client): @lineprotocol class MyPoint(NamedTuple): measurement: aioinflux.MEASUREMENT time: aioinflux.TIMEINT host: aioinflux.TAG running: aioinflux.BOOL users: aioinflux.INT cpu_load: aioinflux.FLOAT cpu_load_level: aioinflux.ENUM cpu_load_level_tag: aioinflux.TAGENUM uuid: aioinflux.STR p = MyPoint( measurement="dp", time=1500, host="us1", running=True, users=1000, cpu_load=99.5, cpu_load_level=CpuLoad.HIGH, cpu_load_level_tag=CpuLoad.LOW, uuid=str(uuid.uuid4()), ) assert p assert hasattr(p, 'to_lineprotocol') assert await client.write(p) logger.info(await client.query('SELECT * FROM dp')) logger.info(await client.query("SHOW FIELD KEYS FROM dp"))
def test_read_dataframe_multistatement(df_client): df_list = df_client.query('SELECT max(*) from m1;SELECT min(*) from m2') logger.info(df_list) assert type(df_list) is list assert 'm1' in df_list[0] assert 'm2' in df_list[1] assert df_list[0]['m1'].shape == (1, 5) assert df_list[1]['m2'].shape == (1, 5)
async def test_iterpoints_with_parser(iter_client): r = await iter_client.query("SELECT * FROM cpu_load LIMIT 3") parser = lambda *x, meta: dict(zip(meta['columns'], x)) # noqa for i in iterpoints(r, parser): logger.info(i) assert 'time' in i assert 'value' in i assert 'host' in i
def test_read_dataframe_groupby(df_client): df_dict = df_client.query('SELECT max(*) from /m[1-2]$/ GROUP BY "tag"') s = ['\n{}:\n{}'.format(k, v) for k, v in df_dict.items()] logger.info('\n'.join(s)) m1 = pd.concat([df for k, df in df_dict.items() if k.split(',')[0] == 'm1']) m2 = pd.concat([df for k, df in df_dict.items() if k.split(',')[0] == 'm2']) assert m1.shape == (5, 6) assert m2.shape == (5, 6)
async def test_write_to_non_default_db(client): points = [p for p in utils.random_points(5)] await client.create_database(db='temp_db') assert client.db != 'temp_db' assert await client.write(points, db='temp_db') resp = await client.query('SELECT * FROM temp_db..test_measurement') logger.info(resp) assert len(resp['results'][0]['series'][0]['values']) == 5 await client.drop_database(db='temp_db')
async def test_write_to_non_default_rp(client): db = client.db await client.query( f"CREATE RETENTION POLICY myrp ON {db} DURATION 1h REPLICATION 1") points = [p for p in utils.random_points(5)] assert await client.write(points, rp='myrp') resp = await client.query(f"SELECT * from {db}.myrp.test_measurement") logger.info(resp) assert len(resp['results'][0]['series'][0]['values']) == 5
async def test_write_without_timestamp(client): points = [p for p in utils.random_points(9)] for p in points: _ = p.pop('time') _ = p.pop('measurement') logger.info(points) assert await client.write(points, measurement='yet_another_measurement') resp = await client.query('SELECT * FROM yet_another_measurement') # Points with the same tag/timestamp set are overwritten assert len(resp['results'][0]['series'][0]['values']) == 1
async def test_write_with_custom_measurement(client): points = [p for p in utils.random_points(5)] for p in points: _ = p.pop('measurement') logger.info(points) with pytest.raises(ValueError): assert await client.write(points) assert await client.write(points, measurement='another_measurement') resp = await client.query('SELECT * FROM another_measurement') assert len(resp['results'][0]['series'][0]['values']) == 5
def test_repr(client): logger.info(client)
def test_read_dataframe(df_client): df = df_client.query('SELECT * from m1') logger.info(f'\n{df.head()}') assert df.shape == (50, 8)
def test_select_into(df_client): df_client.query("SELECT * INTO m2_copy from m2") df = df_client.query('SELECT * from m2_copy') assert df.shape == (50, 8) logger.info(f'\n{df.head()}')
def test_read_dataframe_show_databases(df_client): df = df_client.show_databases() assert isinstance(df.index, pd.RangeIndex) assert 'name' in df.columns logger.info(f'\n{df.head()}')