diff options
| -rw-r--r-- | tablib/core.py | 6 | ||||
| -rwxr-xr-x | test_tablib.py | 15 |
2 files changed, 21 insertions, 0 deletions
diff --git a/tablib/core.py b/tablib/core.py index 3f79a36..e368bab 100644 --- a/tablib/core.py +++ b/tablib/core.py @@ -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): diff --git a/test_tablib.py b/test_tablib.py index 6758daa..54ab6f3 100755 --- a/test_tablib.py +++ b/test_tablib.py @@ -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() |