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