import unittest from segment import VideoSegment from segment.tests import SegmentSharedTestCase class VideoSegmentTestCase(SegmentSharedTestCase): """Test the VideoSegment class.""" EXPECTED_TYPE = "video" EXPECTED_TRIM = "trim" EXPECTED_SETPTS = "setpts" def setUp(self): """Set up for test.""" super().setUp() self.segment = VideoSegment( file=self.EXPECTED_INPUT_FILE, punch_in=self.EXPECTED_PUNCH_IN, punch_out=self.EXPECTED_PUNCH_OUT, input_stream=self.EXPECTED_INPUT_STREAM) self.expected_output_options = [ "-map", "{n}:v".format(n=self.segment.input_stream) ] def test_init(self): """Test segment initialises correctly.""" super().test_init() self.assertEqual( self.segment._temp_frame_file, "", msg="temp frame file = ") # Tricky to test get_last_frame_number() and generate_frame() # because they use pexpect. # Remove SegmentSharedTestCase from the namespace so we don't run # the shared tests twice. See <https://stackoverflow.com/a/22836015>. del(SegmentSharedTestCase)