def test_get_cluster_centroids_as_geo_json(self): test_layer_1 = DBSession.query(Layer).filter_by(name='TestLayer1').one() test_layer_2 = DBSession.query(Layer).filter_by(name='TestLayer2').one() q = GriddedAndBoundMappablePoint.get_points_as_geojson(test_layer_1, grid_size=1) result = q.all() self.assertEqual(result[0].centroid, '{"type":"Point","coordinates":[20,10]}') self.assertEqual(result[1].centroid, '{"type":"Point","coordinates":[30,10]}') self.assertEqual(result[0].cluster_size, 1) self.assertEqual(result[1].cluster_size, 1) q2 = GriddedAndBoundMappablePoint.get_points_as_geojson(test_layer_1, grid_size=100) result2 = q2.one() self.assertEqual(result2.centroid, '{"type":"Point","coordinates":[25,10]}') q3 = GriddedAndBoundMappablePoint.get_points_as_geojson(test_layer_2, grid_size=100) result3 = q3.one() self.assertEqual(result3.centroid, '{"type":"Point","coordinates":[16.6666666666667,15]}')
def test_get_layer_points_as_geo_json(self): test_layer_1 = DBSession.query(Layer).filter_by(name='TestLayer1').one() test_layer_2 = DBSession.query(Layer).filter_by(name='TestLayer2').one() q = GriddedAndBoundMappablePoint.get_points_as_geojson(test_layer_1, grid_size=1) result = q.all() # self.assertEqual(result[0].locations, '{"type":"MultiPoint","coordinates":[[20,10]]}') # self.assertEqual(result[1].locations, '{"type":"MultiPoint","coordinates":[[30,10]]}') q2 = GriddedAndBoundMappablePoint.get_points_as_geojson(test_layer_1, grid_size=100) result2 = q2.all() # self.assertEqual(result2[0].locations, '{"type":"MultiPoint","coordinates":[[30,10],[20,10]]}') self.assertEqual(result2[0].cluster_size, 2) q3 = GriddedAndBoundMappablePoint.get_points_as_geojson(test_layer_2, grid_size=1) result3 = q3.all() # self.assertEqual(result3[0].locations, '{"type":"MultiPoint","coordinates":[[10,15],[10,15]]}') # self.assertEqual(result3[1].locations, '{"type":"MultiPoint","coordinates":[[30,15]]}') self.assertEqual(result3[0].cluster_size, 2) self.assertEqual(result3[1].cluster_size, 1)
def test_get_cluster_grid_size(self): bounds_1 = "-180, -90, 180, 90" grid_size_1 = GriddedAndBoundMappablePoint.get_cluster_grid_size(bounds_1) self.assertEqual(grid_size_1, (270.0 / GriddedAndBoundMappablePoint.GRID_SIZE_WINDOW_FRACTION)) bounds_2 = "0, 0, 180, 90" grid_size_2 = GriddedAndBoundMappablePoint.get_cluster_grid_size(bounds_2) self.assertEqual(grid_size_2, (135.0 / GriddedAndBoundMappablePoint.GRID_SIZE_WINDOW_FRACTION)) bounds_3 = "0, 0, 10, 10" grid_size_3 = GriddedAndBoundMappablePoint.get_cluster_grid_size(bounds_3) self.assertEqual(grid_size_3, (10.0 / GriddedAndBoundMappablePoint.GRID_SIZE_WINDOW_FRACTION)) bounds_4 = "0, 0, 0.0001, 0.0001" grid_size_4 = GriddedAndBoundMappablePoint.get_cluster_grid_size(bounds_4) self.assertTrue( ( grid_size_4 / GriddedAndBoundMappablePoint.GRID_SIZE_WINDOW_FRACTION ) < GriddedAndBoundMappablePoint.MIN_GRID_SIZE_BEFORE_NO_CLUSTERING) self.assertEqual(grid_size_4, 0) bounds_5 = "0, b, 0.0001, 0.0001" grid_size_5 = GriddedAndBoundMappablePoint.get_cluster_grid_size(bounds_5) self.assertEqual(None, grid_size_5)
def generate_cache_clusters(class_, layer, grid_size): log = logging.getLogger(__name__) log.debug("Generating cache for grid size: %s", grid_size) cache_record = class_.CacheRecord(grid_size) layer.cache_records.append(cache_record) DBSession.flush() clusters = GriddedAndBoundMappablePoint.get_points_as_wkt(layer, grid_size=grid_size)\ .filter( MappablePoint.layer_id == layer.id ) i = 0 for cluster in clusters: i += 1 centroid = cluster.centroid cluster_size = cluster.cluster_size # locations = cluster.locations cached_mappable_cluster = class_.CachedMappablePointCluster(cluster_size, centroid) #, locations) cache_record.cached_mappable_point_clusters.append(cached_mappable_cluster) if (i % 10000 == 0): log.debug("Up to cluster: %i", i) DBSession.flush()
def test_get_layer_points_as_wkt(self): test_layer_1 = DBSession.query(Layer).filter_by(name='TestLayer1').one() q = GriddedAndBoundMappablePoint.get_points_as_wkt(test_layer_1, grid_size=1) result = q.all()
def test_bounds_not_intersecting_points(self): test_layer_1 = DBSession.query(Layer).filter_by(name='TestLayer1').one() q = GriddedAndBoundMappablePoint.get_points_as_geojson(test_layer_1, grid_size=1, bbox=[-180,-89,-170,-80]) result = q.all() self.assertEqual(len(result),0)