def test_namespace_get():
ns1 = Namespace()
ns1.add('big', True)
ns1.add('small', False)
ns1.add('medium', True)
ns1.add('huge', False)
assert not ns1.get('huge')
assert ns1.get('tiny') is None
assert ns1.get('tiny', 'sally') == 'sally'
def test_namespace_contains():
ns1 = Namespace()
ns1.add('big', True)
ns1.add('small', False)
ns1.add('medium', True)
ns1.add('huge', False)
assert 'big' in ns1
assert 'tiny' not in ns1
def test_namespace_string():
ns1 = Namespace()
ns1.add('big', True)
ns1.add('small', False)
ns1.add('medium', True)
ns1.add('huge', False)
assert str(ns1) == \
'Namespace(big=True, small=False, medium=True, huge=False)'
ns2 = Namespace(red=True, blue=False)
# Defaults are given in the constructor and are not put into the namespace
# until population is complete
assert str(ns2) == 'Namespace()'
# After population is complete, we finalize to add defaults not found.
# Adding defaults does not preserve order, so the
# option below is required
ns2.finalize()
assert str(ns2) in ('Namespace(red=True, blue=False)',
'Namespace(blue=False, red=True)')
def test_namespace_basic():
ns1 = Namespace()
assert not ns1
assert len(ns1) == 0
ns2 = Namespace()
ns2.add('big', True)
assert ns2
assert len(ns2) == 1
assert ns1 != ns2
ns1.add('big', True)
assert ns1 == ns2
'prompt': '',
'padding_text': '',
'xlm_lang': '',
'length': 20,
'num_samples': 10,
'temperature': 1.0,
'repetition_penalty': 1.0,
'top_k': 20,
'top_p': 0.0,
'no_cuda': False,
'seed': 42,
'stop_token': None
}
from input_reader import Namespace
args = Namespace()
for key, val in params.items():
args.add(key, val)
args.device = torch.device(
"cuda" if torch.cuda.is_available() and not args.no_cuda else "cpu")
args.n_gpu = torch.cuda.device_count()
set_seed(args)
args.model_type = args.model_type.lower()
class ParaphraseModel():
def test_namespace_lists():
ns1 = Namespace()
ns1.add('big', True)
ns1.add('small', False)
ns1.add('medium', True)
ns1.add('huge', False)
assert tuple(ns1.keys()) == ('big', 'small', 'medium', 'huge')
ns1.remove('small')
assert tuple(ns1.keys()) == ('big', 'medium', 'huge')
assert tuple(ns1.values()) == (True, True, False)
assert tuple(ns1.items()) == (('big', True), ('medium', True), ('huge', False))
