def test_read_to_point_rdd(self):
input_location = os.path.join(tests_path, "resources/shapefiles/point")
spatial_rdd = ShapefileReader.readToPointRDD(self.sc, input_location)
geometry_rdd = ShapefileReader.readToGeometryRDD(
self.sc, input_location)
window = Envelope(-180.0, 180.0, -90.0, 90.0)
count = RangeQuery.SpatialRangeQuery(spatial_rdd, window, False,
False).count()
assert spatial_rdd.rawSpatialRDD.count() == count
assert 'org.datasyslab.geospark.spatialRDD.SpatialRDD' in geometry_rdd._srdd.toString(
)
assert 'org.datasyslab.geospark.spatialRDD.PointRDD' in spatial_rdd._srdd.toString(
)
def test_shape_file_end_with_undefined_type(self):
shape_rdd = ShapefileReader.readToGeometryRDD(
sc=self.sc, inputPath=undefined_type_shape_location)
if is_greater_or_equal_version(GeoSparkMeta.version, "1.2.0"):
assert shape_rdd.fieldNames == [
'LGA_CODE16', 'LGA_NAME16', 'STE_CODE16', 'STE_NAME16',
'AREASQKM16'
]
assert shape_rdd.getRawSpatialRDD().count() == 545
def test_read_geometry_rdd(self):
shape_rdd = ShapefileReader.readToGeometryRDD(self.sc,
polygon_shape_location)
assert shape_rdd.fieldNames == []
assert shape_rdd.rawSpatialRDD.collect().__len__() == 10000