tweaker/vp8_cq_unittest.py

#!/usr/bin/python
"""Unit tests for encoder module."""

import unittest
import encoder

import vp8_cq

class TestVp8Cq(unittest.TestCase):
  def test_Init(self):
    codec = vp8_cq.Vp8CodecCqMode()
    self.assertEqual('vp8-cq', codec.name)
    # Verifying that the default config's value for min-q is still 32.
    # This is required for later tests to work properly.
    self.assertEqual('32', encoder.Option('min-q').GetValue(
            codec.start_encoder.parameters))

  def test_Fixup(self):
    codec = vp8_cq.Vp8CodecCqMode()
    config = '--min-q=47 --max-q=33'
    self.assertEqual(codec.ConfigurationFixups(config),
                     '--min-q=47 --max-q=47')

  def test_SuggestTweakIncreasesCq(self):
    codec = vp8_cq.Vp8CodecCqMode()
    videofile = encoder.Videofile('foofile_640_480_30.yuv')
    my_encoder = codec.start_encoder
    encoding = encoder.Encoding(my_encoder, 500, videofile)
    encoding.result = { 'bitrate' : 1000 }
    # Since the bitrate is too high, the suggstion should be to increase it.
    new_encoding = codec.SuggestTweak(encoding)
    self.assertEqual('33', encoder.Option('min-q').GetValue(
            new_encoding.encoder.parameters))

  def test_SuggestTweakDecreasesCq(self):
    codec = vp8_cq.Vp8CodecCqMode()
    videofile = encoder.Videofile('foofile_640_480_30.yuv')
    my_encoder = codec.start_encoder
    encoding = encoder.Encoding(my_encoder, 500, videofile)
    encoding.result = { 'bitrate' : 200 }
    # Since the bitrate is too high, the suggstion should be to increase it.
    new_encoding = codec.SuggestTweak(encoding)
    self.assertEqual('31', encoder.Option('min-q').GetValue(
            new_encoding.encoder.parameters))

if __name__ == '__main__':
    unittest.main()