Skip to content

Commit 3ebd1b1

Browse files
committed
Add and fix the test cases for the module completer
1 parent 505e058 commit 3ebd1b1

1 file changed

Lines changed: 7 additions & 4 deletions

File tree

Lib/test/test_pyrepl/test_pyrepl.py

Lines changed: 7 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -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

Comments
 (0)