From fafaca6a969101870a379b1b802a275e7a0bbe66 Mon Sep 17 00:00:00 2001 From: Chad Whitacre Date: Sat, 13 Feb 2016 11:02:11 -0500 Subject: [PATCH] Fix tests for raising one(default) --- tests/test_records.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) 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'))