def test_get_neighbors_corner():
mask = numpy.array([[True, True, True],
[True, False, False],
[True, False, False]])
result = numpy.ma.array([[None, None, None],
[None, 78, 72],
[None, 74, 67]], mask = mask)
neighbors = pydrology.__get_neighbors__(TEST_ARRAY, 0, 0)
numpy.testing.assert_equal(result, neighbors)
def test_get_neighbors_left_edge():
mask = numpy.array([[True, False, False],
[True, False, False],
[True, False, False]])
result = numpy.ma.array([[None, 78, 72],
[None, 74, 67],
[None, 69, 53]], mask = mask)
neighbors = pydrology.__get_neighbors__(TEST_ARRAY, 1, 0)
numpy.testing.assert_equal(result, neighbors)
def test_get_neighbors_right_edge():
mask = numpy.array([[False, False, True],
[False, False, True],
[False, False, True]])
result = numpy.ma.array([[58, 49, None],
[46, 50, None],
[38, 48, None]], mask = mask)
neighbors = pydrology.__get_neighbors__(TEST_ARRAY, 1, 5)
numpy.testing.assert_equal(result, neighbors)
def test_get_neighbors_middle():
result = numpy.array([[78, 72, 69],
[74, 67, 56],
[69, 53, 44]])
neighbors = pydrology.__get_neighbors__(TEST_ARRAY, 1, 1)
numpy.testing.assert_equal(result, neighbors)
