format() should accept any AST node, not just top-level statements

[pulpdrew]

It would be useful if the format() function accepted any AST node kind, instead of just Statement[]. This would allow consistent formatting of arbitrary expressions, which may not be full statements.