Test attribute completion on other expressions

thomasballinger · thomasballinger · commit b0951cd2a70a · 2015-12-14T17:39:19.000-05:00
diff --git a/bpython/test/test_autocomplete.py b/bpython/test/test_autocomplete.py
@@ -309,6 +309,16 @@ def __getitem__(inner_self, i):
 
         self.com.matches(5, 'a[0].', locals_={'a': FakeList()})
 
+    def test_literals_complete(self):
+        self.assertSetEqual(self.com.matches(10, '[a][0][0].',
+                            locals_={'a': (Foo(),)}),
+                            set(['method', 'a', 'b']))
+
+    def test_dictionaries_complete(self):
+        self.assertSetEqual(self.com.matches(7, 'a["b"].',
+                            locals_={'a': {'b': Foo()}}),
+                            set(['method', 'a', 'b']))
+
 
 class TestMagicMethodCompletion(unittest.TestCase):
 
