From a1748131132ebc83f443b2a8ddc5073bfefa504f Mon Sep 17 00:00:00 2001 From: Armin Rigo Date: Thu, 23 Dec 2004 22:13:13 +0000 Subject: Dima Dorfman's patch for coercion/comparison of C types (patch #995939), with a minor change after the coercion, to accept two objects not necessarily of the same type but with the same tp_compare. --- Lib/test/test_coercion.py | 36 ++++++++++++++++++++++++++++++++++++ 1 file changed, 36 insertions(+) (limited to 'Lib/test/test_coercion.py') diff --git a/Lib/test/test_coercion.py b/Lib/test/test_coercion.py index 331771f672..37d735b3ba 100644 --- a/Lib/test/test_coercion.py +++ b/Lib/test/test_coercion.py @@ -125,9 +125,45 @@ def do_prefix_binops(): else: print '=', format_result(x) +# New-style class version of CoerceNumber +class CoerceTo(object): + def __init__(self, arg): + self.arg = arg + def __coerce__(self, other): + if isinstance(other, CoerceTo): + return self.arg, other.arg + else: + return self.arg, other + +def assert_(expr, msg=None): + if not expr: + raise AssertionError, msg + +def do_cmptypes(): + # Built-in tp_compare slots expect their arguments to have the + # same type, but a user-defined __coerce__ doesn't have to obey. + # SF #980352 + evil_coercer = CoerceTo(42) + # Make sure these don't crash any more + assert_(cmp(u'fish', evil_coercer) != 0) + assert_(cmp(slice(1), evil_coercer) != 0) + # ...but that this still works + class WackyComparer(object): + def __cmp__(self, other): + assert_(other == 42, 'expected evil_coercer, got %r' % other) + return 0 + assert_(cmp(WackyComparer(), evil_coercer) == 0) + # ...and classic classes too, since that code path is a little different + class ClassicWackyComparer: + def __cmp__(self, other): + assert_(other == 42, 'expected evil_coercer, got %r' % other) + return 0 + assert_(cmp(ClassicWackyComparer(), evil_coercer) == 0) + warnings.filterwarnings("ignore", r'complex divmod\(\), // and % are deprecated', DeprecationWarning, r'test.test_coercion$') do_infix_binops() do_prefix_binops() +do_cmptypes() -- cgit v1.2.1