Extend importcompletion tests

sebastinas · sebastinas · commit 337173156c88 · 2021-01-03T00:12:49.000+01:00
diff --git a/bpython/test/test_importcompletion.py b/bpython/test/test_importcompletion.py
@@ -23,37 +23,82 @@ def test_simple_completion(self):
         self.assertSetEqual(
             self.module_gatherer.complete(10, "import zza"), {"zzabc", "zzabd"}
         )
+        self.assertSetEqual(
+            self.module_gatherer.complete(11, "import  zza"), {"zzabc", "zzabd"}
+        )
 
     def test_import_empty(self):
         self.assertSetEqual(
             self.module_gatherer.complete(13, "import zzabc."),
             {"zzabc.e", "zzabc.f"},
         )
+        self.assertSetEqual(
+            self.module_gatherer.complete(14, "import  zzabc."),
+            {"zzabc.e", "zzabc.f"},
+        )
 
     def test_import(self):
         self.assertSetEqual(
             self.module_gatherer.complete(14, "import zzefg.a"),
             {"zzefg.a1", "zzefg.a2"},
         )
+        self.assertSetEqual(
+            self.module_gatherer.complete(15, "import  zzefg.a"),
+            {"zzefg.a1", "zzefg.a2"},
+        )
 
     @unittest.expectedFailure
     def test_import_empty(self):
         self.assertSetEqual(
             self.module_gatherer.complete(7, "import "),
             {"zzabc", "zzabd", "zzefg"},
         )
+        self.assertSetEqual(
+            self.module_gatherer.complete(8, "import  "),
+            {"zzabc", "zzabd", "zzefg"},
+        )
 
     @unittest.expectedFailure
     def test_from_import_empty(self):
+        self.assertSetEqual(
+            self.module_gatherer.complete(5, "from "), {"zzabc", "zzabd", "zzefg"}
+        )
+        self.assertSetEqual(
+            self.module_gatherer.complete(6, "from  "), {"zzabc", "zzabd", "zzefg"}
+        )
+
+    @unittest.expectedFailure
+    def test_from_module_import_empty(self):
         self.assertSetEqual(
             self.module_gatherer.complete(18, "from zzabc import "), {"e", "f"}
         )
+        self.assertSetEqual(
+            self.module_gatherer.complete(19, "from  zzabc import "), {"e", "f"}
+        )
+        self.assertSetEqual(
+            self.module_gatherer.complete(19, "from zzabc  import "), {"e", "f"}
+        )
+        self.assertSetEqual(
+            self.module_gatherer.complete(19, "from zzabc import  "), {"e", "f"}
+        )
 
-    def test_from_import(self):
+    def test_from_module_import(self):
         self.assertSetEqual(
             self.module_gatherer.complete(19, "from zzefg import a"),
             {"a1", "a2"},
         )
+        self.assertSetEqual(
+            self.module_gatherer.complete(20, "from  zzefg import a"),
+            {"a1", "a2"},
+        )
+        self.assertSetEqual(
+            self.module_gatherer.complete(20, "from zzefg  import a"),
+            {"a1", "a2"},
+        )
+        self.assertSetEqual(
+            self.module_gatherer.complete(20, "from zzefg import  a"),
+            {"a1", "a2"},
+        )
 
 
 class TestRealComplete(unittest.TestCase):
@@ -150,3 +195,7 @@ def test_simple_symbolic_link_loop(self):
                 filepaths.remove("Left.toRight")
             filepaths.remove(thing)
         self.assertFalse(filepaths)
+
+
+if __name__ == "__main__":
+    unittest.main()
