From 2cbc13553b113386d23ef225344e61172763ee45 Mon Sep 17 00:00:00 2001 From: Sunny Rahul Date: Fri, 7 Feb 2020 09:41:39 +0100 Subject: [PATCH] import vim module by default #430 Vim-compatible. fixes neovim/neovim#11783 --- pynvim/plugin/script_host.py | 3 ++- test/test_host.py | 9 ++++++++- 2 files changed, 10 insertions(+), 2 deletions(-) diff --git a/pynvim/plugin/script_host.py b/pynvim/plugin/script_host.py index 7f47843a..8def687e 100644 --- a/pynvim/plugin/script_host.py +++ b/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)' diff --git a/test/test_host.py b/test/test_host.py index 83b0908f..b21a398b 100644 --- a/test/test_host.py +++ b/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):