From 8e837349855cef2e54460b705a166bf85e84017c Mon Sep 17 00:00:00 2001 From: Kuba Janoszek Date: Tue, 13 Mar 2012 00:05:24 +0100 Subject: Databook.add_sheet accepts Dataset subclasses --- test_tablib.py | 15 +++++++++++++++ 1 file changed, 15 insertions(+) (limited to 'test_tablib.py') diff --git a/test_tablib.py b/test_tablib.py index 02a1eb3..fbb6eab 100755 --- a/test_tablib.py +++ b/test_tablib.py @@ -669,5 +669,20 @@ Old |Man |100500 ) + def test_databook_add_sheet_accepts_dataset_subclasses(self): + class DatasetSubclass(tablib.Dataset): + pass + + # just checking if subclass of tablib.Dataset can be added to Databook + dataset = DatasetSubclass() + dataset.append(self.john) + dataset.append(self.tom) + + try: + book.add_sheet(dataset) + except tablib.InvalidDatasetType: + self.fail("Subclass of tablib.Dataset should be accepted by Databook.add_sheet") + + if __name__ == '__main__': unittest.main() -- cgit v1.2.1