diff options
| author | Kenneth Reitz <me@kennethreitz.com> | 2010-09-25 06:22:40 -0400 |
|---|---|---|
| committer | Kenneth Reitz <me@kennethreitz.com> | 2010-09-25 06:22:40 -0400 |
| commit | a9c7a5067dffa9f5d67708941c0eefca7aa7c203 (patch) | |
| tree | bde8b5f7c452c90bf7697e53cb8a7fce5a1cff39 | |
| parent | 80cb42e8dd787a2ceffb37fe7a2380a97d000f4e (diff) | |
| download | tablib-a9c7a5067dffa9f5d67708941c0eefca7aa7c203.tar.gz | |
Added dataset wipe.
| -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() |
