summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rwxr-xr-xtest_tablib.py10
1 files changed, 6 insertions, 4 deletions
diff --git a/test_tablib.py b/test_tablib.py
index 342fd45..8687473 100755
--- a/test_tablib.py
+++ b/test_tablib.py
@@ -8,6 +8,7 @@ import unittest
import tablib
+
class TablibTestCase(unittest.TestCase):
"""Tablib test cases."""
@@ -15,6 +16,7 @@ class TablibTestCase(unittest.TestCase):
"""Create simple data set with headers."""
global data, book
+
data = tablib.Dataset()
book = tablib.Databook()
@@ -352,20 +354,20 @@ class TablibTestCase(unittest.TestCase):
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()