mirror of
https://github.com/Zulko/moviepy.git
synced 2021-07-27 01:17:47 +03:00
0751d072d9
* Add module docstrings except in FXs * Update flake8 config * Remove 'find_latest_imagemagick_version.py' * Add tests modules docstrings * Don't execute doctests for tests/ modules * Prevent test error in Windows
45 lines
958 B
Python
45 lines
958 B
Python
"""MoviePy examples tests."""
|
|
|
|
import os
|
|
|
|
import numpy as np
|
|
import pytest
|
|
|
|
from moviepy.video.io.bindings import mplfig_to_npimage
|
|
from moviepy.video.VideoClip import VideoClip
|
|
|
|
from tests.test_helper import TMP_DIR
|
|
|
|
|
|
try:
|
|
import matplotlib.pyplot
|
|
except ImportError:
|
|
matplotlib = None
|
|
else:
|
|
matplotlib = True
|
|
|
|
|
|
@pytest.mark.skipif(not matplotlib, reason="no matplotlib")
|
|
def test_matplotlib_simple_example():
|
|
import matplotlib.pyplot as plt
|
|
|
|
plt.switch_backend("Agg")
|
|
|
|
x = np.linspace(-2, 2, 200)
|
|
duration = 0.5
|
|
|
|
fig, ax = plt.subplots()
|
|
|
|
def make_frame(t):
|
|
ax.clear()
|
|
ax.plot(x, np.sinc(x ** 2) + np.sin(x + 2 * np.pi / duration * t), lw=3)
|
|
ax.set_ylim(-1.5, 2.5)
|
|
return mplfig_to_npimage(fig)
|
|
|
|
animation = VideoClip(make_frame, duration=duration)
|
|
|
|
filename = os.path.join(TMP_DIR, "matplotlib.gif")
|
|
animation.write_gif(filename, fps=20)
|
|
|
|
assert os.path.isfile(filename)
|