import unittest
import itertools
from p6 import *

class P6Tester(unittest.TestCase):

	def test_sample(self):
		enumerate_expected = (1, 1), (2, 1), (1, 2), (1, 3), (3, 1), (4, 1), (3, 2), (2, 3), (1, 4), (1, 5), (5, 1)
		primed_expected = (2, 3), (2, 5), (3, 5)
		self.assertEqual(tuple(itertools.islice(enumerate_rationals(), len(enumerate_expected))), enumerate_expected)
		self.assertEqual(tuple(itertools.islice(primed(enumerate_rationals()), len(primed_expected))), primed_expected)


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