def test_face_type_scalar(): """Test using scalar as face arg.""" node_ids = rfuncs.neighbor_node_at_cell(rmg, 2, (4, )) assert_array_equal(node_ids, np.array([17])) node_ids = rfuncs.neighbor_node_at_cell(rmg, 2, (4, 5)) assert_array_equal(node_ids, np.array([17, 18])) node_ids = rfuncs.neighbor_node_at_cell(rmg, 1) assert_array_equal(node_ids, np.array([5, 6, 7, 10, 11, 12]))
def test_face_type_scalar(): """Test using scalar as face arg.""" rmg = RasterModelGrid((4, 5)) values_at_nodes = np.arange(20.) node_ids = rfuncs.neighbor_node_at_cell(rmg, 2, (4, )) assert_array_equal(node_ids, np.array([17])) node_ids = rfuncs.neighbor_node_at_cell(rmg, 2, (4, 5)) assert_array_equal(node_ids, np.array([17, 18])) node_ids = rfuncs.neighbor_node_at_cell(rmg, 1) assert_array_equal(node_ids, np.array([5, 6, 7, 10, 11, 12]))
def test_face_with_iterable_cell_id(): """Test using iterable arg for cell.""" node_ids = rfuncs.neighbor_node_at_cell(rmg, np.array([0]), (5, )) assert_array_equal(node_ids, np.array([[8]]))
def test_face_with_scalar_cell_id(): """Test using a scalar arg for cell.""" node_ids = rfuncs.neighbor_node_at_cell(rmg, np.array([0]), 0) assert_array_equal(node_ids, np.array([[1]]))
def test_face_type_tuple(): """Test using tuple as face arg.""" node_ids = rfuncs.neighbor_node_at_cell(rmg, (0, 1), (4, 5)) assert_array_equal(node_ids, np.array([[7, 11], [8, 12]]))
def test_face_multiple_corners_and_cells(): """Test getting nodes for more than one corner and more than one cell.""" node_ids = rfuncs.neighbor_node_at_cell(rmg, np.array([0, 1]), (4, 5)) assert_array_equal(node_ids, np.array([[7, 11], [8, 12]]))
def test_face_multiple_faces(): """Test getting nodes for more than one corner.""" rmg = RasterModelGrid((4, 5)) values_at_nodes = np.arange(20.) node_ids = rfuncs.neighbor_node_at_cell(rmg, np.array([0, 1]), (4, )) assert_array_equal(node_ids, np.array([[7, 11]]))
def test_face_type_list(): """Test using list as face arg.""" rmg = RasterModelGrid((4, 5)) node_ids = rfuncs.neighbor_node_at_cell(rmg, [0, 1], (4, 5)) assert_array_equal(node_ids, np.array([[7, 11], [8, 12]]))
def test_face_with_scalar_cell_id(): """Test using a scalar arg for cell.""" rmg = RasterModelGrid((4, 5)) values_at_nodes = np.arange(20.) node_ids = rfuncs.neighbor_node_at_cell(rmg, np.array([0]), 0) assert_array_equal(node_ids, np.array([[1]]))
def test_face_type_tuple(): """Test using tuple as face arg.""" rmg = RasterModelGrid((4, 5)) values_at_nodes = np.arange(20.) node_ids = rfuncs.neighbor_node_at_cell(rmg, (0, 1), (4, 5)) assert_array_equal(node_ids, np.array([[7, 11], [8, 12]]))
def test_face_with_no_cell_id(): """Test without an arg for cells to mean all cells.""" node_ids = rfuncs.neighbor_node_at_cell(rmg, np.array([0])) assert_array_equal(node_ids, np.array([[1, 2, 3, 6, 7, 8]]).T)
def test_face_multiple_faces(): """Test getting nodes for more than one corner.""" node_ids = rfuncs.neighbor_node_at_cell(rmg, np.array([0, 1]), (4, )) assert_array_equal(node_ids, np.array([[7, 11]]))
def test_face_with_no_cell_id(): """Test without an arg for cells to mean all cells.""" rmg = RasterModelGrid((4, 5)) values_at_nodes = np.arange(20.) node_ids = rfuncs.neighbor_node_at_cell(rmg, np.array([0])) assert_array_equal(node_ids, np.array([[1, 2, 3, 6, 7, 8]]).T)