Merge pull request #463 from grlee77/test_doc_fix

TST: fix misc. doctest failures (test_doc.py)

Author: rgommers

.travis.yml

@@ -73,13 +73,15 @@ script:
         pip install PyWavelets*.whl -v
         pushd demo
         nosetests pywt
+        python ../pywt/tests/test_doc.py
         popd
     elif [ "${USE_SDIST}" == "1" ]; then
         python setup.py sdist
         # Move out of source directory to avoid finding local pywt
         pushd dist
         pip install PyWavelets* -v
         nosetests pywt
+        python ../pywt/tests/test_doc.py
         popd
     elif [ "${REFGUIDE_CHECK}" == "1" ]; then
         pip install -e . -v

doc/source/ref/other-functions.rst

@@ -71,5 +71,5 @@ Each can be loaded via a function of the same name.
 
     >>> import pywt
     >>> camera = pywt.data.camera()
-    >>> doppler = pywt.data.demo_signal('doppler')
+    >>> doppler = pywt.data.demo_signal('doppler', 1024)
     >>> available_signals = pywt.data.demo_signal('list')

doc/source/ref/wavelets.rst

@@ -332,7 +332,7 @@ from plain Python lists of filter coefficients and a *filter bank-like* object.
     >>> print(wavelet)
     ContinuousWavelet gaus1
       Family name:    Gaussian
-      Short name:     db
+      Short name:     gaus
       Symmetry:       anti-symmetric
       DWT:            False
       CWT:            True
@@ -390,7 +390,7 @@ Approximating wavelet functions - ``ContinuousWavelet.wavefun()``
     >>> print(wavelet)
     ContinuousWavelet gaus1
       Family name:    Gaussian
-      Short name:     db
+      Short name:     gaus
       Symmetry:       anti-symmetric
       DWT:            False
       CWT:            True

doc/source/regression/dwt-idwt.rst

@@ -74,7 +74,7 @@ extension mode (please refer to the PyWavelets' documentation for the
 :ref:`extension modes <Modes>` available:
 
     >>> pywt.Modes.modes
-    ['zero', 'constant', 'symmetric', 'periodic', 'smooth', 'periodization', 'reflect']
+    ['zero', 'constant', 'symmetric', 'periodic', 'smooth', 'periodization', 'reflect', 'antisymmetric', 'antireflect']
 
 As you see in the above example, the :ref:`periodization <Modes.periodization>`
 (periodization) mode is slightly different from the others. It's aim when

doc/source/regression/modes.rst

@@ -19,7 +19,7 @@ Import :mod:`pywt` first
 List of available signal extension :ref:`modes <Modes>`:
 
     >>> print(pywt.Modes.modes)
-    ['zero', 'constant', 'symmetric', 'periodic', 'smooth', 'periodization', 'reflect']
+    ['zero', 'constant', 'symmetric', 'periodic', 'smooth', 'periodization', 'reflect', 'antisymmetric', 'antireflect']
 
 
 Invalid mode name should rise a :exc:`ValueError`:

pywt/_multilevel.py

@@ -933,6 +933,7 @@ def wavedecn_size(shapes):
 
     Examples
     --------
+    >>> import numpy as np
     >>> import pywt
     >>> data_shape = (64, 32)
     >>> shapes = pywt.wavedecn_shapes(data_shape, 'db2', mode='periodization')

pywt/tests/test_doc.py

@@ -5,6 +5,11 @@
 import os
 import unittest
 
+try:
+    import numpy as np
+    np.set_printoptions(legacy='1.13')
+except TypeError:
+    pass
 
 pdir = os.path.pardir
 docs_base = os.path.abspath(os.path.join(os.path.dirname(__file__),