usse/funda-scraper/venv/lib/python3.10/site-packages/mypyc/test-data/irbuild-dunders.test

188 lines
3.1 KiB
Plaintext

# Test cases for (some) dunder methods
[case testDundersLen]
class C:
def __len__(self) -> int:
return 2
def f(c: C) -> int:
return len(c)
[out]
def C.__len__(self):
self :: __main__.C
L0:
return 4
def f(c):
c :: __main__.C
r0 :: int
r1 :: native_int
r2, r3, r4 :: bit
r5 :: bool
L0:
r0 = c.__len__()
r1 = r0 & 1
r2 = r1 != 0
if r2 goto L1 else goto L2 :: bool
L1:
r3 = CPyTagged_IsLt_(r0, 0)
if r3 goto L3 else goto L4 :: bool
L2:
r4 = r0 >= 0 :: signed
if r4 goto L4 else goto L3 :: bool
L3:
r5 = raise ValueError('__len__() should return >= 0')
unreachable
L4:
return r0
[case testDundersSetItem]
class C:
def __setitem__(self, key: int, value: int) -> None:
pass
def f(c: C) -> None:
c[3] = 4
[out]
def C.__setitem__(self, key, value):
self :: __main__.C
key, value :: int
L0:
return 1
def f(c):
c :: __main__.C
r0 :: None
L0:
r0 = c.__setitem__(6, 8)
return 1
[case testDundersContains]
from typing import Any
class C:
def __contains__(self, x: int) -> bool:
return False
def f(c: C) -> bool:
return 7 in c
def g(c: C) -> bool:
return 7 not in c
class D:
def __contains__(self, x: int) -> Any:
return 'x'
def h(d: D) -> bool:
return 7 not in d
[out]
def C.__contains__(self, x):
self :: __main__.C
x :: int
L0:
return 0
def f(c):
c :: __main__.C
r0 :: bool
L0:
r0 = c.__contains__(14)
return r0
def g(c):
c :: __main__.C
r0, r1 :: bool
L0:
r0 = c.__contains__(14)
r1 = r0 ^ 1
return r1
def D.__contains__(self, x):
self :: __main__.D
x :: int
r0 :: str
L0:
r0 = 'x'
return r0
def h(d):
d :: __main__.D
r0 :: object
r1 :: int32
r2 :: bit
r3, r4 :: bool
L0:
r0 = d.__contains__(14)
r1 = PyObject_IsTrue(r0)
r2 = r1 >= 0 :: signed
r3 = truncate r1: int32 to builtins.bool
r4 = r3 ^ 1
return r4
[case testDundersDelItem]
class C:
def __delitem__(self, x: int) -> None:
pass
def f(c: C) -> None:
del c[5]
[out]
def C.__delitem__(self, x):
self :: __main__.C
x :: int
L0:
return 1
def f(c):
c :: __main__.C
r0 :: None
L0:
r0 = c.__delitem__(10)
return 1
[case testDundersUnary]
class C:
def __neg__(self) -> int:
return 1
def __invert__(self) -> int:
return 2
def __int__(self) -> int:
return 3
def __float__(self) -> float:
return 4.0
def f(c: C) -> None:
-c
~c
int(c)
float(c)
[out]
def C.__neg__(self):
self :: __main__.C
L0:
return 2
def C.__invert__(self):
self :: __main__.C
L0:
return 4
def C.__int__(self):
self :: __main__.C
L0:
return 6
def C.__float__(self):
self :: __main__.C
r0 :: float
L0:
r0 = 4.0
return r0
def f(c):
c :: __main__.C
r0, r1 :: int
r2, r3, r4, r5 :: object
L0:
r0 = c.__neg__()
r1 = c.__invert__()
r2 = load_address PyLong_Type
r3 = PyObject_CallFunctionObjArgs(r2, c, 0)
r4 = load_address PyFloat_Type
r5 = PyObject_CallFunctionObjArgs(r4, c, 0)
return 1