import vim module by default #430

Vim-compatible.
fixes neovim/neovim#11783

Author: Sunny Rahul

pynvim/plugin/script_host.py

@@ -43,7 +43,8 @@ def __init__(self, nvim):
         exec('import sys', self.module.__dict__)
         self.legacy_vim = LegacyVim.from_nvim(nvim)
         sys.modules['vim'] = self.legacy_vim
-
+        # mimic Vim by importing vim module by default.
+        exec('import vim', self.module.__dict__)
         # Handle DirChanged. #296
         nvim.command(
             'au DirChanged * call rpcnotify({}, "python_chdir", v:event.cwd)'

test/test_host.py

@@ -1,6 +1,13 @@
 # -*- coding: utf-8 -*-
-
 from pynvim.plugin.host import Host, host_method_spec
+from pynvim.plugin.script_host import ScriptHost
+
+
+def test_host_imports(vim):
+    h = ScriptHost(vim)
+    assert h.module.__dict__['vim']
+    assert h.module.__dict__['vim'] == h.legacy_vim
+    assert h.module.__dict__['sys']
 
 
 def test_host_clientinfo(vim):