def test_shape_4_by_5(): cell_nodes = sgrid.node_at_cell((4, 5)) assert_array_equal(cell_nodes, np.array([6, 7, 8, 11, 12, 13]))
def test_2d_shape_3_by_3(): cell_nodes = sgrid.node_at_cell((3, 3)) assert_array_equal(cell_nodes, np.array([4]))