plotting_test.py

import unittest

import geopandas as gpd
import pandas as pd
from shapely.geometry import Point

from context import measprocess as mpc


class TestPlottingOSM(unittest.TestCase):
    def setUp(self):
        complete_dataset = pd.read_csv("tests/example_files/gps_test/gps.csv", index_col=0)

        gps_series = complete_dataset[['Lon.', 'Lat.']].values

        self._gps_series = gpd.GeoSeries(
            (Point(lon, lat) for lon, lat in gps_series)
        )

    def test_basic(self):
        #Check if no exception comes up
        series = self._gps_series.set_crs("EPSG:4326")
        mpc.plotting.plot_series_osm(series, show=False)

    def test_exception_epsg(self):
        with self.assertRaises(ValueError):
            series = self._gps_series#.set_crs("EPSG:4326")
            mpc.plotting.plot_series_osm(series, show=False)

    def test_kwargs_basic(self):
        series = self._gps_series.set_crs("EPSG:4326")
        mpc.plotting.plot_series_osm(series, show=False, color="red")

    def test_kwargs_double(self):
        #s is based twice here
        with self.assertRaises(TypeError):
            series = self._gps_series.set_crs("EPSG:4326")
            mpc.plotting.plot_series_osm(series, show=False, s=100, color="red")

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