Databook.add_sheet test for not Dataset subclass added.

1 files changed, 11 insertions, 0 deletions

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