def test_GetValueNotPresent(self):
option = encoder.Option('foo', ['foo', 'bar'])
config = encoder.OptionValueSet(encoder.OptionSet(option),
'--notfoo=foo')
with self.assertRaises(encoder.Error):
config.GetValue('foo')
def test_ReproduceFlag(self):
opts = encoder.OptionSet(encoder.ChoiceOption(['foo']))
valueset = encoder.OptionValueSet(opts, '--foo')
self.assertEqual('--foo', valueset.ToString())
def test_ReproduceOneArg(self):
valueset = encoder.OptionValueSet(encoder.OptionSet(), '--foo=bar')
self.assertEqual('--foo=bar', valueset.ToString())
def test_GetValue(self):
valueset = encoder.OptionValueSet(
encoder.OptionSet(encoder.Option('foo', ['baz', 'bar'])),
'--foo=bar')
self.assertEqual('bar', valueset.GetValue('foo'))
def StartEncoder(self, context):
return encoder.Encoder(
context,
encoder.OptionValueSet(self.option_set,
'--preset slow --tune psnr --threads 1',
formatter=self.option_formatter))
def StartEncoder(self, context):
return encoder.Encoder(
context, encoder.OptionValueSet(self.option_set, "--score=5"))
def StartEncoder(self, context):
return encoder.Encoder(
context,
encoder.OptionValueSet(
self.option_set,
'--passes=1 --good --noise-sensitivity=0 --cpu-used=5'))
def test_ChangeValueOfUnknownOption(self):
opts = encoder.OptionSet(encoder.ChoiceOption(['foo', 'bar']))
valueset = encoder.OptionValueSet(opts, '--foo')
with self.assertRaises(encoder.Error):
# pylint: disable=W0612
newset = valueset.ChangeValue('nosuchname', 'bar')
def test_UnlistedValueFailsToParse(self):
options = encoder.OptionSet(
encoder.Option('foo', ['foo', 'bar']).Mandatory())
with self.assertRaises(encoder.ParseError):
encoder.OptionValueSet(options, '--foo=nonsense')
def EncoderFromParameterString(self, parameter_string):
return encoder.Encoder(self.optimizer.context,
encoder.OptionValueSet(self.optimizer.context.codec.option_set,
parameter_string))
def test_MissingMandatoryFailsToParse(self):
options = encoder.OptionSet(
encoder.Option('foo', ['foo', 'bar']).Mandatory())
with self.assertRaises(encoder.ParseError):
encoder.OptionValueSet(options, '')
def test_ConfigFixups(self):
codec = vp8_mpeg_1d.Vp8CodecMpeg1dMode()
fixups = codec.ConfigurationFixups(
encoder.OptionValueSet(codec.option_set,
'--fixed-q=1 --gold-q=1 --key-q=3'))
self.assertEqual('--fixed-q=6 --gold-q=4 --key-q=3', fixups.ToString())
def StartEncoder(self, context): return encoder.Encoder(context, encoder.OptionValueSet(None, ''))
def test_UnknownFlagPreserved(self):
opts = encoder.OptionSet(encoder.ChoiceOption(['foo']))
valueset = encoder.OptionValueSet(opts, '--bar')
self.assertEqual('--bar', valueset.ToString())
def test_RandomlyChangeConfig(self):
codec = DummyCodec()
otherconfig = codec.RandomlyChangeConfig(
encoder.OptionValueSet(codec.option_set, '--score=5'))
self.assertNotEqual(otherconfig, '--score=5')
self.assertIn(otherconfig, ['--score=0', '--score=10'])
def test_FlagsSorted(self):
opts = encoder.OptionSet(encoder.ChoiceOption(['foo']))
valueset = encoder.OptionValueSet(opts, '--foo --bar')
self.assertEqual('--bar --foo', valueset.ToString())
def test_ParametersCanChangeMayReturnFalse(self):
context = encoder.Context(DummyCodec())
my_encoder = encoder.Encoder(
context, encoder.OptionValueSet(encoder.OptionSet(),
'--parameters'))
self.assertFalse(my_encoder.ParametersCanChange())
def test_RandomlyRemoveParameterSuccessfully(self):
config = encoder.OptionValueSet(
encoder.OptionSet(encoder.Option('foo', ['foo', 'bar'])),
'--foo=foo')
newconfig = config.RandomlyRemoveParameter()
self.assertEqual('', newconfig.ToString())
def StartEncoder(self, context):
return encoder.Encoder(
context,
encoder.OptionValueSet(self.option_set,
self.start_encoder_parameters))
