diff options
| author | Sebastian Berg <sebastian@sipsolutions.net> | 2016-01-21 22:23:53 +0100 |
|---|---|---|
| committer | Sebastian Berg <sebastian@sipsolutions.net> | 2016-02-13 22:26:40 +0100 |
| commit | f7e64cc699567124114e41ad7f518d6825807cc3 (patch) | |
| tree | 42808673dcf3c1daa99ce960889cc15fa47a75c4 /numpy/lib/tests/test_stride_tricks.py | |
| parent | db0304e4fe483a9d591424b35affd0d0e197000d (diff) | |
| download | numpy-f7e64cc699567124114e41ad7f518d6825807cc3.tar.gz | |
DOC: Add documentation for as_strided
Diffstat (limited to 'numpy/lib/tests/test_stride_tricks.py')
| -rw-r--r-- | numpy/lib/tests/test_stride_tricks.py | 17 |
1 files changed, 17 insertions, 0 deletions
diff --git a/numpy/lib/tests/test_stride_tricks.py b/numpy/lib/tests/test_stride_tricks.py index 06e659002..95df135cf 100644 --- a/numpy/lib/tests/test_stride_tricks.py +++ b/numpy/lib/tests/test_stride_tricks.py @@ -317,6 +317,23 @@ def test_as_strided(): a_view = as_strided(a, shape=(3, 4), strides=(0, a.itemsize)) assert_equal(a.dtype, a_view.dtype) +def as_strided_writeable(): + arr = np.ones(10) + view = as_strided(arr, writeable=False) + assert_(not view.flags.writeable) + + # Check that writeable also is fine: + view = as_strided(arr, writeable=True) + assert_(view.flags.writeable) + view[...] = 3 + assert_array_equal(arr, np.full_like(arr, 3)) + + # Test that things do not break down for readonly: + arr.flags.writeable = False + view = as_strided(arr, writeable=False) + view = as_strided(arr, writeable=True) + assert_(not view.flags.writeable) + class VerySimpleSubClass(np.ndarray): def __new__(cls, *args, **kwargs): |