from common import * from trezor.crypto import base32 class TestCryptoBase32(unittest.TestCase): # test vectors from: # https://tools.ietf.org/html/rfc4648 # https://github.com/emn178/hi-base32/blob/master/tests/test.js vectors = [ (b'', ''), (b'f', 'MY======'), (b'fo', 'MZXQ===='), (b'foo', 'MZXW6==='), (b'foob', 'MZXW6YQ='), (b'fooba', 'MZXW6YTB'), (b'foobar', 'MZXW6YTBOI======'), (b'H', 'JA======'), (b'He', 'JBSQ===='), (b'Hel', 'JBSWY==='), (b'Hell', 'JBSWY3A='), (b'Hello', 'JBSWY3DP'), (b'zlutoucky kun upel dabelske ody', 'PJWHK5DPOVRWW6JANN2W4IDVOBSWYIDEMFRGK3DTNNSSA33EPE======'), (b'中文', '4S4K3ZUWQ4======'), (b'中文1', '4S4K3ZUWQ4YQ===='), (b'中文12', '4S4K3ZUWQ4YTE==='), (b'aécio', 'MHB2SY3JN4======'), (b'𠜎', '6CQJZDQ='), (b'Base64是一種基於64個可列印字元來表示二進制資料的表示方法', 'IJQXGZJWGTTJRL7EXCAOPKFO4WP3VZUWXQ3DJZMARPSY7L7FRCL6LDNQ4WWZPZMFQPSL5BXIUGUOPJF24S5IZ2MAWLSYRNXIWOD6NFUZ46NIJ2FBVDT2JOXGS246NM4V') ] def test_encode(self): for a, b in self.vectors: self.assertEqual(base32.encode(a), b) def test_decode(self): for a, b in self.vectors: self.assertEqual(base32.decode(b), a) if __name__ == '__main__': unittest.main()