import unittest
-from rfc1751 import rfc1751
+from rfc1751 import encode
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'])
+ self.assertEqual(encode(0), ['A'])
+ self.assertEqual(encode(1), ['ABE'])
+ self.assertEqual(encode(2), ['ACE'])
+ self.assertEqual(encode(2047), ['YOKE'])
+ self.assertEqual(encode(2048), ['ABE', 'A'])
+ self.assertEqual(encode(2049), ['ABE', 'ABE'])
+ self.assertEqual(encode(2050), ['ABE', 'ACE'])
# Note: These can get NSFW
- self.assertEqual(rfc1751(2112325), ['FIRM', 'COCK'])
+ self.assertEqual(encode(2112325), ['FIRM', 'COCK'])
if __name__ == '__main__':