kennethreitz/tablib
1
0
Fork 0
mirror of https://github.com/kennethreitz/tablib.git synced 2026-06-05 23:10:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

Maintain dataset title after sort.

Browse Source
This commit is contained in:
Iuri de Silvio
2013-09-20 10:06:24 -03:00
parent c225a64d68
commit c8a5563309
2 changed files with 4 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tablib/core.py
+2 -2
View File
@@ -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:
test_tablib.py
+2 -1
View File
@@ -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]