@@ -925,6 +925,7 @@ def tearDown(self):
925925 def prepare_reader (self , events , namespace ):
926926 console = FakeConsole (events )
927927 config = ReadlineConfig ()
928+ config .module_completer = ModuleCompleter (namespace )
928929 config .readline_completer = rlcompleter .Completer (namespace ).complete
929930 reader = ReadlineAlikeReader (console = console , config = config )
930931 return reader
@@ -961,13 +962,15 @@ def test_import_completions(self):
961962
962963 def test_relative_import_completions (self ):
963964 cases = (
964- ("from .readl\t \n " , "from .readline" ),
965- ("from . import readl\t \n " , "from . import readline" ),
965+ (None , "from .readl\t \n " , "from .readl" ),
966+ (None , "from . import readl\t \n " , "from . import readl" ),
967+ ("_pyrepl" , "from .readl\t \n " , "from .readline" ),
968+ ("_pyrepl" , "from . import readl\t \n " , "from . import readline" ),
966969 )
967- for code , expected in cases :
970+ for package , code , expected in cases :
968971 with self .subTest (code = code ):
969972 events = code_to_events (code )
970- reader = self .prepare_reader (events , namespace = {})
973+ reader = self .prepare_reader (events , namespace = {"__package__" : package })
971974 output = reader .readline ()
972975 self .assertEqual (output , expected )
973976
0 commit comments