changeset 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 2f1625a65d56
children e7b82aa558ab
files src/ast_dump.rs tests/test_parse_files/test_classdef.py
diffstat 2 files changed, 4 insertions(+), 1 deletions(-) [+]
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)),
new file mode 100644
--- /dev/null
+++ b/tests/test_parse_files/test_classdef.py
@@ -0,0 +1,3 @@
+class a():
+    def b():
+        print("a")