Exemplo n.º 1
0
def test_beyond_grid():
    rmg = RasterModelGrid(4, 5, dx=2.)

    assert_equal(rfuncs.find_nearest_node(rmg, (-.999, .2)), 0)
    assert_raises(ValueError, rfuncs.find_nearest_node, rmg, (-1.001, .2))

    assert_equal(rfuncs.find_nearest_node(rmg, (8.999, .2)), 4)
    assert_raises(ValueError, rfuncs.find_nearest_node, rmg, (9.001, .2))

    assert_equal(rfuncs.find_nearest_node(rmg, (.2, -.999)), 0)
    assert_raises(ValueError, rfuncs.find_nearest_node, rmg, (.2, -1.001))

    assert_equal(rfuncs.find_nearest_node(rmg, (.2, 6.999)), 15)
    assert_raises(ValueError, rfuncs.find_nearest_node, rmg, (.2, 7.001))
Exemplo n.º 2
0
def test_beyond_grid():
    """Raise an error if points are outside the bounds of the grid."""
    rmg = RasterModelGrid((4, 5), spacing=(2., 2.))

    assert_equal(rfuncs.find_nearest_node(rmg, (-.999, .2)), 0)
    assert_raises(ValueError, rfuncs.find_nearest_node, rmg, (-1.001, .2))

    assert_equal(rfuncs.find_nearest_node(rmg, (8.999, .2)), 4)
    assert_raises(ValueError, rfuncs.find_nearest_node, rmg, (9.001, .2))

    assert_equal(rfuncs.find_nearest_node(rmg, (.2, -.999)), 0)
    assert_raises(ValueError, rfuncs.find_nearest_node, rmg, (.2, -1.001))

    assert_equal(rfuncs.find_nearest_node(rmg, (.2, 6.999)), 15)
    assert_raises(ValueError, rfuncs.find_nearest_node, rmg, (.2, 7.001))
Exemplo n.º 3
0
def test_with_scalars():
    """Test scalar args."""
    rmg = RasterModelGrid(4, 5)
    id = rfuncs.find_nearest_node(rmg, (0.2, 0.6))
    assert id == 5
    assert id.ndim == 0
    assert_array_is_int(id)
Exemplo n.º 4
0
def test_with_ndarray():
    """Test with 1d numpy arrays as args."""
    rmg = RasterModelGrid((4, 5))
    coords = (np.array([0.1, 0.2]), np.array([3.4, 2.6]))
    id = rfuncs.find_nearest_node(rmg, coords)
    assert_array_equal(id, np.array([15, 15], dtype=int))
    assert isinstance(id, np.ndarray)
Exemplo n.º 5
0
def test_with_scalars():
    """Test scalar args."""
    rmg = RasterModelGrid((4, 5))
    id = rfuncs.find_nearest_node(rmg, (0.2, 0.6))
    assert id == 5
    assert id.ndim == 0
    assert id.dtype == np.int32 or id.dtype == np.int64
def test_with_ndarray():
    """Test with 1d numpy arrays as args."""
    rmg = RasterModelGrid(4, 5)
    coords = (np.array([0.1, .2]), np.array([3.4, 2.6]))
    id = rfuncs.find_nearest_node(rmg, coords)
    assert_array_equal(id, np.array([15, 15], dtype=int))
    assert isinstance(id, np.ndarray)
def test_with_ndarray_with_length_0():
    """Test with 0d numpy arrays as args."""
    rmg = RasterModelGrid(4, 5)
    id = rfuncs.find_nearest_node(rmg, (np.array(0.2), np.array(0.6)))
    assert_array_equal(id, np.array(5, dtype=int))
    assert id.ndim == 0
    assert_array_is_int(id)
Exemplo n.º 8
0
def test_with_ndarray_with_length_0():
    """Test with 0d numpy arrays as args."""
    rmg = RasterModelGrid((4, 5))
    id = rfuncs.find_nearest_node(rmg, (np.array(0.2), np.array(0.6)))
    assert_array_equal(id, np.array(5, dtype=int))
    assert id.ndim == 0
    assert id.dtype == np.int32 or id.dtype == np.int64
Exemplo n.º 9
0
def test_beyond_grid():
    rmg = RasterModelGrid(4, 5, dx=2.)

    assert_equal(rfuncs.find_nearest_node(rmg, (-.999, .2)), 0)
    assert_raises(ValueError,
                  rfuncs.find_nearest_node, rmg, (-1.001, .2))

    assert_equal(rfuncs.find_nearest_node(rmg, (8.999, .2)), 4)
    assert_raises(ValueError,
                  rfuncs.find_nearest_node, rmg, (9.001, .2))

    assert_equal(rfuncs.find_nearest_node(rmg, (.2, -.999)), 0)
    assert_raises(ValueError,
                  rfuncs.find_nearest_node, rmg, (.2, -1.001))

    assert_equal(rfuncs.find_nearest_node(rmg, (.2, 6.999)), 15)
    assert_raises(ValueError,
                  rfuncs.find_nearest_node, rmg, (.2, 7.001))
Exemplo n.º 10
0
def test_beyond_grid():
    """Raise an error if points are outside the bounds of the grid."""
    rmg = RasterModelGrid((4, 5), spacing=(2., 2.))

    assert_equal(rfuncs.find_nearest_node(rmg, (-.999, .2)), 0)
    assert_raises(ValueError,
                  rfuncs.find_nearest_node, rmg, (-1.001, .2))

    assert_equal(rfuncs.find_nearest_node(rmg, (8.999, .2)), 4)
    assert_raises(ValueError,
                  rfuncs.find_nearest_node, rmg, (9.001, .2))

    assert_equal(rfuncs.find_nearest_node(rmg, (.2, -.999)), 0)
    assert_raises(ValueError,
                  rfuncs.find_nearest_node, rmg, (.2, -1.001))

    assert_equal(rfuncs.find_nearest_node(rmg, (.2, 6.999)), 15)
    assert_raises(ValueError,
                  rfuncs.find_nearest_node, rmg, (.2, 7.001))
