From fa6f5b3af316bfc8195921527418297681674ea2 Mon Sep 17 00:00:00 2001 From: Kuba Janoszek Date: Tue, 13 Mar 2012 00:21:32 +0100 Subject: [PATCH] Databook.add_sheet test for not Dataset subclass added. --- test_tablib.py | 11 +++++++++++ 1 file changed, 11 insertions(+) diff --git a/test_tablib.py b/test_tablib.py index fbb6eab..aab1b0f 100755 --- a/test_tablib.py +++ b/test_tablib.py @@ -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