Fix some old regression tests.

2016-12-07 01:30:30 +01:00
parent 97ccb74ebb
commit 2be5da3f85
3 changed files with 9 additions and 4 deletions
--- a/test/test_regression.py
+++ b/test/test_regression.py
@@ -174,7 +174,8 @@ class TestRegression(TestCase):
        for i in range(2):
            completions = Script('').completions()
            c = get_str_completion(completions)
-            n = len(c._definition.subscopes[0].children[-1].children)
+            str_context, = c._name.infer()
+            n = len(str_context.classdef.children[-1].children)
            if i == 0:
                limit = n
            else: