Improve handling of printing source code only output.

2023-05-07 16:32:24 +02:00
parent e0fac306cb
commit 6ae0e7d084
3 changed files with 38 additions and 23 deletions
@@ -201,9 +201,9 @@ class TestCreateParser(unittest.TestCase):
            parser = create_parser()
            self.assertIsInstance(parser, argparse.ArgumentParser)
            mock_add_mutually_exclusive_group.assert_called_once_with(required=True)
-            mock_group.add_argument.assert_any_call('-p', '--print', help='YAML file to print')
+            mock_group.add_argument.assert_any_call('-p', '--print', help='File to print')
            mock_group.add_argument.assert_any_call('-q', '--question', nargs='*', help='Question to ask')
            mock_group.add_argument.assert_any_call('-D', '--chat-dump', help="Print chat as Python structure", action='store_true')
            mock_group.add_argument.assert_any_call('-d', '--chat', help="Print chat as readable text", action='store_true')
            self.assertTrue('.config.yaml' in parser.get_default('config'))
-            self.assertEqual(parser.get_default('number'), 3)
+            self.assertEqual(parser.get_default('number'), 1)