def test_splittypes(self): parser = yargparse.YArgumentParser() args = parser.parse_args(['--train_param.deltas=[.5, .6, 7]']) self.assertEqual(args.train_param.deltas, [.5, .6, 7]) parser = yargparse.YArgumentParser() args = parser.parse_args(['--train_param.deltas [.5, .6, 7]']) self.assertEqual(args.train_param.deltas, [.5, .6, 7]) parser = yargparse.YArgumentParser() args = parser.parse_args(['--train_param.deltas:[.5, .6, 7]']) self.assertEqual(args.train_param.deltas, [.5, .6, 7])
def test_base_config(self): parser = yargparse.YArgumentParser() args = parser.parse_args() self.assertEqual(args.config, ['config.yaml']) self.assertEqual(args.train_param.deltas, [1, 2, 3]) self.assertEqual(args.train_param.lr, [.1, .2, .3]) self.assertEqual(args.features[0].dim, 100) self.assertEqual(args.features[0].type, "sparse") self.assertEqual(args.features[1].dim, 200) self.assertEqual(args.features[1].type, "dense")
def test_lists(self): parser = yargparse.YArgumentParser() args = parser.parse_args() self.assertEqual(len(args.class_weights), 1)
def test_multiple_configs(self): parser = yargparse.YArgumentParser() args = parser.parse_args('-c config.yaml config2.yaml'.split(' ')) self.assertEqual(len(args.class_weights), 1)
def test_nonevals(self): parser = yargparse.YArgumentParser() parser.add_argument('--train_epochs', default=None) args = parser.parse_args(['--matrix[1][0] 18']) self.assertEqual(args.train_epochs, 10)
def test_list_assign(self): parser = yargparse.YArgumentParser() args = parser.parse_args(['--matrix[1][0] 18']) self.assertEqual(args.matrix[1][0], 18)
def test_update_vals(self): parser = yargparse.YArgumentParser() args = parser.parse_args(['--train_param.deltas=[.5, .6, 7]']) self.assertEqual(args.config, ['config.yaml']) self.assertEqual(args.train_param.deltas, [.5, .6, 7])