psycopg/psycopg2
1
1
Fork 0
mirror of https://github.com/psycopg/psycopg2.git synced 2025-02-18 01:50:32 +03:00
Code Issues Packages Projects Releases Wiki Activity

cater for comparison of subclasses

Browse Source
This commit is contained in:
Chris Withers 2013-05-26 22:20:07 +01:00
parent b6a9e0ffaf
commit 28276e3eb1
2 changed files with 10 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
lib/_range.py
View File

@ -121,7 +121,7 @@ class Range(object):
return self._bounds is not None return self._bounds is not None
def __eq__(self, other): def __eq__(self, other):
if not isinstance(other, self.__class__): if not isinstance(other, Range):
return False return False
return (self._lower == other._lower return (self._lower == other._lower
and self._upper == other._upper and self._upper == other._upper

9
tests/test_types_extras.py
View File

@ -1216,6 +1216,15 @@ class RangeTestCase(unittest.TestCase):
from psycopg2.extras import Range from psycopg2.extras import Range
self.assertFalse(Range(10, 20)==()) self.assertFalse(Range(10, 20)==())
def test_eq_subclass(self):
from psycopg2.extras import Range, NumericRange
class IntRange(NumericRange): pass
class PositiveIntRange(IntRange): pass
self.assertTrue(Range(10, 20)==IntRange(10, 20))
self.assertTrue(PositiveIntRange(10, 20)==IntRange(10, 20))
def test_not_ordered(self): def test_not_ordered(self):
from psycopg2.extras import Range from psycopg2.extras import Range
self.assertRaises(TypeError, lambda: Range(empty=True) < Range(0,4)) self.assertRaises(TypeError, lambda: Range(empty=True) < Range(0,4))