def test_corner_type_scalar():
"""Test using scalar as corner arg."""
node_ids = rfuncs.corner_node_at_cell(rmg, 2, (4, ))
assert_array_equal(node_ids, np.array([16]))
node_ids = rfuncs.corner_node_at_cell(rmg, 2, (4, 5))
assert_array_equal(node_ids, np.array([16, 17]))
node_ids = rfuncs.corner_node_at_cell(rmg, 1)
assert_array_equal(node_ids, np.array([0, 1, 2, 5, 6, 7]))
def test_corner_type_scalar():
"""Test using scalar as corner arg."""
node_ids = rfuncs.corner_node_at_cell(rmg, 2, (4, ))
assert_array_equal(node_ids, np.array([16]))
node_ids = rfuncs.corner_node_at_cell(rmg, 2, (4, 5))
assert_array_equal(node_ids, np.array([16, 17]))
node_ids = rfuncs.corner_node_at_cell(rmg, 1)
assert_array_equal(node_ids, np.array([0, 1, 2, 5, 6, 7]))
def test_corner_type_tuple():
"""Test using tuple as corner arg."""
node_ids = rfuncs.corner_node_at_cell(rmg, (0, 1), (4, 5))
assert_array_equal(node_ids, np.array([[8, 6], [9, 7]]))
def test_corner_multiple_corners_and_cells():
"""Test getting nodes for more than one corner and more than one cell."""
node_ids = rfuncs.corner_node_at_cell(rmg, np.array([0, 1]), (4, 5))
assert_array_equal(node_ids, np.array([[8, 6], [9, 7]]))
def test_corner_with_no_cell_id():
"""Test without an arg for cells to mean all cells."""
node_ids = rfuncs.corner_node_at_cell(rmg, np.array([0]))
assert_array_equal(node_ids, np.array([[2, 3, 4, 7, 8, 9]]).T)
def test_corner_with_iterable_cell_id():
"""Test using iterable arg for cell."""
node_ids = rfuncs.corner_node_at_cell(rmg, np.array([0]), (5, ))
assert_array_equal(node_ids, np.array([[9]]))
def test_corner_with_scalar_cell_id():
"""Test using a scalar arg for cell."""
node_ids = rfuncs.corner_node_at_cell(rmg, np.array([0]), 0)
assert_array_equal(node_ids, np.array([[2]]))
def test_corner_type_list():
"""Test using list as corner arg."""
node_ids = rfuncs.corner_node_at_cell(rmg, [0, 1], (4, 5))
assert_array_equal(node_ids, np.array([[8, 6], [9, 7]]))
def test_corner_type_tuple():
"""Test using tuple as corner arg."""
node_ids = rfuncs.corner_node_at_cell(rmg, (0, 1), (4, 5))
assert_array_equal(node_ids, np.array([[8, 6], [9, 7]]))
def test_corner_multiple_corners_and_cells():
"""Test getting nodes for more than one corner and more than one cell."""
node_ids = rfuncs.corner_node_at_cell(rmg, np.array([0, 1]), (4, 5))
assert_array_equal(node_ids, np.array([[8, 6], [9, 7]]))
def test_corner_with_no_cell_id():
"""Test without an arg for cells to mean all cells."""
node_ids = rfuncs.corner_node_at_cell(rmg, np.array([0]))
assert_array_equal(node_ids, np.array([[2, 3, 4, 7, 8, 9]]).T)
def test_corner_with_iterable_cell_id():
"""Test using iterable arg for cell."""
node_ids = rfuncs.corner_node_at_cell(rmg, np.array([0]), (5, ))
assert_array_equal(node_ids, np.array([[9]]))
def test_corner_with_scalar_cell_id():
"""Test using a scalar arg for cell."""
node_ids = rfuncs.corner_node_at_cell(rmg, np.array([0]), 0)
assert_array_equal(node_ids, np.array([[2]]))
def test_corner_type_list():
"""Test using list as corner arg."""
node_ids = rfuncs.corner_node_at_cell(rmg, [0, 1], (4, 5))
assert_array_equal(node_ids, np.array([[8, 6], [9, 7]]))
