Add return statement test
Add return statement test

tree-sitter/tree-sitter-python,tree-sitter/tree-sitter-python,tree-sitter/tree-sitter-python,tree-sitter/tree-sitter-python,tree-sitter/tree-sitter-python