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))
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))
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)
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)
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)
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
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)
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))
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
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)
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)
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))
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
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)
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)
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)
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
def test_non_unit_spacing(): rmg = RasterModelGrid(4, 5, dx=2.) id = rfuncs.find_nearest_node(rmg, (.9, .2)) assert_equal(id, 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)