From 734775524e14a8ae3997933afba64a4ac6a3cd47 Mon Sep 17 00:00:00 2001 From: Damien George Date: Mon, 27 Mar 2017 11:29:11 +1100 Subject: [PATCH] tests/basics: Add test for super() when self is closed over. --- tests/basics/class_super_closure.py | 18 ++++++++++++++++++ 1 file changed, 18 insertions(+) create mode 100644 tests/basics/class_super_closure.py diff --git a/tests/basics/class_super_closure.py b/tests/basics/class_super_closure.py new file mode 100644 index 0000000000..41acae90de --- /dev/null +++ b/tests/basics/class_super_closure.py @@ -0,0 +1,18 @@ +# test that no-arg super() works when self is closed over + +class A: + def __init__(self): + self.val = 4 + def foo(self): + # we access a member of self to check that self is correct + return list(range(self.val)) +class B(A): + def foo(self): + # self is closed over because it's referenced in the list comprehension + # and then super() must detect this and load from the closure cell + return [self.bar(i) for i in super().foo()] + def bar(self, x): + return 2 * x + +print(A().foo()) +print(B().foo())