nth fibonacci

fibonacci.py

import unittest

class FibonacciTest(unittest.TestCase):

    def test_sequence(self):
        self.assertEqual(fibonacci_nth(0), 0)
        self.assertEqual(fibonacci_nth(1), 1)
        self.assertEqual(fibonacci_nth(2), 1)
        self.assertEqual(fibonacci_nth(3), 2)
        self.assertEqual(fibonacci_nth(4), 3)
        self.assertEqual(fibonacci_nth(5), 5)
        self.assertEqual(fibonacci_nth(6), 8)

    def test_LargerValue(self):
        self.assertEqual(fibonacci_nth(100), 354224848179261915075)


def fibonacci_generator():
    a = 0
    b = 1
    while True:
        yield a
        a, b = b, a + b

def fibonacci_nth(n):
    for i, v in enumerate(fibonacci_generator()):
        if i == n:
            return v


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