summaryrefslogtreecommitdiff
path: root/portato/TEST_helper.py
blob: 88c4cd33d6263cea8cf1dcd2fc9e95d348043a8c (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
#!/usr/bin/python

from __future__ import absolute_import

import unittest
import .helper

class HelperTest (unittest.TestCase):

	def testFlatten(self):
		list = [[1,2],[3,4],[[5],[6,7,8], 9]]
		flist = helper.flatten(list)
		self.assertEqual(flist, [1,2,3,4,5,6,7,8,9], "List not flattend correctly.")

	def testUniqueArray(self):

		def equal (l1, l2):
			for i in l1:
				if i not in l2:
					return False
				l2.remove(i)
			return True

		list1 = [1,4,5,2,1,7,9,11,2,4,7,12]
		result1 = [1,4,5,2,7,9,11,12]

		list2 = [[x] for x in list1]
		result2 = [[x] for x in result1]

		self.assert_(equal(helper.unique_array(list1), result1), "Make hashable list unique does not work.")
		self.assert_(equal(helper.unique_array(list2), result2), "Make unhashable list unique does not work.")

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