import unittest

from rfc1751 import rfc1751


class TestRFC1751(unittest.TestCase):
    def testRFC1751(self) -> None:
        self.assertEqual(rfc1751(0), ['A'])
        self.assertEqual(rfc1751(1), ['ABE'])
        self.assertEqual(rfc1751(2), ['ACE'])
        self.assertEqual(rfc1751(2047), ['YOKE'])
        self.assertEqual(rfc1751(2048), ['ABE', 'A'])
        self.assertEqual(rfc1751(2049), ['ABE', 'ABE'])
        self.assertEqual(rfc1751(2050), ['ABE', 'ACE'])

        # Note: These can get NSFW
        self.assertEqual(rfc1751(2112325), ['FIRM', 'COCK'])


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