overpy_series_test.py

import unittest

import geopandas as gpd
from shapely.geometry import LineString, Point

from context import measprocess as mpc


class TestOSMAPI(unittest.TestCase):
    def setUp(self):
        self._coords = [(48.201914, 16.363859), (48.194170, 16.385466)]

        self._measurement_coords_not_set = gpd.GeoSeries(
            (Point(lon, lat) for lat, lon in self._coords)
        )

        self._measurement_coords_set = gpd.GeoSeries(
            (Point(lon, lat) for lat, lon in self._coords)
        ).set_crs("EPSG:4326")

    def test_basic_API(self):
        street_series = mpc.geospatial.fetch.geoseries_streets(
            self._measurement_coords_set
        )

        blockage_series = mpc.geospatial.fetch.geoseries_buildings(
            self._measurement_coords_set
        )

        self.assertTrue("EPSG:4326" == street_series.crs)

        self.assertTrue("EPSG:4326" == blockage_series.crs)

    def test_raise_error_if_crs_not_set(self):

        with self.assertRaises(ValueError):
            mpc.geospatial.fetch.geoseries_buildings(
                self._measurement_coords_not_set
            )

            mpc.geospatial.fetch.geoseries_streets(self._measurement_coords_not_set)

    def test_output_coordinates_streets(self):

        street_series = mpc.geospatial.fetch.geoseries_streets(
            self._measurement_coords_set
        )

        street_series = street_series.to_crs("EPSG:31287")
        projected = self._measurement_coords_set.to_crs("EPSG:31287")

        self.assertTrue(street_series.distance(projected[0]).mean() < 1000)

    def test_output_coordinates_blockages(self):

        blockage_series = mpc.geospatial.fetch.geoseries_buildings(
            self._measurement_coords_set
        )

        blockage_series = blockage_series.to_crs("EPSG:31287")
        projected = self._measurement_coords_set.to_crs("EPSG:31287")

        self.assertTrue(blockage_series.distance(projected[0]).mean() < 1000)


if __name__ == "__main__":
    unittest.main()