Added dataset wipe.

This commit is contained in:
Kenneth Reitz
2010-09-25 06:22:40 -04:00
parent 80cb42e8dd
commit a9c7a5067d
2 changed files with 21 additions and 0 deletions
+6
View File
@@ -204,6 +204,12 @@ class Dataset(object):
self._data.insert(i, tuple(row))
elif col:
pass
def wipe(self):
"""Erases all data from Dataset."""
self._data = list()
self.__headers = None
class Databook(object):
+15
View File
@@ -230,6 +230,21 @@ class TablibTestCase(unittest.TestCase):
new_data = tablib.formats.csv.import_set(str(data.csv))
# print new_data
def test_wipe(self):
"""Purge a dataset."""
new_row = (1, 2, 3)
data.append(new_row)
# Verify width/data
self.assertTrue(data.width == len(new_row))
self.assertTrue(data[0] == new_row)
data.wipe()
new_row = (1, 2, 3, 4)
data.append(new_row)
self.assertTrue(data.width == len(new_row))
self.assertTrue(data[0] == new_row)
if __name__ == '__main__':
unittest.main()