コード例 #1
0
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]))
コード例 #2
0
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]))
コード例 #3
0
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]))
コード例 #4
0
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]))
コード例 #5
0
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]))
コード例 #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]]))
コード例 #7
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]]))
コード例 #8
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]]))
コード例 #9
0
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]]))
コード例 #10
0
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]))
コード例 #11
0
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]))
コード例 #12
0
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]))
コード例 #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]]))
コード例 #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]]))
コード例 #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)
コード例 #16
0
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]))
コード例 #17
0
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]]))
コード例 #18
0
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]))
コード例 #19
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]]))
コード例 #20
0
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]]))
コード例 #21
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)
コード例 #22
0
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]]))
コード例 #23
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]]))
コード例 #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]]))