diff options
| author | Kenneth Reitz <me@kennethreitz.org> | 2016-02-07 06:36:12 -0500 |
|---|---|---|
| committer | Kenneth Reitz <me@kennethreitz.org> | 2016-02-07 06:36:12 -0500 |
| commit | d90358bf6991ec19a4c6a07173f9127e970d96fb (patch) | |
| tree | 7e76962d170635d1073360cc8bbb65b7d94686ff /test_tablib.py | |
| parent | c5920249ded38dc5fe70f22021f4a0dd73dd97cf (diff) | |
| parent | 5fbdd56fba67b259a2bee7979b20fbbd9a31dc63 (diff) | |
| download | tablib-d90358bf6991ec19a4c6a07173f9127e970d96fb.tar.gz | |
Merge branch 'develop' of https://github.com/rabinnankhwa/tablib into develop
# Conflicts:
# AUTHORS
Diffstat (limited to 'test_tablib.py')
| -rwxr-xr-x | test_tablib.py | 20 |
1 files changed, 20 insertions, 0 deletions
diff --git a/test_tablib.py b/test_tablib.py index 2a4f687..87f35f9 100755 --- a/test_tablib.py +++ b/test_tablib.py @@ -811,6 +811,26 @@ class TablibTestCase(unittest.TestCase): self.assertTrue(data[0] == new_row) + def test_subset(self): + """Create a subset of a dataset""" + + rows = (0, 2) + columns = ('first_name','gpa') + + data.headers = self.headers + + data.append(self.john) + data.append(self.george) + data.append(self.tom) + + #Verify data is truncated + subset = data.subset(rows=rows, cols=columns) + self.assertEqual(type(subset), tablib.Dataset) + self.assertEqual(subset.headers, list(columns)) + self.assertEqual(subset._data[0].list, ['John', 90]) + self.assertEqual(subset._data[1].list, ['Thomas', 50]) + + def test_formatters(self): """Confirm formatters are being triggered.""" |
