def test_geospatial_positive_query_for_circle(self):
"""
Perform a positive geospatial query for a circle
"""
records = []
query = TestGeospatial.client.query("test", "demo")
geo_object2 = aerospike.GeoJSON({
"type": "AeroCircle",
"coordinates": [[-122.0, 37.5], 250.2]
})
query.where(p.geo_within("loc", geo_object2.dumps()))
def callback((key, metadata, record)):
records.append(record)
query.foreach(callback)
assert len(records) == 1
assert records == [{
'loc': {
'coordinates': [-122.0, 37.5],
'type': 'Point'
}
}]
def test_geospatial_positive_query_with_geojson(self):
"""
Perform a positive geospatial query for a polygon with geojson
"""
records = []
query = TestGeospatial.client.query("test", "demo")
geo_object2 = aerospike.geojson(
'{"type": "Polygon", "coordinates": [[[-122.500000, 37.000000], [-121.000000, 37.000000], [-121.000000, 38.080000],[-122.500000, 38.080000], [-122.500000, 37.000000]]]}'
)
query.where(p.geo_within("loc", geo_object2.dumps()))
def callback((key, metadata, record)):
records.append(record)
query.foreach(callback)
assert len(records) == 3
assert records == [{
'loc': {
'coordinates': [-122.0, 37.5],
'type': 'Point'
}
}, {
'loc': {
'coordinates': [-121.8, 37.7],
'type': 'Point'
}
}, {
'loc': {
'coordinates': [-121.6, 37.9],
'type': 'Point'
}
}]
def test_geospatial_loads_positive_with_query(self):
"""
Perform a positive loads on geoJSON raw string followed by a query
"""
geo_object_loads = aerospike.GeoJSON({"type": "Polygon", "coordinates": [[[-124.500000,
37.000000],[-125.000000, 37.000000], [-121.000000, 38.080000],[-122.500000,
38.080000], [-124.500000, 37.000000]]]})
geo_object_loads.loads('{"type": "Polygon", "coordinates": [[[-122.500000, 37.000000], [-121.000000, 37.000000], [-121.000000, 38.080000],[-122.500000, 38.080000], [-122.500000, 37.000000]]]}')
assert geo_object_loads.unwrap() == {'coordinates': [[[-122.5, 37.0], [-121.0, 37.0], [-121.0, 38.08],
[-122.5, 38.08], [-122.5, 37.0]]], 'type': 'Polygon'}
records = []
query = TestGeospatial.client.query("test", "demo")
query.where(p.geo_within("loc", geo_object_loads.dumps()))
def callback((key, metadata, record)):
records.append(record)
query.foreach(callback)
assert len(records) == 3
assert records == [{'loc': {'coordinates': [-122.0, 37.5], 'type': 'Point'}}, {'loc': {'coordinates': [-121.8, 37.7], 'type':
'Point'}}, {'loc': {'coordinates': [-121.6, 37.9], 'type': 'Point'}}]
def test_geospatial_positive_query_outside_shape(self):
"""
Perform a positive geospatial query for polygon where all points are
outside polygon
"""
records = []
query = TestGeospatial.client.query("test", "demo")
geo_object2 = aerospike.GeoJSON({
"type":
"Polygon",
"coordinates":
[[[-126.500000, 37.000000], [-124.000000, 37.000000],
[-124.000000, 38.080000], [-126.500000, 38.080000],
[-126.500000, 37.000000]]]
})
query.where(p.geo_within("loc", geo_object2.dumps()))
def callback((key, metadata, record)):
records.append(record)
query.foreach(callback)
assert len(records) == 0
def test_geospatial_positive_query_without_set(self):
"""
Perform a positive geospatial query for a polygon without a set
"""
keys = []
for i in xrange(1, 10):
key = ('test', None, i)
lng = -122 + (0.2 * i)
lat = 37.5 + (0.2 * i)
geo_object = aerospike.GeoJSON({"type": "Point", "coordinates": [lng, lat] })
TestGeospatial.client.put(key, {"loc": geo_object})
keys.append(key)
TestGeospatial.client.index_geo2dsphere_create("test", None, "loc", "loc_index_no_set")
records = []
query = TestGeospatial.client.query("test", None)
geo_object2 = aerospike.GeoJSON({"type": "Polygon", "coordinates": [[[-122.500000,
37.000000],[-121.000000, 37.000000], [-121.000000, 38.080000],[-122.500000,
38.080000], [-122.500000, 37.000000]]]})
query.where(p.geo_within("loc", geo_object2.dumps()))
def callback((key, metadata, record)):
records.append(record)
query.foreach(callback)
TestGeospatial.client.index_remove('test', 'loc_index_no_set')
for key in keys:
TestGeospatial.client.remove(key)
assert len(records) == 2
assert records == [{'loc': {'coordinates': [-121.8, 37.7], 'type': 'Point'}}, {'loc': {'coordinates': [-121.6, 37.9], 'type': 'Point'}}]
def test_geospatial_positive_query(self):
"""
Perform a positive geospatial query for a polygon
"""
records = []
query = TestGeospatial.client.query("test", "demo")
geo_object2 = aerospike.GeoJSON(
{
"type": "Polygon",
"coordinates": [
[
[-122.500000, 37.000000],
[-121.000000, 37.000000],
[-121.000000, 38.080000],
[-122.500000, 38.080000],
[-122.500000, 37.000000],
]
],
}
)
query.where(p.geo_within("loc", geo_object2.dumps()))
def callback((key, metadata, record)):
records.append(record)
query.foreach(callback)
assert len(records) == 3
assert records == [
{"loc": {"coordinates": [-122.0, 37.5], "type": "Point"}},
{"loc": {"coordinates": [-121.8, 37.7], "type": "Point"}},
{"loc": {"coordinates": [-121.6, 37.9], "type": "Point"}},
]
def test_geospatial_positive_query_outside_shape(self):
"""
Perform a positive geospatial query for polygon where all points are
outside polygon
"""
records = []
query = TestGeospatial.client.query("test", "demo")
geo_object2 = aerospike.GeoJSON(
{
"type": "Polygon",
"coordinates": [
[
[-126.500000, 37.000000],
[-124.000000, 37.000000],
[-124.000000, 38.080000],
[-126.500000, 38.080000],
[-126.500000, 37.000000],
]
],
}
)
query.where(p.geo_within("loc", geo_object2.dumps()))
def callback((key, metadata, record)):
records.append(record)
query.foreach(callback)
assert len(records) == 0
def test_geospatial_wrap_positive_with_query(self):
"""
Perform a positive wrap on geospatial data followed by a query
"""
geo_object_wrap = aerospike.GeoJSON({
"type":
"Polygon",
"coordinates":
[[[-124.500000, 37.000000], [-125.000000, 37.000000],
[-121.000000, 38.080000], [-122.500000, 38.080000],
[-124.500000, 37.000000]]]
})
geo_object_wrap.wrap({
"type":
"Polygon",
"coordinates":
[[[-122.500000, 37.000000], [-121.000000, 37.000000],
[-121.000000, 38.080000], [-122.500000, 38.080000],
[-122.500000, 37.000000]]]
})
assert geo_object_wrap.unwrap() == {
'coordinates': [[[-122.5, 37.0], [-121.0, 37.0], [-121.0, 38.08],
[-122.5, 38.08], [-122.5, 37.0]]],
'type':
'Polygon'
}
records = []
query = TestGeospatial.client.query("test", "demo")
query.where(p.geo_within("loc", geo_object_wrap.dumps()))
def callback((key, metadata, record)):
records.append(record)
query.foreach(callback)
assert len(records) == 3
expected = [{
'loc': {
'coordinates': [-122.0, 37.5],
'type': 'Point'
}
}, {
'loc': {
'coordinates': [-121.8, 37.7],
'type': 'Point'
}
}, {
'loc': {
'coordinates': [-121.6, 37.9],
'type': 'Point'
}
}]
for r in records:
assert r in expected
def test_geospatial_positive_query_without_set(self):
"""
Perform a positive geospatial query for a polygon without a set
"""
keys = []
for i in xrange(1, 10):
key = ('test', None, i)
lng = -122 + (0.2 * i)
lat = 37.5 + (0.2 * i)
geo_object = aerospike.GeoJSON({
"type": "Point",
"coordinates": [lng, lat]
})
TestGeospatial.client.put(key, {"loc": geo_object})
keys.append(key)
TestGeospatial.client.index_geo2dsphere_create("test", None, "loc",
"loc_index_no_set")
records = []
query = TestGeospatial.client.query("test", None)
geo_object2 = aerospike.GeoJSON({
"type":
"Polygon",
"coordinates":
[[[-122.500000, 37.000000], [-121.000000, 37.000000],
[-121.000000, 38.080000], [-122.500000, 38.080000],
[-122.500000, 37.000000]]]
})
query.where(p.geo_within("loc", geo_object2.dumps()))
def callback((key, metadata, record)):
records.append(record)
query.foreach(callback)
TestGeospatial.client.index_remove('test', 'loc_index_no_set')
for key in keys:
TestGeospatial.client.remove(key)
assert len(records) == 2
expected = [{
'loc': {
'coordinates': [-121.8, 37.7],
'type': 'Point'
}
}, {
'loc': {
'coordinates': [-121.6, 37.9],
'type': 'Point'
}
}]
for r in records:
assert r in expected
def test_geospatial_wrap_positive_with_query(self):
"""
Perform a positive wrap on geospatial data followed by a query
"""
geo_object_wrap = aerospike.GeoJSON(
{
"type": "Polygon",
"coordinates": [
[
[-124.500000, 37.000000],
[-125.000000, 37.000000],
[-121.000000, 38.080000],
[-122.500000, 38.080000],
[-124.500000, 37.000000],
]
],
}
)
geo_object_wrap.wrap(
{
"type": "Polygon",
"coordinates": [
[
[-122.500000, 37.000000],
[-121.000000, 37.000000],
[-121.000000, 38.080000],
[-122.500000, 38.080000],
[-122.500000, 37.000000],
]
],
}
)
assert geo_object_wrap.unwrap() == {
"coordinates": [[[-122.5, 37.0], [-121.0, 37.0], [-121.0, 38.08], [-122.5, 38.08], [-122.5, 37.0]]],
"type": "Polygon",
}
records = []
query = TestGeospatial.client.query("test", "demo")
query.where(p.geo_within("loc", geo_object_wrap.dumps()))
def callback((key, metadata, record)):
records.append(record)
query.foreach(callback)
assert len(records) == 3
assert records == [
{"loc": {"coordinates": [-122.0, 37.5], "type": "Point"}},
{"loc": {"coordinates": [-121.8, 37.7], "type": "Point"}},
{"loc": {"coordinates": [-121.6, 37.9], "type": "Point"}},
]
def test_geospatial_positive_query_for_circle(self):
"""
Perform a positive geospatial query for a circle
"""
records = []
query = TestGeospatial.client.query("test", "demo")
geo_object2 = aerospike.GeoJSON({"type": "Circle", "coordinates": [[-122.0, 37.5], 250.2]})
query.where(p.geo_within("loc", geo_object2.dumps()))
def callback((key, metadata, record)):
records.append(record)
query.foreach(callback)
assert len(records) == 1
assert records == [{'loc': {'coordinates': [-122.0, 37.5], 'type': 'Point'}}]