# HG changeset patch
# User Emmanuel Gil Peyrot <linkmauve@linkmauve.fr>
# Date 1465775704 -3600
# Node ID 5df52b40fe54c76c0b295df18f80d874f486131e
# Parent  d255a932553f526688cdc6493e60fb8e41219fd4
Implement FunctionDef.returns.

diff --git a/src/ast_convert.rs b/src/ast_convert.rs
--- a/src/ast_convert.rs
+++ b/src/ast_convert.rs
@@ -564,13 +564,14 @@ fn parse_statement(py: Python, ast: PyOb
         let name = ast.getattr(py, "name").unwrap();
         let args = ast.getattr(py, "args").unwrap();
         let body = ast.getattr(py, "body").unwrap();
+        let returns = ast.getattr(py, "returns").unwrap();
 
         let name = get_str(py, name);
         let args = parse_arguments(py, args);
         let body = parse_list(py, body, parse_statement);
 
         let decorators = vec!();
-        let returns = None;
+        let returns = parse_optional_expr(py, returns);
 
         stmt::FunctionDef(name, args, body, decorators, returns)
     } else if is_instance(&ast, &global_type) {
diff --git a/src/ast_dump.rs b/src/ast_dump.rs
--- a/src/ast_dump.rs
+++ b/src/ast_dump.rs
@@ -226,7 +226,18 @@ impl stmt {
         let current_indent = make_indent(indent);
         match self.clone() {
             stmt::ClassDef(name, bases, keywords, body, decorator_list) => format!("{}class {}({}):\n{}", current_indent, name, vec_to_strings_vec(bases).join(", "), statements_to_string(indent, body)),
-            stmt::FunctionDef(name, arguments, body, decorator_list, returns) => format!("{}def {}({}):\n{}", current_indent, name, arguments.to_string(), statements_to_string(indent, body)),
+            stmt::FunctionDef(name, arguments, body, decorator_list, returns) => {
+                format!("{}def {}({}){}:\n{}",
+                    current_indent,
+                    name,
+                    arguments.to_string(),
+                    match returns {
+                        None => String::new(),
+                        Some(returns) => format!(" -> {}", returns.to_string())
+                    },
+                    statements_to_string(indent, body)
+                )
+            }
             stmt::Global(names) => format!("{}global {}", current_indent, names.join(", ")),
             stmt::Nonlocal(names) => format!("{}nonlocal {}", 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_functiondef.py b/tests/test_parse_files/test_functiondef.py
--- a/tests/test_parse_files/test_functiondef.py
+++ b/tests/test_parse_files/test_functiondef.py
@@ -1,6 +1,6 @@
 def func0():
     pass
-def func1(a, b: str):
+def func1(a, b: str) -> None:
     pass
 def func2(a, *, b: composite.type, c=2, d: int=3):
     pass