import unittest
import p6
from itertools import islice

class P6Tester(unittest.TestCase):
	def testCombinePrimes(self):
		self.assertEqual(self.getCombinedList(0), [])
		self.assertEqual(self.getCombinedList(5), [(2, 3), (2, 5), (3, 5), (2, 7), (3, 7)])

	def testDoubleSum(self):
		self.assertEqual(map(sum, self.getCombinedList(11)), [5, 7, 8, 9, 10, 12, 14, 16, 18, 18, 20])	

	def getCombinedList(self, n):
		return list(islice(p6.combinePrimes(), n))
	

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