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()