def test_dump_header_invalid(self):
parser = ArgumentParser()
with self.assertRaises(ValueError):
parser.dump_header = True
def test_dump_header_yaml(self):
parser = ArgumentParser()
parser.add_argument('--int', type=int, default=1)
parser.dump_header = ['line 1', 'line 2']
dump = parser.dump(parser.get_defaults())
self.assertEqual(dump, '# line 1\n# line 2\nint: 1\n')
def test_dump_header_json(self):
parser = ArgumentParser()
parser.add_argument('--int', type=int, default=1)
parser.dump_header = ['line 1', 'line 2']
dump = parser.dump(parser.get_defaults(), format='json')
self.assertEqual(dump, '{"int":1}')
