Add wrapper for vim_command_output

Author: buztard

neovim/api/nvim.py

@@ -116,6 +116,10 @@ def command(self, string):
         """Execute a single ex command."""
         return self._session.request('vim_command', string)
 
+    def command_output(self, string):
+        """Execute a single ex command and return the output."""
+        return self._session.request('vim_command_output', string)
+
     def eval(self, string):
         """Evaluate a vimscript expression."""
         return self._session.request('vim_eval', string)

test/test_vim.py

@@ -18,6 +18,11 @@ def test_command():
     os.unlink(fname)
 
 
+@with_setup
+def test_command_output():
+    eq(vim.command_output('echo test'), 'test')
+
+
 @with_setup(setup=cleanup)
 def test_eval():
     vim.command('let g:v1 = "a"')