From c8a5563309f6b580748183ac2c9cfad88dd8ebec Mon Sep 17 00:00:00 2001 From: Iuri de Silvio Date: Fri, 20 Sep 2013 10:06:24 -0300 Subject: [PATCH] Maintain dataset title after sort. --- tablib/core.py | 4 ++-- test_tablib.py | 3 ++- 2 files changed, 4 insertions(+), 3 deletions(-) diff --git a/tablib/core.py b/tablib/core.py index 851b25c..86f75b2 100644 --- a/tablib/core.py +++ b/tablib/core.py @@ -800,7 +800,7 @@ class Dataset(object): raise HeadersNeeded _sorted = sorted(self.dict, key=itemgetter(col), reverse=reverse) - _dset = Dataset(headers=self.headers) + _dset = Dataset(headers=self.headers, title=self.title) for item in _sorted: row = [item[key] for key in self.headers] @@ -811,7 +811,7 @@ class Dataset(object): col = self.headers[col] _sorted = sorted(self.dict, key=itemgetter(col), reverse=reverse) - _dset = Dataset(headers=self.headers) + _dset = Dataset(headers=self.headers, title=self.title) for item in _sorted: if self.headers: diff --git a/test_tablib.py b/test_tablib.py index 4b4ad93..6125408 100755 --- a/test_tablib.py +++ b/test_tablib.py @@ -29,7 +29,7 @@ class TablibTestCase(unittest.TestCase): self.george = ('George', 'Washington', 67) self.tom = ('Thomas', 'Jefferson', 50) - self.founders = tablib.Dataset(headers=self.headers) + self.founders = tablib.Dataset(headers=self.headers, title='Founders') self.founders.append(self.john) self.founders.append(self.george) self.founders.append(self.tom) @@ -559,6 +559,7 @@ class TablibTestCase(unittest.TestCase): """Sort columns.""" sorted_data = self.founders.sort(col="first_name") + self.assertEqual(sorted_data.title, 'Founders') first_row = sorted_data[0] second_row = sorted_data[2]