from unittest import main, TestCase
import zbase32
class Test(TestCase):
def test_encode_decode(self):
self.assertEqual(b"foo", zbase32.decode((zbase32.encode(b"foo"))))
self.assertEqual("c3zs6", zbase32.encode(b"foo"))
self.assertEqual(b"foo", zbase32.decode("c3zs6"))
def test_exception(self):
with self.assertRaises(zbase32.DecodeError):
zbase32.decode("invalid@char")
if __name__ == "__main__":
main()