# HG changeset patch
# User Bastien Orivel <eijebong@bananium.fr>
# Date 1465046887 -7200
# Node ID e244c492ff07f5e5ca9f517e8d7f5cf6d3f65440
# Parent  2f1625a65d5687dbd7bb29598abb8f12ad353d00
Add a test for ClassDef and fix ast_dump for smt::ClassDef.

diff --git a/src/ast_dump.rs b/src/ast_dump.rs
--- 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)),
diff --git a/tests/test_parse_files/test_classdef.py b/tests/test_parse_files/test_classdef.py
new file mode 100644
--- /dev/null
+++ b/tests/test_parse_files/test_classdef.py
@@ -0,0 +1,3 @@
+class a():
+    def b():
+        print("a")