Add some animation tests.

anntzer · anntzer · commit ef8df261d43f · 2018-03-15T19:17:24.000-07:00
diff --git a/.appveyor.yml b/.appveyor.yml
@@ -63,7 +63,7 @@ install:
   - activate test-environment
   - echo %PYTHON_VERSION% %TARGET_ARCH%
   # pytest-cov>=2.3.1 due to https://github.com/pytest-dev/pytest-cov/issues/124
-  - pip install -q "pytest!=3.3.0,>=3.2.0" "pytest-cov>=2.3.1" pytest-rerunfailures pytest-timeout pytest-xdist
+  - pip install -q "pytest>=3.3.1" "pytest-cov>=2.3.1" pytest-rerunfailures pytest-timeout pytest-xdist
 
   # Apply patch to `subprocess` on Python versions > 2 and < 3.6.3
   # https://github.com/matplotlib/matplotlib/issues/9176
diff --git a/.travis.yml b/.travis.yml
@@ -18,14 +18,16 @@ addons:
     paths:
     - result_images.tar.bz2
   apt:
+    sources:
+      - ppa:jonathonf/ffmpeg-3
     packages:
       - cm-super
       - dvipng
+      - ffmpeg
       - gdb
       - gir1.2-gtk-3.0
       - graphviz
       - inkscape
-      - libav-tools
       - libcairo2
       - libgeos-dev
       - libgirepository-1.0.1
@@ -52,7 +54,7 @@ env:
     - NUMPY=numpy
     - PANDAS=
     - PYPARSING=pyparsing
-    - PYTEST='pytest!=3.3.0,>=3.2.0'
+    - PYTEST='pytest>=3.3.1'
     - PYTEST_COV=pytest-cov
     - PYTEST_PEP8=
     - SPHINX=sphinx
@@ -74,7 +76,7 @@ matrix:
         - NUMPY=numpy==1.10.0
         - PANDAS='pandas<0.21.0'
         - PYPARSING=pyparsing==2.0.1
-        - PYTEST=pytest==3.1.0
+        - PYTEST=pytest==3.3.1
         - PYTEST_COV=pytest-cov==2.3.1
         - SPHINX=sphinx==1.3
     - python: 3.5
diff --git a/doc/devel/contributing.rst b/doc/devel/contributing.rst
@@ -151,7 +151,7 @@ environment is set up properly::
 
 .. note::
 
-  **Additional dependencies for testing**: pytest_ (version 3.1 or later),
+  **Additional dependencies for testing**: pytest_ (version 3.3.1 or later),
   Ghostscript_, Inkscape_
 
 .. seealso::
diff --git a/doc/devel/testing.rst b/doc/devel/testing.rst
@@ -25,7 +25,7 @@ local FreeType build
 
 The following software is required to run the tests:
 
-  - pytest_ (>=3.1)
+  - pytest_ (>=3.3.1)
   - Ghostscript_ (to render PDF files)
   - Inkscape_ (to render SVG files)
 
diff --git a/lib/matplotlib/animation.py b/lib/matplotlib/animation.py
@@ -687,8 +687,7 @@ def _args(self):
                 '-vframes', str(self._frame_counter)] + self.output_args
 
 
