mirror of
https://github.com/micropython/micropython.git
synced 2025-09-06 01:40:34 +02:00
4b57330465
So that user types can implement reverse operators and have them work with str on the left-hand-side, eg `"a" + UserType()`. Signed-off-by: Damien George <damien@micropython.org>
49 lines
876 B
Python
49 lines
876 B
Python
# Test reverse operators.
|
|
|
|
# Test user type with integers.
|
|
class A:
|
|
def __init__(self, v):
|
|
self.v = v
|
|
|
|
def __add__(self, o):
|
|
if isinstance(o, A):
|
|
return A(self.v + o.v)
|
|
return A(self.v + o)
|
|
|
|
def __radd__(self, o):
|
|
return A(self.v + o)
|
|
|
|
def __repr__(self):
|
|
return "A({})".format(self.v)
|
|
|
|
|
|
print(A(3) + 1)
|
|
print(2 + A(5))
|
|
|
|
|
|
# Test user type with strings.
|
|
class B:
|
|
def __init__(self, v):
|
|
self.v = v
|
|
|
|
def __repr__(self):
|
|
return "B({})".format(self.v)
|
|
|
|
def __ror__(self, o):
|
|
return B(o + "|" + self.v)
|
|
|
|
def __radd__(self, o):
|
|
return B(o + "+" + self.v)
|
|
|
|
def __rmul__(self, o):
|
|
return B(o + "*" + self.v)
|
|
|
|
def __rtruediv__(self, o):
|
|
return B(o + "/" + self.v)
|
|
|
|
|
|
print("a" | B("b"))
|
|
print("a" + B("b"))
|
|
print("a" * B("b"))
|
|
print("a" / B("b"))
|