aboutsummaryrefslogtreecommitdiff
path: root/tests/functional/o/overloaded_operator.py
blob: c305af689a0359084dc0b9c8758f7f0b6a2f16ea (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 wan't an error for astype access