summaryrefslogtreecommitdiff
path: root/tests/functional/o/overloaded_operator.py
blob: c14fb651f178c6eb82acead1d82092cfb07212a5 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
# pylint: disable=missing-docstring,too-few-public-methods,useless-object-inheritance
"""#3291"""
from __future__ import print_function

class Myarray(object):
    def __init__(self, array):
        self.array = array

    def __mul__(self, val):
        return Myarray(val)

    def astype(self):
        return "ASTYPE", self

def randint(maximum):
    if maximum is not None:
        return Myarray([1, 2, 3]) * 2

    return int(5)

print(randint(1).astype()) # we don't want an error for astype access