def test_movie_maker(mock_image, ImageSequenceClip):
from nglview.contrib.movie import MovieMaker
ImageSequenceClip.write_gif = MagicMock()
ImageSequenceClip.write_videofile = MagicMock()
traj = pytraj.datafiles.load_tz2()
view = nglview.show_pytraj(traj)
movie = MovieMaker(view,
in_memory=True,
download_folder='here',
render_params={'factor': 4},
moviepy_params={},
stop=2)
movie.make()
def test_movie_maker(ImageSequenceClip):
from nglview.contrib.movie import MovieMaker
ImageSequenceClip.write_gif = MagicMock()
ImageSequenceClip.write_videofile = MagicMock()
traj = pytraj.datafiles.load_tz2()
view = nglview.show_pytraj(traj)
movie = MovieMaker(view, in_memory=False)
movie.download_folder = os.path.join(os.path.dirname(__file__), 'data')
# fake _event
movie._event = FakeEvent()
movie._event._event_set = True
movie.make()
movie._event._event_set = False
movie.make()
movie = MovieMaker(view, in_memory=False)
movie.skip_render = True
movie.make()
movie.output = 'hello.mp4'
movie.make()
movie.interupt()
movie._event = None
movie.interupt()
movie.in_memory = True
movie.make()
movie._event._event_set = False
movie.make()
movie = MovieMaker(view,
download_folder='here',
render_params=dict(factor=4),
moviepy_params={},
stop=2)
movie.make()