def test_passing_incorrect_custom_filter(test_pbf):
from pyrosm import OSM
osm = OSM(filepath=test_pbf)
try:
osm.get_buildings(custom_filter="wrong")
except ValueError as e:
if "dictionary" in str(e):
pass
except Exception as e:
raise e
def test_reading_with_custom_filters_with_excluding(test_pbf):
from pyrosm import OSM
from shapely.geometry import Polygon
from geopandas import GeoDataFrame
# Get first all data
osm = OSM(filepath=test_pbf)
gdf_all = osm.get_buildings()
# Find out all 'building' tags
cnts = gdf_all['building'].value_counts()
n = len(gdf_all)
for filter_, cnt in cnts.items():
# Use the custom filter
filtered = osm.get_data_by_custom_criteria(
custom_filter={'building': [filter_]}, filter_type="exclude")
assert isinstance(filtered, GeoDataFrame)
assert isinstance(filtered.loc[0, "geometry"], Polygon)
assert len(filtered) == n - cnt
# Now should not have the filter_ in buildings
assert filter_ not in filtered["building"].unique()
required_cols = ['building', 'id', 'timestamp', 'version', 'geometry']
for col in required_cols:
assert col in filtered.columns
def test_parse_buildings_with_bbox(test_pbf):
from pyrosm import OSM
from geopandas import GeoDataFrame
from shapely.geometry import Polygon
bounds = [26.94, 60.525, 26.96, 60.535]
# Init with bounding box
osm = OSM(filepath=test_pbf, bounding_box=bounds)
gdf = osm.get_buildings()
assert isinstance(gdf.loc[0, 'geometry'], Polygon)
assert isinstance(gdf, GeoDataFrame)
# Test shape
assert gdf.shape == (569, 15)
required_cols = [
'building', 'addr:street', 'addr:postcode', 'addr:housenumber',
'opening_hours', 'id', 'timestamp', 'version', 'geometry', 'tags'
]
for col in required_cols:
assert col in gdf.columns
# The total bounds of the result should not be larger than the filter
# (allow some rounding error)
result_bounds = gdf.total_bounds
for coord1, coord2 in zip(bounds, result_bounds):
assert round(coord2, 3) >= round(coord1, 3)
def test_adding_extra_attribute(helsinki_pbf):
from pyrosm import OSM
from geopandas import GeoDataFrame
osm = OSM(filepath=helsinki_pbf)
gdf = osm.get_buildings()
extra_col = "wikidata"
extra = osm.get_buildings(extra_attributes=[extra_col])
# The extra should have one additional column compared to the original one
assert extra.shape[1] == gdf.shape[1] + 1
# Should have same number of rows
assert extra.shape[0] == gdf.shape[0]
assert extra_col in extra.columns
assert len(extra[extra_col].dropna().unique()) > 0
assert isinstance(gdf, GeoDataFrame)
def test_passing_custom_filter_without_element_key(test_pbf):
from pyrosm import OSM
from geopandas import GeoDataFrame
osm = OSM(filepath=test_pbf)
gdf = osm.get_buildings(custom_filter={"start_date": True})
assert isinstance(gdf, GeoDataFrame)
def test_reading_buildings_with_filters(test_pbf):
from pyrosm import OSM
from shapely.geometry import Polygon
from geopandas import GeoDataFrame
# Get first all data
osm = OSM(filepath=test_pbf)
gdf_all = osm.get_buildings()
# Find out all 'building' tags
cnts = gdf_all['building'].value_counts()
for filter_, cnt in cnts.items():
filtered = osm.get_buildings({'building': [filter_]})
assert isinstance(filtered, GeoDataFrame)
assert isinstance(filtered.loc[0, "geometry"], Polygon)
assert len(filtered) == cnt
# Now should only have buildings with given key
assert len(filtered["building"].unique()) == 1
required_cols = ['building', 'id', 'timestamp', 'version', 'geometry']
for col in required_cols:
assert col in filtered.columns
def test_reading_buildings_with_relations(helsinki_pbf):
from pyrosm import OSM
from shapely.geometry import Polygon
from geopandas import GeoDataFrame
osm = OSM(filepath=helsinki_pbf)
gdf = osm.get_buildings()
assert isinstance(gdf, GeoDataFrame)
assert isinstance(gdf.loc[0, "geometry"], Polygon)
assert gdf.shape == (486, 34)
required_cols = [
'building', 'id', 'timestamp', 'version', 'tags', 'geometry'
]
for col in required_cols:
assert col in gdf.columns
def test_reading_buildings_from_area_having_none(helsinki_pbf):
from pyrosm import OSM
from geopandas import GeoDataFrame
# Bounding box for area that does not have any data
bbox = [24.940514, 60.173849, 24.942, 60.175892]
osm = OSM(filepath=helsinki_pbf, bounding_box=bbox)
# The tool should warn if no buildings were found
with pytest.warns(UserWarning) as w:
gdf = osm.get_buildings()
# Check the warning text
if "could not find any buildings" in str(w):
pass
# Result should be None
assert gdf is None
def test_reading_buildings_with_defaults(test_pbf):
from pyrosm import OSM
from shapely.geometry import Polygon
from geopandas import GeoDataFrame
osm = OSM(filepath=test_pbf)
gdf = osm.get_buildings()
assert isinstance(gdf, GeoDataFrame)
assert isinstance(gdf.loc[0, "geometry"], Polygon)
assert gdf.shape == (2193, 19)
required_cols = [
'building', 'addr:city', 'addr:street', 'addr:country',
'addr:postcode', 'addr:housenumber', 'source', 'opening_hours',
'building:levels', 'id', 'timestamp', 'version', 'geometry'
]
for col in required_cols:
assert col in gdf.columns
def test_saving_buildings_to_geopackage(test_pbf, test_output_dir):
import os
from pyrosm import OSM
import geopandas as gpd
import shutil
if not os.path.exists(test_output_dir):
os.makedirs(test_output_dir)
temp_path = os.path.join(test_output_dir, "pyrosm_test.gpkg")
osm = OSM(filepath=test_pbf)
gdf = osm.get_buildings()
gdf.to_file(temp_path, driver="GPKG")
# Ensure it can be read and matches with original one
gdf2 = gpd.read_file(temp_path)
cols = gdf.columns
for col in cols:
assert gdf[col].tolist() == gdf2[col].tolist()
# Clean up
shutil.rmtree(test_output_dir)
def test_buildings(test_pbf):
from pyrosm import OSM
from geopandas import GeoDataFrame
osm = OSM(test_pbf)
gdf = osm.get_buildings()
assert isinstance(gdf, GeoDataFrame)