#110 add nearest_cell test

mdbartos · Apr 10, 2020 · 35baf17 · 35baf17
1 parent 3dba26c
commit 35baf17
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/tests/test_grid.py b/tests/test_grid.py
@@ -51,6 +51,16 @@ def test_constructors():
 def test_dtype():
     assert(grid.dir.dtype == np.uint8)
 
+def test_nearest_cell():
+    '''
+    corner: snaps to nearest top/left
+    center: snaps to index of cell that contains the geometry
+    '''
+    col, row = grid.nearest_cell(x, y, snap='corner')
+    assert (col, row) == (229, 101)
+    col, row = grid.nearest_cell(x, y, snap='center')
+    assert (col, row) == (228, 100)
+
 def test_catchment():
     # Reference routing
     grid.catchment(x, y, data='dir', dirmap=dirmap, out_name='catch',