Fix tests for raising one(default)

This commit is contained in:
Chad Whitacre
2016-02-13 11:02:11 -05:00
parent ef9938623b
commit fafaca6a96
+2 -2
View File
@@ -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'))