CInema: Clean up size issues

src/platform/python/cinema/movie.py

@@ -8,15 +8,17 @@ Output = namedtuple('Output', ['video'])
 class Tracer(object):
     def __init__(self, core):
         self.core = core
-        self.framebuffer = Image(*core.desired_video_dimensions())
-        self.core.set_video_buffer(self.framebuffer)
         self._video_fifo = []
 
     def yield_frames(self, skip=0, limit=None):
+        self.framebuffer = Image(*self.core.desired_video_dimensions())
+        self.core.set_video_buffer(self.framebuffer)
         self.core.reset()
         skip = (skip or 0) + 1
         while skip > 0:
             frame = self.core.frame_counter
+            self.framebuffer = Image(*self.core.desired_video_dimensions())
+            self.core.set_video_buffer(self.framebuffer)
             self.core.run_frame()
             skip -= 1
         while frame <= self.core.frame_counter and limit != 0:

src/platform/python/cinema/test.py

@@ -3,7 +3,6 @@ import os.path
 import mgba.core
 import mgba.image
 import cinema.movie
-import itertools
 import glob
 import re
 from copy import deepcopy
@@ -73,7 +72,7 @@ class VideoTest(CinemaTest):
         self.tracer = cinema.movie.Tracer(self.core)
 
     def generate_frames(self):
-        for i, frame in zip(itertools.count(), self.tracer.video(**self.output_settings())):
+        for i, frame in enumerate(self.tracer.video(**self.output_settings())):
             try:
                 baseline = VideoFrame.load(os.path.join(self.path, self.BASELINE % i))
                 yield baseline, frame, VideoFrame.diff(baseline, frame)
@@ -85,7 +84,7 @@ class VideoTest(CinemaTest):
         assert not any(any(diffs[0].image.convert("L").point(bool).getdata()) for diffs in self.diffs)
 
     def generate_baseline(self):
-        for i, frame in zip(itertools.count(), self.tracer.video(**self.output_settings())):
+        for i, frame in enumerate(self.tracer.video(**self.output_settings())):
             frame.save(os.path.join(self.path, self.BASELINE % i))