snakecase segment_length

Deltares · Dec 12, 2024 · ee01f89 · ee01f89
1 parent cc7edcc
commit ee01f89
Show file tree

Hide file tree

Showing 4 changed files with 9 additions and 9 deletions.
diff --git a/examples/add_roadnetwork.ipynb b/examples/add_roadnetwork.ipynb
diff --git a/hydromt_fiat/workflows/exposure_vector.py b/hydromt_fiat/workflows/exposure_vector.py
@@ -298,11 +298,11 @@ def setup_roads(
             roads[
                 [
                     "max_damage_structure",
-                    "Segment Length",
+                    "segment_length",
                 ]
             ] = get_max_potential_damage_roads(roads, road_damage)
         elif isinstance(road_damage, (int, float)) or road_damage is None:
-            roads["Segment Length"] = road_length
+            roads["segment_length"] = road_length
             roads["max_damage_structure"] = road_damage
 
         # Convert crs to exposure buildings crs

diff --git a/hydromt_fiat/workflows/roads.py b/hydromt_fiat/workflows/roads.py
@@ -17,7 +17,7 @@ def get_max_potential_damage_roads(
     roads = gpd.GeoDataFrame(
         {
             "lanes": pd.to_numeric(roads["lanes"], errors="coerce"),
-            "Segment Length": roads.length,
+            "segment_length": roads.length,
             "geometry": roads["geometry"],
         }
     )
@@ -31,15 +31,15 @@ def get_max_potential_damage_roads(
     roads["damage_value"] = roads["lanes"].map(damage_dic)
 
     # Potentially convert the length to meters
-    roads["max_damage_structure"] = roads["damage_value"] * roads["Segment Length"]
+    roads["max_damage_structure"] = roads["damage_value"] * roads["segment_length"]
     if unit == "meters" or unit == "metre" or unit == "m":
         roads["max_damage_structure"] = roads["max_damage_structure"] * 0.3048
     else:
         print(
             "You are using the wrong unit for the segment length. Please use <'foot/feet/ft'> or <'meters/metre/m'>"
         )
 
-    return roads[["max_damage_structure", "Segment Length"]]
+    return roads[["max_damage_structure", "segment_length"]]
 
 
 def get_road_lengths(roads):

diff --git a/tests/test_roads.py b/tests/test_roads.py
@@ -89,7 +89,7 @@ def test_setup_roads(case):
         "extract_method",
         "ground_flht",
         "max_damage_structure",
-        "Segment Length",
+        "segment_length",
     ]
     assert set(required_columns) == set(exposure.columns)