tablib/test_tablib.py

#!/usr/bin/env python
# -*- coding: utf-8 -*-

import unittest

import tablib

class TablibTestCase(unittest.TestCase):
	def setUp(self):
		pass

	def tearDown(self):
		pass

	def test_empty_append(self):

		data = tablib.Dataset()

		new_row = (1,2,3)
		data.append(new_row)

		self.assertTrue(data.width == len(new_row))


	def test_empty_append_with_headers(self):

		data = tablib.Dataset()

		data.headers = ['first', 'second']
		new_row = (1,2,3,4)
		
		self.assertRaises(tablib.InvalidDimensions, data.append, new_row)
		
	# def test_adding_header with (self):
				



if __name__ == '__main__':
	unittest.main()