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)