test changes

thomasballinger · thomasballinger · commit 2bd4a10d6176 · 2014-07-06T14:30:43.000-04:00
diff --git a/bpython/autocomplete.py b/bpython/autocomplete.py
@@ -60,7 +60,10 @@
 def get_completer(cursor_offset, current_line, locals_, argspec, full_code, mode, complete_magic_methods):
     """Returns a list of matches and a class for what kind of completion is happening
 
-    If no completion type is relevant, returns None, None"""
+    If no completion type is relevant, returns None, None
+
+    argspec is an output of inspect.getargspec
+    """
 
     kwargs = {'locals_':locals_, 'argspec':argspec, 'full_code':full_code,
               'mode':mode, 'complete_magic_methods':complete_magic_methods}
diff --git a/bpython/test/test_autocomplete.py b/bpython/test/test_autocomplete.py
@@ -1,15 +1,16 @@
 from bpython import autocomplete
 from functools import partial
+import inspect
 
 import unittest
 
 class TestSimpleComplete(unittest.TestCase):
 
-    simple_config = type('', (), {})()
-    simple_config.autocomplete_mode = autocomplete.SIMPLE
     complete = partial(autocomplete.complete,
                        namespace={'zabcdef':1, 'zabcqwe':2, 'ze':3},
                        config=simple_config)
+    kwargs = {locals_={'zabcdef':1, 'zabcqwe':2, 'ze':3},
+              argspec=inspect.getargspec(lambda x: x),
 
     def test_simple_completion(self):
         self.assertEqual(self.complete('zab'), ['zabcdef', 'zabcqwe'])
