Commit 628ffd74 authored by Stefan Behnel's avatar Stefan Behnel

Fix the first argument special method signatures of "__eq__", "__lt__" and the...

Fix the first argument special method signatures of "__eq__", "__lt__" and the other richcmp methods to be of "self" type, in accordance with to the CPython specs.
See https://docs.python.org/3/reference/datamodel.html#emulating-container-types
Closes #1935.
parent 4c461c93
...@@ -576,7 +576,7 @@ def get_special_method_signature(name): ...@@ -576,7 +576,7 @@ def get_special_method_signature(name):
if slot: if slot:
return slot.signature return slot.signature
elif name in richcmp_special_methods: elif name in richcmp_special_methods:
return binaryfunc return ibinaryfunc
else: else:
return None return None
......
...@@ -209,21 +209,21 @@ Rich comparison operators ...@@ -209,21 +209,21 @@ Rich comparison operators
https://docs.python.org/3/reference/datamodel.html#basic-customization https://docs.python.org/3/reference/datamodel.html#basic-customization
+-----------------------+---------------------------------------+-------------+-----------------------------------------------------+ +-----------------------+---------------------------------------+-------------+--------------------------------------------------------+
| __richcmp__ |x, y, int op | object | Rich comparison (no direct Python equivalent) | | __richcmp__ |x, y, int op | object | Rich comparison (no direct Python equivalent) |
+-----------------------+---------------------------------------+-------------+-----------------------------------------------------+ +-----------------------+---------------------------------------+-------------+--------------------------------------------------------+
| __eq__ |x, y | object | x == y | | __eq__ |self, y | object | self == y |
+-----------------------+---------------------------------------+-------------+-----------------------------------------------------+ +-----------------------+---------------------------------------+-------------+--------------------------------------------------------+
| __ne__ |x, y | object | x != y (falls back to ``__eq__`` if not available) | | __ne__ |self, y | object | self != y (falls back to ``__eq__`` if not available) |
+-----------------------+---------------------------------------+-------------+-----------------------------------------------------+ +-----------------------+---------------------------------------+-------------+--------------------------------------------------------+
| __lt__ |x, y | object | x < y | | __lt__ |self, y | object | self < y |
+-----------------------+---------------------------------------+-------------+-----------------------------------------------------+ +-----------------------+---------------------------------------+-------------+--------------------------------------------------------+
| __gt__ |x, y | object | x > y | | __gt__ |self, y | object | self > y |
+-----------------------+---------------------------------------+-------------+-----------------------------------------------------+ +-----------------------+---------------------------------------+-------------+--------------------------------------------------------+
| __le__ |x, y | object | x <= y | | __le__ |self, y | object | self <= y |
+-----------------------+---------------------------------------+-------------+-----------------------------------------------------+ +-----------------------+---------------------------------------+-------------+--------------------------------------------------------+
| __ge__ |x, y | object | x >= y | | __ge__ |self, y | object | self >= y |
+-----------------------+---------------------------------------+-------------+-----------------------------------------------------+ +-----------------------+---------------------------------------+-------------+--------------------------------------------------------+
Arithmetic operators Arithmetic operators
^^^^^^^^^^^^^^^^^^^^ ^^^^^^^^^^^^^^^^^^^^
......
...@@ -9,7 +9,7 @@ IS_PY2 = sys.version_info[0] == 2 ...@@ -9,7 +9,7 @@ IS_PY2 = sys.version_info[0] == 2
@cython.cclass @cython.cclass
class X(object): class X(object):
x = cython.declare(cython.int, visibility="public") x = cython.declare(cython.int)
def __init__(self, x): def __init__(self, x):
self.x = x self.x = x
...@@ -18,6 +18,12 @@ class X(object): ...@@ -18,6 +18,12 @@ class X(object):
return "<%d>" % self.x return "<%d>" % self.x
@cython.cfunc
@cython.locals(x=X)
def x_of(x):
return x.x
@cython.cclass @cython.cclass
class ClassEq(X): class ClassEq(X):
""" """
...@@ -74,9 +80,12 @@ class ClassEq(X): ...@@ -74,9 +80,12 @@ class ClassEq(X):
TypeError... TypeError...
""" """
def __eq__(self, other): def __eq__(self, other):
if isinstance(self, X): assert 1 <= self.x <= 2
assert isinstance(self, ClassEq), type(self)
if isinstance(other, X): if isinstance(other, X):
return self.x == other.x return self.x == x_of(other)
elif isinstance(other, int):
return self.x < other
return NotImplemented return NotImplemented
...@@ -134,9 +143,12 @@ class ClassEqNe(ClassEq): ...@@ -134,9 +143,12 @@ class ClassEqNe(ClassEq):
TypeError... TypeError...
""" """
def __ne__(self, other): def __ne__(self, other):
if isinstance(self, X): assert 1 <= self.x <= 2
assert isinstance(self, ClassEqNe), type(self)
if isinstance(other, X): if isinstance(other, X):
return self.x != other.x return self.x != x_of(other)
elif isinstance(other, int):
return self.x < other
return NotImplemented return NotImplemented
...@@ -208,11 +220,34 @@ class ClassEqNeGe(ClassEqNe): ...@@ -208,11 +220,34 @@ class ClassEqNeGe(ClassEqNe):
... else: a > b ... else: a > b
Traceback (most recent call last): Traceback (most recent call last):
TypeError... TypeError...
>>> 2 <= a
False
>>> a >= 2
False
>>> 1 <= a
True
>>> a >= 1
True
>>> a >= 2
False
>>> if IS_PY2: raise TypeError # doctest: +ELLIPSIS
... else: 'x' <= a
Traceback (most recent call last):
TypeError...
>>> if IS_PY2: raise TypeError # doctest: +ELLIPSIS
... else: a >= 'x'
Traceback (most recent call last):
TypeError...
""" """
def __ge__(self, other): def __ge__(self, other):
if isinstance(self, X): assert 1 <= self.x <= 2
assert isinstance(self, ClassEqNeGe), type(self)
if isinstance(other, X): if isinstance(other, X):
return self.x >= other.x return self.x >= x_of(other)
elif isinstance(other, int):
return self.x >= other
return NotImplemented return NotImplemented
...@@ -274,11 +309,34 @@ class ClassLe(X): ...@@ -274,11 +309,34 @@ class ClassLe(X):
True True
>>> b >= c >>> b >= c
True True
>>> 2 >= a
True
>>> a <= 2
True
>>> 1 >= a
True
>>> a <= 1
True
>>> a <= 0
False
>>> if IS_PY2: raise TypeError # doctest: +ELLIPSIS
... else: 'x' >= a
Traceback (most recent call last):
TypeError...
>>> if IS_PY2: raise TypeError # doctest: +ELLIPSIS
... else: a <= 'x'
Traceback (most recent call last):
TypeError...
""" """
def __le__(self, other): def __le__(self, other):
if isinstance(self, X): assert 1 <= self.x <= 2
assert isinstance(self, ClassLe), type(self)
if isinstance(other, X): if isinstance(other, X):
return self.x <= other.x return self.x <= x_of(other)
elif isinstance(other, int):
return self.x <= other
return NotImplemented return NotImplemented
...@@ -320,11 +378,37 @@ class ClassLt(X): ...@@ -320,11 +378,37 @@ class ClassLt(X):
[<1>, <1>, <2>] [<1>, <1>, <2>]
>>> sorted([b, a, c]) >>> sorted([b, a, c])
[<1>, <1>, <2>] [<1>, <1>, <2>]
>>> 2 > a
True
>>> a < 2
True
>>> 1 > a
False
>>> a < 1
False
>>> if IS_PY2: raise TypeError # doctest: +ELLIPSIS
... else: 1 < a
Traceback (most recent call last):
TypeError...
>>> if IS_PY2: raise TypeError # doctest: +ELLIPSIS
... else: 'x' > a
Traceback (most recent call last):
TypeError...
>>> if IS_PY2: raise TypeError # doctest: +ELLIPSIS
... else: a < 'x'
Traceback (most recent call last):
TypeError...
""" """
def __lt__(self, other): def __lt__(self, other):
if isinstance(self, X): assert 1 <= self.x <= 2
assert isinstance(self, ClassLt), type(self)
if isinstance(other, X): if isinstance(other, X):
return self.x < other.x return self.x < x_of(other)
elif isinstance(other, int):
return self.x < other
return NotImplemented return NotImplemented
...@@ -368,9 +452,12 @@ class ClassLtGtInherited(X): ...@@ -368,9 +452,12 @@ class ClassLtGtInherited(X):
[<1>, <1>, <2>] [<1>, <1>, <2>]
""" """
def __gt__(self, other): def __gt__(self, other):
if isinstance(self, X): assert 1 <= self.x <= 2
assert isinstance(self, ClassLtGtInherited), type(self)
if isinstance(other, X): if isinstance(other, X):
return self.x > other.x return self.x > x_of(other)
elif isinstance(other, int):
return self.x > other
return NotImplemented return NotImplemented
...@@ -412,17 +499,49 @@ class ClassLtGt(X): ...@@ -412,17 +499,49 @@ class ClassLtGt(X):
[<1>, <1>, <2>] [<1>, <1>, <2>]
>>> sorted([b, a, c]) >>> sorted([b, a, c])
[<1>, <1>, <2>] [<1>, <1>, <2>]
>>> 2 > a
True
>>> 2 < a
False
>>> a < 2
True
>>> a > 2
False
>>> if IS_PY2: raise TypeError # doctest: +ELLIPSIS
... else: 'x' > a
Traceback (most recent call last):
TypeError...
>>> if IS_PY2: raise TypeError # doctest: +ELLIPSIS
... else: 'x' < a
Traceback (most recent call last):
TypeError...
>>> if IS_PY2: raise TypeError # doctest: +ELLIPSIS
... else: a < 'x'
Traceback (most recent call last):
TypeError...
>>> if IS_PY2: raise TypeError # doctest: +ELLIPSIS
... else: a > 'x'
Traceback (most recent call last):
TypeError...
""" """
def __lt__(self, other): def __lt__(self, other):
if isinstance(self, X): assert 1 <= self.x <= 2
assert isinstance(self, ClassLtGt), type(self)
if isinstance(other, X): if isinstance(other, X):
return self.x < other.x return self.x < x_of(other)
elif isinstance(other, int):
return self.x < other
return NotImplemented return NotImplemented
def __gt__(self, other): def __gt__(self, other):
if isinstance(self, X): assert 1 <= self.x <= 2
assert isinstance(self, ClassLtGt), type(self)
if isinstance(other, X): if isinstance(other, X):
return self.x > other.x return self.x > x_of(other)
elif isinstance(other, int):
return self.x > other
return NotImplemented return NotImplemented
......
Markdown is supported
0%
or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment