Fix regular grid doctest

This commit is contained in:
Johannes Schönberger
2014-03-13 18:24:59 -04:00
parent b6e7fc8624
commit f280e0b13e
2 changed files with 5 additions and 5 deletions
+2 -2
View File
@@ -1043,10 +1043,10 @@ def warp(image, inverse_map=None, map_args={}, output_shape=None, order=1,
Use a callable (slow):
>>> def shift(xy):
>>> def shift_bottom(xy):
... xy[:, 1] -= 10
... return xy
>>> warped = warp(image, shift_right)
>>> warped = warp(image, shift_bottom)
Use a transformation matrix to warp an image (fast):
+3 -3
View File
@@ -29,21 +29,21 @@ def regular_grid(ar_shape, n_points):
>>> ar = np.zeros((20, 40))
>>> g = regular_grid(ar.shape, 8)
>>> g
[slice(5.0, None, 10.0), slice(5.0, None, 10.0)]
[slice(5, None, 10), slice(5, None, 10)]
>>> ar[g] = 1
>>> ar.sum()
8.0
>>> ar = np.zeros((20, 40))
>>> g = regular_grid(ar.shape, 32)
>>> g
[slice(2.0, None, 5.0), slice(2.0, None, 5.0)]
[slice(2, None, 5), slice(2, None, 5)]
>>> ar[g] = 1
>>> ar.sum()
32.0
>>> ar = np.zeros((3, 20, 40))
>>> g = regular_grid(ar.shape, 8)
>>> g
[slice(1.0, None, 3.0), slice(5.0, None, 10.0), slice(5.0, None, 10.0)]
[slice(1, None, 3), slice(5, None, 10), slice(5, None, 10)]
>>> ar[g] = 1
>>> ar.sum()
8.0