diff --git a/tests/test_records.py b/tests/test_records.py index 9166446..f45b141 100644 --- a/tests/test_records.py +++ b/tests/test_records.py @@ -77,12 +77,12 @@ class TestRecordCollection: raises(ValueError, rows.one) def test_one_raises_default_if_its_an_exception_subclass(self): - rows = records.RecordCollection(IdRecord(i) for i in range(1)) + rows = records.RecordCollection(iter([])) class Cheese(Exception): pass raises(Cheese, rows.one, Cheese) def test_one_raises_default_if_its_an_exception_instance(self): - rows = records.RecordCollection(IdRecord(i) for i in range(1)) + rows = records.RecordCollection(iter([])) class Cheese(Exception): pass raises(Cheese, rows.one, Cheese('cheddar'))