STY: Move

Author: larsoner

lib/matplotlib/tests/test_animation.py

@@ -13,6 +13,33 @@
 from matplotlib import animation
 
 
+@pytest.fixture()
+def anim(request):
+    """Create a simple animation (with options)."""
+    fig, ax = plt.subplots()
+    line, = ax.plot([], [])
+
+    ax.set_xlim(0, 10)
+    ax.set_ylim(-1, 1)
+
+    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,
+
+    # "klass" can be passed to determine the class returned by the fixture
+    kwargs = dict(getattr(request, 'param', {}))  # make a copy
+    klass = kwargs.pop('klass', animation.FuncAnimation)
+    if 'frames' not in kwargs:
+        kwargs['frames'] = 5
+    return klass(fig=fig, func=animate, init_func=init, **kwargs)
+
+
 class NullMovieWriter(animation.AbstractMovieWriter):
     """
     A minimal MovieWriter.  It doesn't actually write anything.
@@ -118,32 +145,6 @@ def isAvailable(cls):
     (writer, Path(output)) for writer, output in WRITER_OUTPUT]
 
 
-@pytest.fixture()
-def anim(request):
-    fig, ax = plt.subplots()
-    line, = ax.plot([], [])
-
-    ax.set_xlim(0, 10)
-    ax.set_ylim(-1, 1)
-
-    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,
-
-    # "klass" can be passed to determine the class returned by the fixture
-    kwargs = dict(getattr(request, 'param', {}))  # make a copy
-    klass = kwargs.pop('klass', animation.FuncAnimation)
-    if 'frames' not in kwargs:
-        kwargs['frames'] = 5
-    return klass(fig=fig, func=animate, init_func=init, **kwargs)
-
-
 # Smoke test for saving animations.  In the future, we should probably
 # design more sophisticated tests which compare resulting frames a-la
 # matplotlib.testing.image_comparison