from .. import fpeck
import unittest

class FpeckTests(unittest.TestCase):
    def test_5peck_n_10(self):
        k = fpeck.expand(0x000102030405060708090a0b0c0d0e0f)
        self.assertEqual(fpeck.encrypt(k, 10, 0), 917905)
        self.assertEqual(fpeck.decrypt(k, 10, 917905), 0)
        self.assertEqual(fpeck.encrypt(k, 10, 0b1111111111), 314126)
        self.assertEqual(fpeck.decrypt(k, 10, 314126), 0b1111111111)
        # Look, anything else can be covered by the TESID tests.
        # They cover at least two values from each 5peck instance.


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