add tests

tests/test_core.py

@@ -6,6 +6,7 @@
 import os
 import geopandas as gpd
 from shapely.geometry import Point, Polygon
+from shapely import LineString
 import shutil
 
 
@@ -50,6 +51,12 @@ def test_selector():
     assert G == 1
 
 
+def test_selector_non_int():
+    """Test selector() if not an int."""
+    x = core.selector("y")
+    assert x == "y"
+
+
 def test_flatten():
     """Test the flatten() function."""
     # call the function
@@ -331,6 +338,19 @@ def test_custom_simplify():
     pass
 
 
+def test_cut():
+    """Test the cut function."""
+    # define parameters
+    line = LineString([[0, 0], [1, 0], [1, 1]])
+    distance = 0.5
+    # call function
+    two_lines = core.cut(line, distance)
+    assert line.length == 2.0
+    assert two_lines[0].length == 0.5
+    assert two_lines[1].length == 1.5
+    assert line.length == (two_lines[0].length + two_lines[1].length)
+
+
 def test_salt_long_lines():
     """Test the salt_long_lines function."""
     pass

tests/test_core_integrative.py

@@ -0,0 +1 @@
+"""Integrative tests for core.py"""

tests/test_load_osm_integrative.py

@@ -14,3 +14,13 @@ def test_OSM_to_network():
     assert isinstance(otn, load_osm.OSM_to_network)
     assert isinstance(otn.osmFile, str)
     assert isinstance(otn.roads_raw, gpd.GeoDataFrame)
+
+
+def test_OSM_to_network_ferries():
+    """Test the OSM_to_network class with ferries."""
+    fpath = os.path.join("Tutorials", "tutorial_data", "iceland-latest.osm.pbf")
+    # load the class
+    otn = load_osm.OSM_to_network(fpath, includeFerries=True)
+    assert isinstance(otn, load_osm.OSM_to_network)
+    assert isinstance(otn.osmFile, str)
+    assert isinstance(otn.roads_raw, gpd.GeoDataFrame)