Test attribute completion

thomasballinger · thomasballinger · commit 9579c0619ee9 · 2015-01-24T00:30:33.000-05:00
diff --git a/bpython/test/test_autocomplete.py b/bpython/test/test_autocomplete.py
@@ -175,3 +175,16 @@ def test_obj_that_does_not_allow_conversion_to_bool(self):
         com = autocomplete.DictKeyCompletion()
         local={'mNumPy':MockNumPy(),}
         self.assertSetEqual(com.matches(7,"mNumPy[" , local), set())
+
+class Foo(object):
+    a = 10
+    def __init__(self):
+        self.b = 20
+    def method(self, x):
+        pass
+
+class TestAttrCompletion(unittest.TestCase):
+
+    def test_att_matches_found_on_instance(self):
+        com = autocomplete.AttrCompletion()
+        self.assertSetEqual(com.matches(2, 'a.', {'a': Foo()}), set(['a.method', 'a.a', 'a.b']))
