def test_parse_hparam_args__no_hparams(): hparam_args = ['--foo', '0.01'] with pytest.raises(ValueError): parse_hparam_args(hparam_args)
def test_parse_hparam_args__list(): hparam_args = ['--foo', 'HParams(boo=[1,2])'] assert parse_hparam_args(hparam_args) == {'foo': HParams(boo=[1, 2])}
def test_parse_hparam_args__single_flag(): hparam_args = ['-foo', 'HParams(boo=10**-6)'] with pytest.raises(ValueError): parse_hparam_args(hparam_args)
def test_parse_hparam_args__exponent(): hparam_args = ['--foo', 'HParams(boo=10**-6)'] assert parse_hparam_args(hparam_args) == {'foo': HParams(boo=10**-6)}
def test_parse_hparam_args__nesting(): hparam_args = ['--moo.foo', 'HParams(boo=1)'] assert parse_hparam_args(hparam_args) == {'moo.foo': HParams(boo=1)}
def test_parse_hparam_args__equals(): hparam_args = ['--foo=HParams(boo=1)'] assert parse_hparam_args(hparam_args) == {'foo': HParams(boo=1)}
def test_parse_hparam_args__string(): hparam_args = ['--foo', 'HParams(boo="WaveNet")'] assert parse_hparam_args(hparam_args) == {'foo': HParams(boo='WaveNet')}
def test_parse_hparam_args__decimal(): hparam_args = ['--foo', 'HParams(boo=0.01)'] assert parse_hparam_args(hparam_args) == {'foo': HParams(boo=0.01)}