summaryrefslogtreecommitdiffstatshomepage
path: root/tests/basics/subclass_classmethod.py
blob: 3089726aa9bf5e113ee480570783407fc542b7b1 (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
# Calling an inherited classmethod
class Base:

    @classmethod
    def foo(cls):
        print(cls.__name__)

try:
    Base.__name__
except AttributeError:
    print("SKIP")
    raise SystemExit

class Sub(Base):
    pass


Sub.foo()

# overriding a member and accessing it via a classmethod

class A(object):
    foo = 0

    @classmethod
    def bar(cls):
        print(cls.foo)

    def baz(self):
        print(self.foo)

class B(A):
    foo = 1

B.bar() # class calling classmethod
B().bar() # instance calling classmethod
B().baz() # instance calling normal method

# super inside a classmethod
# ensure the argument of the super method that is called is the child type


class C:
    @classmethod
    def f(cls):
        print("C.f", cls.__name__)  # cls should be D

    @classmethod
    def g(cls):
        print("C.g", cls.__name__)  # cls should be D


class D(C):
    @classmethod
    def f(cls):
        print("D.f", cls.__name__)
        super().f()

    @classmethod
    def g(cls):
        print("D.g", cls.__name__)
        super(D, cls).g()


D.f()
D.g()
D().f()
D().g()