def test_vectors_aggregate_query_complex(self):
wkt = 'POLYGON((-76.65 40.10, -76.65 40.14, -76.55 40.14, -76.55 40.10, -76.65 40.10))'
child_agg = AggregationDef(agg_type='date_hist', value='M')
aggs = AggregationDef(agg_type='geohash',
value='4',
children=child_agg)
v = Vectors()
query = 'item_type:tweet'
start_date = 'now-6M'
end_date = 'now'
result = v.aggregate_query(wkt,
aggs,
index=None,
query=query,
start_date=start_date,
end_date=end_date)
assert len(result) == 1
assert 'name' in result[0]
assert result[0]['name'] == 'geohash:4'
assert 'terms' in result[0]
terms = result[0]['terms']
assert len(terms) == 1
assert terms[0]['term'] == 'dr1s'
assert len(terms[0]['aggregations']) == 1
assert len(terms[0]['aggregations'][0]['terms']) == 4
def test_vectors_aggregate_query_agg_def(self):
wkt = 'POLYGON((-76.65 40.10, -76.65 40.14, -76.55 40.14, -76.55 40.10, -76.65 40.10))'
aggs = AggregationDef(agg_type='terms', value='ingest_source')
v = Vectors()
result = v.aggregate_query(wkt, aggs, index=None)
assert len(result) == 1
assert 'name' in result[0]
assert result[0]['name'] == 'terms:ingest_source'
assert 'terms' in result[0]
assert len(result[0]['terms']) == 6
def test_vectors_aggregate_query_with_defined_index(self):
wkt = 'POLYGON((-76.65 40.10, -76.65 40.14, -76.55 40.14, -76.55 40.10, -76.65 40.10))'
aggs = 'terms:item_type'
v = Vectors()
result = v.aggregate_query(wkt, aggs, index='read-vector-osm-*')
assert len(result) == 1
assert 'name' in result[0]
assert result[0]['name'] == 'terms:item_type'
assert 'terms' in result[0]
assert len(result[0]['terms']) == 10
def test_vectors_aggregate_query_complex(self):
wkt = 'POLYGON((-76.65 40.10, -76.65 40.14, -76.55 40.14, -76.55 40.10, -76.65 40.10))'
child_agg = AggregationDef(agg_type='date_hist', value='M')
aggs = AggregationDef(agg_type='geohash', value='4', children=child_agg)
v = Vectors()
query = 'item_type:tweet'
start_date = 'now-6M'
end_date = 'now'
result = v.aggregate_query(wkt, aggs, index=None, query=query, start_date=start_date, end_date=end_date)
assert len(result) == 1
assert 'name' in result[0]
assert result[0]['name'] == 'geohash:4'
assert 'terms' in result[0]
terms = result[0]['terms']
assert len(terms) == 1
assert terms[0]['term'] == 'dr1s'
assert len(terms[0]['aggregations']) == 1
assert len(terms[0]['aggregations'][0]['terms']) == 4
