def test_corner_type_scalar():
    node_ids = rfuncs.node_id_of_cell_corner(rmg, 2, (4, ))
    assert_array_equal(node_ids, np.array([16]))

    node_ids = rfuncs.node_id_of_cell_corner(rmg, 2, (4, 5))
    assert_array_equal(node_ids, np.array([16, 17]))

    node_ids = rfuncs.node_id_of_cell_corner(rmg, 1)
    assert_array_equal(node_ids, np.array([0, 1, 2, 5, 6, 7]))
def test_corner_type_scalar():
    node_ids = rfuncs.node_id_of_cell_corner(rmg, 2, (4, ))
    assert_array_equal(node_ids, np.array([16]))

    node_ids = rfuncs.node_id_of_cell_corner(rmg, 2, (4, 5))
    assert_array_equal(node_ids, np.array([16, 17]))

    node_ids = rfuncs.node_id_of_cell_corner(rmg, 1)
    assert_array_equal(node_ids, np.array([0, 1, 2, 5, 6, 7]))
def test_corner_with_no_cell_id():
    node_ids = rfuncs.node_id_of_cell_corner(rmg, np.array([0]))
    assert_array_equal(node_ids, np.array([2, 3, 4, 7, 8, 9]))
def test_corner_with_iterable_cell_id():
    node_ids = rfuncs.node_id_of_cell_corner(rmg, np.array([0]), (5, ))
    assert_array_equal(node_ids, np.array([9]))
def test_corner_with_scalar_cell_id():
    node_ids = rfuncs.node_id_of_cell_corner(rmg, np.array([0]), 0)
    assert_array_equal(node_ids, np.array([2]))
Example #6
0
def test_corner_with_iterable_cell_id():
    """Test using iterable arg for cell."""
    node_ids = rfuncs.node_id_of_cell_corner(rmg, np.array([0]), (5, ))
    assert_array_equal(node_ids, np.array([[9]]))
def test_corner_multiple_corners_and_cells():
    """Test getting nodes for more than one corner and more than one cell."""
    node_ids = rfuncs.node_id_of_cell_corner(rmg, np.array([0, 1]), (4, 5))
    assert_array_equal(node_ids, np.array([[8, 6], [9, 7]]))
def test_corner_with_iterable_cell_id():
    """Test using iterable arg for cell."""
    node_ids = rfuncs.node_id_of_cell_corner(rmg, np.array([0]), (5, ))
    assert_array_equal(node_ids, np.array([[9]]))
def test_corner_type_list():
    """Test using list as corner arg."""
    node_ids = rfuncs.node_id_of_cell_corner(rmg, [0, 1], (4, 5))
    assert_array_equal(node_ids, np.array([[8, 6], [9, 7]]))
def test_corner_with_no_cell_id():
    node_ids = rfuncs.node_id_of_cell_corner(rmg, np.array([0]))
    assert_array_equal(node_ids, np.array([2, 3, 4, 7, 8, 9]))
def test_corner_with_iterable_cell_id():
    node_ids = rfuncs.node_id_of_cell_corner(rmg, np.array([0]), (5, ))
    assert_array_equal(node_ids, np.array([9]))
def test_corner_with_scalar_cell_id():
    node_ids = rfuncs.node_id_of_cell_corner(rmg, np.array([0]), 0)
    assert_array_equal(node_ids, np.array([2]))
Example #13
0
def test_corner_type_tuple():
    """Test using tuple as corner arg."""
    node_ids = rfuncs.node_id_of_cell_corner(rmg, (0, 1), (4, 5))
    assert_array_equal(node_ids, np.array([[8, 6], [9, 7]]))
Example #14
0
def test_corner_multiple_corners_and_cells():
    """Test getting nodes for more than one corner and more than one cell."""
    node_ids = rfuncs.node_id_of_cell_corner(rmg, np.array([0, 1]), (4, 5))
    assert_array_equal(node_ids, np.array([[8, 6], [9, 7]]))
Example #15
0
def test_corner_with_no_cell_id():
    """Test without an arg for cells to mean all cells."""
    node_ids = rfuncs.node_id_of_cell_corner(rmg, np.array([0]))
    assert_array_equal(node_ids, np.array([[2, 3, 4, 7, 8, 9]]).T)
def test_corner_multiple_corners():
    node_ids = rfuncs.node_id_of_cell_corner(rmg, np.array([0, 1]), (4, ))
    assert_array_equal(node_ids, np.array([8, 6]))
def test_corner_multiple_corners_and_cells():
    node_ids = rfuncs.node_id_of_cell_corner(rmg, np.array([0, 1]), (4, 5))
    assert_array_equal(node_ids, np.array([[8, 6], [9, 7]]))
def test_corner_multiple_corners():
    node_ids = rfuncs.node_id_of_cell_corner(rmg, np.array([0, 1]), (4, ))
    assert_array_equal(node_ids, np.array([8, 6]))
def test_corner_with_scalar_cell_id():
    """Test using a scalar arg for cell."""
    node_ids = rfuncs.node_id_of_cell_corner(rmg, np.array([0]), 0)
    assert_array_equal(node_ids, np.array([[2]]))
def test_corner_multiple_corners_and_cells():
    node_ids = rfuncs.node_id_of_cell_corner(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.node_id_of_cell_corner(rmg, np.array([0]))
    assert_array_equal(node_ids, np.array([[2, 3, 4, 7, 8, 9]]).T)
def test_corner_type_list():
    node_ids = rfuncs.node_id_of_cell_corner(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.node_id_of_cell_corner(rmg, (0, 1), (4, 5))
    assert_array_equal(node_ids, np.array([[8, 6], [9, 7]]))
Example #24
0
def test_corner_with_scalar_cell_id():
    """Test using a scalar arg for cell."""
    node_ids = rfuncs.node_id_of_cell_corner(rmg, np.array([0]), 0)
    assert_array_equal(node_ids, np.array([[2]]))