[draw] fix extrude without out-of-bounds slice
This commit is contained in:
parent
8895b06f08
commit
481b56874e
2 changed files with 30 additions and 10 deletions
|
|
@ -238,6 +238,23 @@ def test_get_slice_uses_shifted_grid_bounds() -> None:
|
|||
grid.get_slice(cell_data, Plane(x=2.1), which_shifts=0)
|
||||
|
||||
|
||||
def test_draw_extrude_rectangle_uses_boundary_slice() -> None:
|
||||
grid = Grid([[0, 1, 2], [0, 1, 2], [0, 1, 2]], shifts=[[0, 0, 0]])
|
||||
cell_data = grid.allocate(0)
|
||||
source = numpy.array([[1, 2],
|
||||
[3, 4]], dtype=float)
|
||||
cell_data[0, :, :, 1] = source
|
||||
|
||||
grid.draw_extrude_rectangle(
|
||||
cell_data,
|
||||
rectangle=[[0, 0, 2], [2, 2, 2]],
|
||||
direction=2,
|
||||
polarity=-1,
|
||||
distance=2,
|
||||
)
|
||||
|
||||
assert_allclose(cell_data[0, :, :, 0], source)
|
||||
assert_allclose(cell_data[0, :, :, 1], source)
|
||||
|
||||
|
||||
def test_sampled_preview_exyz_tracks_nonuniform_centers() -> None:
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue