Commit 3f7eb13
committed
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
@@ -15,19 +15,45 @@ def setUp(self): | |||
| 15 | 15 | "zzefg", | |
| 16 | 16 | "zzabc.e", | |
| 17 | 17 | "zzabc.f", | |
| 18 | + "zzefg.a1", | ||
| 19 | + "zzefg.a2" | ||
| 18 | 20 | ] | |
| 19 | 21 | ||
| 20 | 22 | def test_simple_completion(self): | |
| 21 | 23 | self.assertSetEqual( | |
| 22 | 24 | self.module_gatherer.complete(10, "import zza"), {"zzabc", "zzabd"} | |
| 23 | 25 | ) | |
| 24 | 26 | ||
| 25 | - def test_package_completion(self): | ||
| 27 | + def test_import_empty(self): | ||
| 26 | 28 | self.assertSetEqual( | |
| 27 | 29 | self.module_gatherer.complete(13, "import zzabc."), | |
| 28 | 30 | {"zzabc.e", "zzabc.f"}, | |
| 29 | 31 | ) | |
| 30 | 32 | ||
| 33 | + def test_import(self): | ||
| 34 | + self.assertSetEqual( | ||
| 35 | + self.module_gatherer.complete(14, "import zzefg.a"), | ||
| 36 | + {"zzefg.a1", "zzefg.a2"}, | ||
| 37 | + ) | ||
| 38 | + | ||
| 39 | + | ||
| 40 | + @unittest.expectedFailure | ||
| 41 | + def test_import_empty(self): | ||
| 42 | + self.assertSetEqual( | ||
| 43 | + self.module_gatherer.complete(7, "import "), {"zzabc", "zzabd", "zzefg"} | ||
| 44 | + ) | ||
| 45 | + | ||
| 46 | + @unittest.expectedFailure | ||
| 47 | + def test_from_import_empty(self): | ||
| 48 | + self.assertSetEqual( | ||
| 49 | + self.module_gatherer.complete(18, "from zzabc import "), {"e", "f"} | ||
| 50 | + ) | ||
| 51 | + | ||
| 52 | + def test_from_import(self): | ||
| 53 | + self.assertSetEqual( | ||
| 54 | + self.module_gatherer.complete(19, "from zzefg import a"), {"a1", "a2"} | ||
| 55 | + ) | ||
| 56 | + | ||
| 31 | 57 | ||
| 32 | 58 | class TestRealComplete(unittest.TestCase): | |
| 33 | 59 | def setUp(self): | |
@@ -52,6 +78,11 @@ def test_from_package(self): | |||
| 52 | 78 | self.module_gatherer.complete(17, "from xml import d"), {"dom"} | |
| 53 | 79 | ) | |
| 54 | 80 | ||
| 81 | + def test_from_package(self): | ||
| 82 | + self.assertSetEqual( | ||
| 83 | + self.module_gatherer.complete(17, "from xml import d"), {"dom"} | ||
| 84 | + ) | ||
| 85 | + | ||
| 55 | 86 | ||
| 56 | 87 | class TestAvoidSymbolicLinks(unittest.TestCase): | |
| 57 | 88 | def setUp(self): | |
0 commit comments