summaryrefslogtreecommitdiff
path: root/test_tablib.py
diff options
context:
space:
mode:
authorKenneth Reitz <me@kennethreitz.org>2016-02-07 06:36:12 -0500
committerKenneth Reitz <me@kennethreitz.org>2016-02-07 06:36:12 -0500
commitd90358bf6991ec19a4c6a07173f9127e970d96fb (patch)
tree7e76962d170635d1073360cc8bbb65b7d94686ff /test_tablib.py
parentc5920249ded38dc5fe70f22021f4a0dd73dd97cf (diff)
parent5fbdd56fba67b259a2bee7979b20fbbd9a31dc63 (diff)
downloadtablib-d90358bf6991ec19a4c6a07173f9127e970d96fb.tar.gz
Merge branch 'develop' of https://github.com/rabinnankhwa/tablib into develop
# Conflicts: # AUTHORS
Diffstat (limited to 'test_tablib.py')
-rwxr-xr-xtest_tablib.py20
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."""