def test_default_args(self): args = parse_args([]) assert args.filename is None assert args.text_attribute == 'text' assert args.pretty == False assert args.indent == 2 assert args.outputfile is None
def main(argv=None, stdin=sys.stdin): if not argv: argv = sys.argv[1:] args = parse_args(argv) if args.filename: nodes = parse_file(args.filename, args.text_attribute) else: nodes = parse_string(stdin.read(), args.text_attribute) if args.pretty: try: output = '\n'.join(pprint(nodes, args.indent)) except ValueError, e: errmsg = 'Error: %s' % e # prepend a newline before the error message if no filename was # given to distinguish it from the input if not args.filename: errmsg = '\n' + errmsg return errmsg
def test_indent_negative(self): args = parse_args(['-i', '-1337']) assert args.indent == 2
def test_indent_basic(self, argv): args = parse_args(argv) assert args.indent == 42
def test_filename_with_directory(self, option, tmpdir, capsys): with pytest.raises(SystemExit): parse_args([option, str(tmpdir)]) out, err = capsys.readouterr() assert out == '' assert err.strip().endswith('is not a file')
def test_nonexistent_filename(self, argv, capsys): with pytest.raises(SystemExit): parse_args(argv) out, err = capsys.readouterr() assert out == '' assert err.strip().endswith('does not exist')