psycopg/psycopg2
1
1
Fork 0
mirror of https://github.com/psycopg/psycopg2.git synced 2025-08-02 19:30:09 +03:00
Code Issues Packages Projects Releases Wiki Activity

Merge 5d8c22cf9d into 153b0983c1

Browse Source
This commit is contained in:
Jon Dufresne 2018-07-25 15:33:52 +00:00 committed by GitHub
commit e81d073e87
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 18 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
lib/_range.py
View File

@ -105,6 +105,10 @@ class Range(object):
return False return False
return self._bounds[1] == ']' return self._bounds[1] == ']'
def __getitem__(self, key):
t = self._lower, self._upper
return t[key]
def __contains__(self, x): def __contains__(self, x):
if self._bounds is None: if self._bounds is None:
return False return False

14
tests/test_types_extras.py
View File

@ -1386,6 +1386,20 @@ class RangeTestCase(unittest.TestCase):
r = Range(0, 4) r = Range(0, 4)
self.assertEqual(loads(dumps(r)), r) self.assertEqual(loads(dumps(r)), r)
def test_getitem(self):
from psycopg2.extras import Range
r = Range(0, 4)
self.assertEqual(tuple(r), (0, 4))
self.assertEqual(list(r), [0, 4])
lower, upper = r
self.assertEqual(lower, 0)
self.assertEqual(upper, 4)
self.assertEqual(r[0], 0)
self.assertEqual(r[1], 4)
# String indexes are not allowed.
with self.assertRaises(TypeError):
r['abc']
def skip_if_no_range(f): def skip_if_no_range(f):
@wraps(f) @wraps(f)