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

"""Tests for tablib"""

import unittest

import tablib


class TablibTestCase(unittest.TestCase):
	"""Tablib test cases"""

	def setUp(self):
		"""Create simple data set with headers"""
		headers = ('first_name', 'last_name', 'gpa')
		self.john = ('John', 'Adams', 90)
		self.george = ('George', 'Washington', 67)

		self.founders = tablib.Dataset(headers=headers)
		self.founders.append(self.john)
		self.founders.append(self.george)

	def tearDown(self):
		"""teardown"""
		pass

	def test_empty_append(self):
		"""Verify append() correctly adds tuple with no headers"""
		data = tablib.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)

	def test_empty_append_with_headers(self):
		"""Verify append() correctly detects mismatch of number of
		headers and data
		"""
		data = tablib.Dataset()

		data.headers = ['first', 'second']
		new_row = (1, 2, 3, 4)

		self.assertRaises(tablib.InvalidDimensions, data.append, new_row)

	def test_header_slicing(self):
		"""Verify slicing by headers"""

		self.assertEqual(self.founders['first_name'], [self.john[0], self.george[0]])
		self.assertEqual(self.founders['last_name'], [self.john[1], self.george[1]])
		self.assertEqual(self.founders['gpa'], [self.john[2], self.george[2]])

	def test_data_slicing(self):
		"""Verify slicing by data"""

		# Slice individual rows
		self.assertEqual(self.founders[0], self.john)
		self.assertEqual(self.founders[:1], [self.john])
		self.assertEqual(self.founders[1:2], [self.george])
		self.assertEqual(self.founders[2:], [])
		self.assertEqual(self.founders[-1], self.george)

		# Slice multiple rows
		self.assertEqual(self.founders[:], [self.john, self.george])
		self.assertEqual(self.founders[0:2], [self.john, self.george])
		self.assertEqual(self.founders[2:], [])

	# def test_adding_header with (self):


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