def test_first_raises_default_if_its_an_exception_instance(self):
rows = camus.RecordCollection(iter([]))
class Cheese(Exception):
pass
raises(Cheese, rows.first, Cheese("cheddar"))
def test_one_raises_default_if_its_an_exception_subclass(self):
rows = camus.RecordCollection(iter([]))
class Cheese(Exception):
pass
raises(Cheese, rows.one, Cheese)
def test_slice_iter(self):
rows = camus.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
def test_multiple_iter(self):
rows = camus.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_first_default_is_overridable(self):
rows = camus.RecordCollection(iter([]))
assert rows.first("Cheese") == "Cheese"
def test_first_defaults_to_None(self):
rows = camus.RecordCollection(iter([]))
assert rows.first() is None
def test_first_returns_a_single_record(self):
rows = camus.RecordCollection(IdRecord(i) for i in range(1))
assert rows.first() == IdRecord(0)
def test_all_returns_a_list_of_records(self):
rows = camus.RecordCollection(IdRecord(i) for i in range(3))
assert rows.all() == [IdRecord(0), IdRecord(1), IdRecord(2)]
def test_next(self):
rows = camus.RecordCollection(IdRecord(i) for i in range(10))
for i in range(10):
check_id(i, next(rows))
def test_iter(self):
rows = camus.RecordCollection(IdRecord(i) for i in range(10))
for i, row in enumerate(rows):
check_id(i, row)
def test_scalar_raises_when_more_than_one(self):
rows = camus.RecordCollection(IdRecord(i) for i in range(3))
raises(ValueError, rows.scalar)
def test_scalar_default_is_overridable(self):
rows = camus.RecordCollection(iter([]))
assert rows.scalar("Kaffe") == "Kaffe"
def test_scalar_defaults_to_None(self):
rows = camus.RecordCollection(iter([]))
assert rows.scalar() is None
def test_scalar_returns_a_single_record(self):
rows = camus.RecordCollection(IdRecord(i) for i in range(1))
assert rows.scalar() == 0
