def test_show_formats(self): with captured_stdout() as stdout: show_formats() # the output should be a header line + one line per format num_formats = len(ARCHIVE_FORMATS.keys()) output = [line for line in stdout.getvalue().split("\n") if line.strip().startswith("--formats=")] self.assertEqual(len(output), num_formats)
def test_show_formats(self): with captured_stdout() as stdout: show_formats() num_formats = len(ARCHIVE_FORMATS.keys()) output = [ line for line in stdout.getvalue().split('\n') if line.strip().startswith('--formats=') ] self.assertEqual(len(output), num_formats)