diff src/ast_dump.rs @ 38:e244c492ff07

Add a test for ClassDef and fix ast_dump for smt::ClassDef.
author Bastien Orivel <eijebong@bananium.fr>
date Sat, 04 Jun 2016 15:28:07 +0200
parents bc3437ecb12b
children 5f1d285471af
line wrap: on
line diff
--- a/src/ast_dump.rs
+++ b/src/ast_dump.rs
@@ -178,7 +178,7 @@ impl stmt {
     fn to_string(&self, indent: usize) -> String {
         let current_indent = make_indent(indent);
         match self.clone() {
-            stmt::ClassDef(name, bases, keywords, body, decorator_list) => format!("{}def {}({}):\n{}", current_indent, name, args_to_string(bases), statements_to_string(indent, body)),
+            stmt::ClassDef(name, bases, keywords, body, decorator_list) => format!("{}class {}({}):\n{}", current_indent, name, args_to_string(bases), statements_to_string(indent, body)),
             stmt::FunctionDef(name, arguments, body, decorator_list, returns) => format!("{}def {}({}):\n{}", current_indent, name, arguments_to_string(arguments), statements_to_string(indent, body)),
             stmt::Global(names) => format!("{}global {}", current_indent, names.join(", ")),
             stmt::If(test, body, orelse) => format!("{}if {}:\n{}", current_indent, test.to_string(), if_else_statements_to_string(indent, body, orelse)),