Add unit test for extract_profiles

mdbartos · Feb 27, 2022 · 6a04a2f · 6a04a2f
1 parent 3553a14
commit 6a04a2f
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/tests/test_grid.py b/tests/test_grid.py
@@ -408,6 +408,13 @@ def test_extract_river_network():
     grid.extract_river_network(catch, acc > 20, algorithm='recursive')
     # TODO: Need more checks here. Check if endnodes equals next startnode
 
+def test_extract_profiles():
+    fdir = d.fdir
+    catch = d.catch
+    acc = d.acc
+    grid.clip_to(catch)
+    profiles, connections = grid.extract_profiles(catch, acc > 20)
+
 def test_view_methods():
     dem = d.dem
     catch = d.catch