-# Base class of avconv information.  AVConv has identical arguments to
-# FFMpeg
+# Base class of avconv information.  AVConv has identical arguments to FFMpeg.
 class AVConvBase(FFMpegBase):
     '''Mixin class for avconv output.
 
diff --git a/lib/matplotlib/tests/test_animation.py b/lib/matplotlib/tests/test_animation.py
@@ -1,7 +1,5 @@
-from __future__ import absolute_import, division, print_function
-
-import six
-
+import os
+from pathlib import Path
 import sys
 import tempfile
 
@@ -42,29 +40,35 @@ def finish(self):
         pass
 
 
-def test_null_movie_writer():
-    # Test running an animation with NullMovieWriter.
-
-    fig = plt.figure()
+def make_animation(**kwargs):
+    fig, ax = plt.subplots()
+    line, = ax.plot([])
 
     def init():
         pass
 
     def animate(i):
-        pass
+        line.set_data([0, 1], [0, i])
+        return line,
+
+    return animation.FuncAnimation(fig, animate, **kwargs)
+
+
+def test_null_movie_writer():
+    # Test running an animation with NullMovieWriter.
 
     num_frames = 5
+    anim = make_animation(frames=num_frames)
+
     filename = "unused.null"
     dpi = 50
     savefig_kwargs = dict(foo=0)
-
-    anim = animation.FuncAnimation(fig, animate, init_func=init,
-                                   frames=num_frames)
     writer = NullMovieWriter()
+
     anim.save(filename, dpi=dpi, writer=writer,
               savefig_kwargs=savefig_kwargs)
 
-    assert writer.fig == fig
+    assert writer.fig == plt.figure(1)  # The figure used by make_animation.
     assert writer.outfile == filename
     assert writer.dpi == dpi
     assert writer.args == ()
@@ -178,23 +182,8 @@ def animate(i):
 
 
 def test_no_length_frames():
-    fig, ax = plt.subplots()
-    line, = ax.plot([], [])
-
-    def init():
-        line.set_data([], [])
-        return line,
-
-    def animate(i):
-        x = np.linspace(0, 10, 100)
-        y = np.sin(x + i)
-        line.set_data(x, y)
-        return line,
-
-    anim = animation.FuncAnimation(fig, animate, init_func=init,
-                                   frames=iter(range(5)))
-    writer = NullMovieWriter()
-    anim.save('unused.null', writer=writer)
+    (make_animation(frames=iter(range(5)))
+     .save('unused.null', writer=NullMovieWriter()))
 
 
 def test_movie_writer_registry():
@@ -212,10 +201,64 @@ def test_movie_writer_registry():
     assert not animation.writers.is_available("ffmpeg")
     # something which is guaranteed to be available in path
     # and exits immediately
-    bin = u"true" if sys.platform != 'win32' else u"where"
+    bin = "true" if sys.platform != 'win32' else "where"
     mpl.rcParams['animation.ffmpeg_path'] = bin
     assert animation.writers._dirty
     animation.writers.list()  # resets
     assert not animation.writers._dirty
     assert animation.writers.is_available("ffmpeg")
     mpl.rcParams['animation.ffmpeg_path'] = ffmpeg_path
+
+
+@pytest.mark.skipif(
+    not animation.writers.is_available(mpl.rcParams["animation.writer"]),
+    reason="animation writer not installed")
+@pytest.mark.parametrize("method_name", ["to_html5_video", "to_jshtml"])
+def test_embed_limit(method_name, caplog):
+    with mpl.rc_context({"animation.embed_limit": 1e-6}):  # ~1 byte.
+        getattr(make_animation(frames=1), method_name)()
+    assert len(caplog.records) == 1
+    record, = caplog.records
+    assert (record.name == "matplotlib.animation"
+            and record.levelname == "WARNING")
+
+
+@pytest.mark.skipif(
+    not animation.writers.is_available(mpl.rcParams["animation.writer"]),
+    reason="animation writer not installed")
+@pytest.mark.parametrize(
+    "method_name",
+    ["to_html5_video",
+     pytest.mark.xfail("to_jshtml")])  # Needs to be fixed.
+def test_cleanup_temporaries(method_name, tmpdir):
+    with tmpdir.as_cwd():
+        getattr(make_animation(frames=1), method_name)()
+        assert list(Path(tmpdir).iterdir()) == []
+
+
+# Currently, this fails with a ValueError after we try to communicate() twice
+# with the Popen.
+@pytest.mark.xfail
+@pytest.mark.skipif(os.name != "posix", reason="requires a POSIX OS")
+def test_failing_ffmpeg(tmpdir, monkeypatch):
+    """
+    Test that we correctly raise an OSError when ffmpeg fails.
+
+    To do so, mock ffmpeg using a simple executable shell script that
+    succeeds when called with no arguments (so that it gets registered by
+    `isAvailable`), but fails otherwise, and add it to the $PATH.
+    """
+    try:
+        with tmpdir.as_cwd():
+            monkeypatch.setenv("PATH", ".:" + os.environ["PATH"])
+            exe_path = Path(tmpdir, "ffmpeg")
+            exe_path.write_text("""\
+#!/bin/sh
+[[ "$@" -gt 0 ]] && exit 1 || exit 0
+""")
+            os.chmod(str(exe_path), 0o755)
+            animation.writers.reset_available_writers()
+            with pytest.raises(OSError):
+                make_animation().save("test.mpeg")
+    finally:
+        animation.writers.reset_available_writers()
diff --git a/setupext.py b/setupext.py
@@ -739,7 +739,7 @@ def get_namespace_packages(self):
 
 class Tests(OptionalPackage):
     name = "tests"
-    pytest_min_version = '3.1'
+    pytest_min_version = '3.3.1'
     default_config = False
 
     def check(self):
