mirror of
https://github.com/kennethreitz/records.git
synced 2026-06-05 14:50:18 +00:00
Chad Whitacre
100 lines
2.9 KiB
Python
100 lines
2.9 KiB
Python
from collections import namedtuple
|
|
|
|
import records
|
|
|
|
from pytest import raises
|
|
|
|
|
|
IdRecord = namedtuple('IdRecord', 'id')
|
|
|
|
|
|
def check_id(i, row):
|
|
assert row.id == i
|
|
|
|
|
|
class TestRecordCollection:
|
|
def test_iter(self):
|
|
rows = records.RecordCollection(IdRecord(i) for i in range(10))
|
|
for i, row in enumerate(rows):
|
|
check_id(i, row)
|
|
|
|
def test_next(self):
|
|
rows = records.RecordCollection(IdRecord(i) for i in range(10))
|
|
for i in range(10):
|
|
check_id(i, next(rows))
|
|
|
|
def test_iter_and_next(self):
|
|
rows = records.RecordCollection(IdRecord(i) for i in range(10))
|
|
i = enumerate(iter(rows))
|
|
check_id(*next(i)) # Cache first row.
|
|
next(rows) # Cache second row.
|
|
check_id(*next(i)) # Read second row from cache.
|
|
|
|
def test_multiple_iter(self):
|
|
rows = records.RecordCollection(IdRecord(i) for i in range(10))
|
|
i = enumerate(iter(rows))
|
|
j = enumerate(iter(rows))
|
|
|
|
check_id(*next(i)) # Cache first row.
|
|
|
|
check_id(*next(j)) # Read first row from cache.
|
|
check_id(*next(j)) # Cache second row.
|
|
|
|
check_id(*next(i)) # Read second row from cache.
|
|
|
|
def test_slice_iter(self):
|
|
rows = records.RecordCollection(IdRecord(i) for i in range(10))
|
|
for i, row in enumerate(rows[:5]):
|
|
check_id(i, row)
|
|
for i, row in enumerate(rows):
|
|
check_id(i, row)
|
|
assert len(rows) == 10
|
|
|
|
|
|
# all
|
|
|
|
def test_all_returns_a_list_of_records(self):
|
|
rows = records.RecordCollection(IdRecord(i) for i in range(3))
|
|
assert rows.all() == [IdRecord(0), IdRecord(1), IdRecord(2)]
|
|
|
|
|
|
# first
|
|
|
|
def test_first_returns_a_single_record(self):
|
|
rows = records.RecordCollection(IdRecord(i) for i in range(1))
|
|
assert rows.first() == IdRecord(0)
|
|
|
|
def test_first_defaults_to_Nfirst(self):
|
|
rows = records.RecordCollection(iter([]))
|
|
assert rows.first() is None
|
|
|
|
def test_first_default_is_overridable(self):
|
|
rows = records.RecordCollection(iter([]))
|
|
assert rows.first('Cheese') == 'Cheese'
|
|
|
|
def test_first_raises_when_more_than_first(self):
|
|
rows = records.RecordCollection(IdRecord(i) for i in range(3))
|
|
raises(ValueError, rows.first)
|
|
|
|
def test_first_raises_default_if_its_an_exception_subclass(self):
|
|
rows = records.RecordCollection(iter([]))
|
|
class Cheese(Exception): pass
|
|
raises(Cheese, rows.first, Cheese)
|
|
|
|
def test_first_raises_default_if_its_an_exception_instance(self):
|
|
rows = records.RecordCollection(iter([]))
|
|
class Cheese(Exception): pass
|
|
raises(Cheese, rows.first, Cheese('cheddar'))
|
|
|
|
|
|
class TestRecord:
|
|
|
|
def test_record_dir(self):
|
|
keys, values = ['id', 'name', 'email'], [1, '', '']
|
|
record = records.Record(keys, values)
|
|
_dir = dir(record)
|
|
for key in keys:
|
|
assert key in _dir
|
|
for key in dir(object):
|
|
assert key in _dir
|