Exemplo n.º 11
0
def test_with_scalars():
    rmg = RasterModelGrid(4, 5)
    id = rfuncs.find_nearest_node(rmg, (0.2, 0.6))
    assert_equal(id, 5)
    assert_equal(id.ndim, 0)
    assert_is_instance(id, np.int)
Exemplo n.º 12
0
def test_beyond_grid():
    """Raise an error if points are outside the bounds of the grid."""
    rmg = RasterModelGrid((4, 5), xy_spacing=(2.0, 2.0))

    assert rfuncs.find_nearest_node(rmg, (-0.999, 0.2)) == 0
    with pytest.raises(ValueError):
        rfuncs.find_nearest_node(rmg, (-1.001, 0.2))

    assert rfuncs.find_nearest_node(rmg, (8.999, 0.2)) == 4
    with pytest.raises(ValueError):
        rfuncs.find_nearest_node(rmg, (9.001, 0.2))

    assert rfuncs.find_nearest_node(rmg, (0.2, -0.999)) == 0
    with pytest.raises(ValueError):
        rfuncs.find_nearest_node(rmg, (0.2, -1.001))

    assert rfuncs.find_nearest_node(rmg, (0.2, 6.999)) == 15
    with pytest.raises(ValueError):
        rfuncs.find_nearest_node(rmg, (0.2, 7.001))
Exemplo n.º 13
0
def test_non_unit_spacing():
    """Test with a grid of non-unit spacing."""
    rmg = RasterModelGrid((4, 5), xy_spacing=(2.0, 2.0))
    id = rfuncs.find_nearest_node(rmg, (0.9, 0.2))
    assert id == 0
Exemplo n.º 14
0
def test_with_ndarray():
    rmg = RasterModelGrid(4, 5)
    coords = (np.array([0.1, .2]), np.array([3.4, 2.6]))
    id = rfuncs.find_nearest_node(rmg, coords)
    assert_array_equal(id, np.array([15, 15], dtype=int))
    assert_is_instance(id, np.ndarray)
Exemplo n.º 15
0
def test_non_unit_spacing():
    """Test with a grid of non-unit spacing."""
    rmg = RasterModelGrid(4, 5, dx=2.)
    id = rfuncs.find_nearest_node(rmg, (.9, .2))
    assert_equal(id, 0)
Exemplo n.º 16
0
def test_beyond_grid():
    """Raise an error if points are outside the bounds of the grid."""
    rmg = RasterModelGrid((4, 5), spacing=(2., 2.))

    assert rfuncs.find_nearest_node(rmg, (-.999, .2)) == 0
    with pytest.raises(ValueError):
        rfuncs.find_nearest_node(rmg, (-1.001, .2))

    assert rfuncs.find_nearest_node(rmg, (8.999, .2)) == 4
    with pytest.raises(ValueError):
        rfuncs.find_nearest_node(rmg, (9.001, .2))

    assert rfuncs.find_nearest_node(rmg, (.2, -.999)) == 0
    with pytest.raises(ValueError):
        rfuncs.find_nearest_node(rmg, (.2, -1.001))

    assert rfuncs.find_nearest_node(rmg, (.2, 6.999)) == 15
    with pytest.raises(ValueError):
        rfuncs.find_nearest_node(rmg, (.2, 7.001))
Exemplo n.º 17
0
def test_non_unit_spacing():
    """Test with a grid of non-unit spacing."""
    rmg = RasterModelGrid((4, 5), spacing=(2., 2.))
    id = rfuncs.find_nearest_node(rmg, (.9, .2))
    assert id == 0
Exemplo n.º 18
0
def test_with_iterable():
    """Test iterable args."""
    rmg = RasterModelGrid(4, 5)
    id = rfuncs.find_nearest_node(rmg, ([0.2], [0.6]))
    assert_array_equal(id, np.array([5], dtype=int))
    assert_array_is_int(id)
Exemplo n.º 19
0
def test_with_iterable():
    rmg = RasterModelGrid(4, 5)
    id = rfuncs.find_nearest_node(rmg, ([0.2], [0.6]))
    assert_array_equal(id, np.array([5], dtype=int))
    assert_is_instance(id, np.ndarray)
Exemplo n.º 20
0
def test_with_ndarray_with_length_0():
    rmg = RasterModelGrid(4, 5)
    id = rfuncs.find_nearest_node(rmg, (np.array(0.2), np.array(0.6)))
    assert_array_equal(id, np.array(5, dtype=int))
    assert_equal(id.ndim, 0)
    assert_is_instance(id, np.int)
Exemplo n.º 21
0
def test_with_iterable():
    """Test iterable args."""
    rmg = RasterModelGrid((4, 5))
    id = rfuncs.find_nearest_node(rmg, ([0.2], [0.6]))
    assert_array_equal(id, np.array([5], dtype=int))
    assert id.dtype == np.int32 or id.dtype == np.int64
Exemplo n.º 22
0
def test_non_unit_spacing():
    rmg = RasterModelGrid(4, 5, dx=2.)
    id = rfuncs.find_nearest_node(rmg, (.9, .2))
    assert_equal(id, 0)
Exemplo n.º 23
0
def test_non_unit_spacing():
    """Test with a grid of non-unit spacing."""
    rmg = RasterModelGrid((4, 5), spacing=(2., 2.))
    id = rfuncs.find_nearest_node(rmg, (.9, .2))
    assert_equal(id, 0)