Databook.add_sheet test for not Dataset subclass added.

This commit is contained in:
Kuba Janoszek
2012-03-13 00:21:32 +01:00
parent 0528e0a500
commit fa6f5b3af3
+11
View File
@@ -669,6 +669,17 @@ Old |Man |100500
)
def test_databook_add_sheet_accepts_only_dataset_instances(self):
class NotDataset(object):
def append(self, item):
pass
dataset = NotDataset()
dataset.append(self.john)
self.assertRaises(tablib.InvalidDatasetType, book.add_sheet, dataset)
def test_databook_add_sheet_accepts_dataset_subclasses(self):
class DatasetSubclass(tablib.Dataset):
pass