Add py__class__ to ContextSet.

jedi/evaluate/context.py

@@ -207,7 +207,8 @@ class ContextualizedName(ContextualizedNode):
 
 
 class ContextSet(BaseContextSet):
-    pass
+    def py__class__(self):
+        return ContextSet.from_iterable(c.py__class__() for c in self._set)
 
 
 NO_CONTEXTS = ContextSet()

jedi/evaluate/stdlib.py

@@ -170,7 +170,7 @@ def builtins_type(evaluator, objects, bases, dicts):
         # It's a type creation... maybe someday...
         return NO_CONTEXTS
     else:
-        return ContextSet.from_iterable(o.py__class__() for o in objects)
+        return objects.py__class__()
 
 
 class SuperInstance(AbstractInstanceContext):