master: decimal and float format in config file

1 files changed, 4 insertions, 3 deletions

diff --git a/tests/test_main.py b/tests/test_main.py
index ba11298c..e461fc0d 100644
--- a/tests/test_main.py
+++ b/tests/test_main.py
@@ -45,21 +45,22 @@ def test_obfuscate_process_password():
 
     setproctitle.setproctitle(original_title)
 
+
 def test_format_output():
     results = format_output('Title', [('abc', 'def')], ['head1', 'head2'],
-                            'test status', 'psql')
+                            'test status', 'psql',  dcmlfmt='d', floatfmt='g',)
     expected = ['Title', '+---------+---------+\n| head1   | head2   |\n|---------+---------|\n| abc     | def     |\n+---------+---------+', 'test status']
     assert results == expected
 
 def test_format_output_auto_expand():
     table_results = format_output('Title', [('abc', 'def')],
-                                  ['head1', 'head2'], 'test status', 'psql',
+                                  ['head1', 'head2'], 'test status', 'psql', dcmlfmt='d', floatfmt='g',
                                   max_width=100)
     table = ['Title', '+---------+---------+\n| head1   | head2   |\n|---------+---------|\n| abc     | def     |\n+---------+---------+', 'test status']
     assert table_results == table
 
     expanded_results = format_output('Title', [('abc', 'def')],
-                                     ['head1', 'head2'], 'test status', 'psql',
+                                     ['head1', 'head2'], 'test status', 'psql', dcmlfmt='d', floatfmt='g',
                                      max_width=1)
     expanded = ['Title', u'-[ RECORD 0 ]-------------------------\nhead1 | abc\nhead2 | def\n', 'test status']
     assert expanded_results == expanded