From c8a5563309f6b580748183ac2c9cfad88dd8ebec Mon Sep 17 00:00:00 2001 From: Iuri de Silvio Date: Fri, 20 Sep 2013 10:06:24 -0300 Subject: Maintain dataset title after sort. --- test_tablib.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) (limited to 'test_tablib.py') 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] -- cgit v1.2.1