def test_change_maximum_and_origin(): grid = StructuredGrid(origin=np.zeros(3), nsteps=np.array([5, 5, 5])) grid.origin = np.array([-1.0, -1.0, -1.0]) assert np.all(np.isclose(grid.origin, np.array([-1, -1, -1]))) assert np.all(np.isclose(grid.nsteps, np.array([6, 6, 6]))) assert np.all(np.isclose(grid.step_vector, np.ones(3))) grid.maximum = np.array([7.0, 7.0, 7.0]) assert np.all(np.isclose(grid.nsteps, np.array([9, 9, 9]))) assert np.all(np.isclose(grid.maximum, np.array([7.0, 7.0, 7.0]))) assert np.all(np.isclose(grid.step_vector, np.ones(3)))
def test_change_origin(): grid = StructuredGrid(origin=np.zeros(3), nsteps=np.array([5, 5, 5])) grid.origin = np.array([-1, -1, -1]) assert np.all(grid.origin == np.array([-1, -1, -1])) assert np.all(grid.nsteps == np.array([6, 6, 6])) assert np.all(grid.step_vector == np.ones(3))