import unittest

import mnemonicode
from mnemonicode import _to_base, _from_base


class TestBaseConversion(unittest.TestCase):
    def test_encode_zero(self):
        self.assertEqual([], _to_base(12, 0))

    def test_encode_base_ten(self):
        self.assertEqual([1, 2, 3, 4, 5, 6], _to_base(10, 123456))

    def test_encode_negative(self):
        self.assertRaises(ValueError, _to_base, 64, -10)

    def test_decode_base_ten(self):
        self.assertEqual(123456, _from_base(10, [1, 2, 3, 4, 5, 6]))

    def test_decode_zero(self):
        self.assertEqual(0, _from_base(16, []))

    def test_decode_empty_digits(self):
        self.assertEqual(456, _from_base(10, [0, 0, 0, 4, 5, 6]))

    def test_decode_invalid_digits(self):
        self.assertRaises(ValueError, _from_base, 8, [128])
        self.assertRaises(ValueError, _from_base, 8, [8])


class TestMnemonicode(unittest.TestCase):
    def test_divide(self):
        self.assertEqual(
            list(mnemonicode._divide(b'12345678', 4)),
            [b'1234', b'5678']
        )

        self.assertEqual(
            list(mnemonicode._divide(b'1234567', 4)),
            [b'1234', b'567']
        )

        self.assertEqual(
            list(mnemonicode._divide(b'12345', 4)),
            [b'1234', b'5']
        )


class TestEncode(unittest.TestCase):
    def test_block_to_words(self):
        def test(string, words):
            self.assertEqual(tuple(mnemonicode._block_to_words(string)), words)

        test(b"a", ("camera",))
        test(b"ab", ("zero", "albert"))
        test(b"abc", ("hazard", "velvet", "jet"))
        test(b"abcd", ("bogart", "atlas", "safari"))

        # test padding
        test(b'x\x00', ("cecilia", "academy"))
        test(b'x\x00\x00', ("cecilia", "academy", "ego"))
        test(b'x\x00\x00\x00', ("cecilia", "academy", "academy"))

        # test all zeros
        test(b'\x00', ("academy",))
        test(b'\x00\x00', ("academy", "academy"))
        test(b'\x00\x00\x00', ("academy", "academy", "ego"))
        test(b'\x00\x00\x00\x00', ("academy", "academy", "academy"))

    def test_examples(self):
        def test(string, words):
            self.assertEqual(list(mnemonicode.mnencode(string)), words)

        test(b"a", [("camera",)])
        test(b"ab", [("zero", "albert")])
        test(b"abc", [("hazard", "velvet", "jet")])
        test(b"abcd", [
            ("bogart", "atlas", "safari")])
        test(b"abcde", [
            ("bogart", "atlas", "safari"),
            ("cannon",),
        ])
        test(b"abcdef", [
            ("bogart", "atlas", "safari"),
            ("david", "albino"),
        ])
        test(b"abcdefg", [
            ("bogart", "atlas", "safari"),
            ("emerald", "infant", "jet"),
        ])
        test(b"abcdefgh", [
            ("bogart", "atlas", "safari"),
            ("airport", "cabaret", "shock"),
        ])


class TestDecode(unittest.TestCase):
    def test_words_to_block(self):
        def test(string, words):
            self.assertEqual(mnemonicode._words_to_block(words), string)

        test(b"a", ("camera",))
        test(b"ab", ("zero", "albert"))
        test(b"abc", ("hazard", "velvet", "jet"))
        test(b"abcd", ("bogart", "atlas", "safari"))

        # test padding
        test(b'x\x00', ("cecilia", "academy"))
        test(b'x\x00\x00', ("cecilia", "academy", "ego"))
        test(b'x\x00\x00\x00', ("cecilia", "academy", "academy"))

        # test all zeros
        test(b'\x00', ("academy",))
        test(b'\x00\x00', ("academy", "academy"))
        test(b'\x00\x00\x00', ("academy", "academy", "ego"))
        test(b'\x00\x00\x00\x00', ("academy", "academy", "academy"))

    def test_examples(self):
        def test(string, words):
            self.assertEqual(mnemonicode.mndecode(words), string)

        test(b"a", [("camera",)])
        test(b"ab", [("zero", "albert")])
        test(b"abc", [("hazard", "velvet", "jet")])
        test(b"abcd", [
            ("bogart", "atlas", "safari")])
        test(b"abcde", [
            ("bogart", "atlas", "safari"),
            ("cannon",),
        ])
        test(b"abcdef", [
            ("bogart", "atlas", "safari"),
            ("david", "albino"),
        ])
        test(b"abcdefg", [
            ("bogart", "atlas", "safari"),
            ("emerald", "infant", "jet"),
        ])
        test(b"abcdefgh", [
            ("bogart", "atlas", "safari"),
            ("airport", "cabaret", "shock"),
        ])

    def test_early_three_byte_word(self):
        self.assertRaises(
            ValueError,
            mnemonicode.mndecode, [("jet",)]
        )

    def test_unknown_word(self):
        self.assertRaises(
            ValueError,
            mnemonicode.mndecode, [("supercalifragilisticexpialidocious",)]
        )

    def test_missing_wrapper_tuple(self):
        self.assertRaises(
            TypeError,
            mnemonicode.mndecode, ["academy"]
        )

    def test_empty_word_group(self):
        self.assertRaises(
            ValueError,
            mnemonicode.mndecode, [tuple()]
        )

    def testing_really_long_word_group(self):
        self.assertRaises(
            ValueError,
            mnemonicode.mndecode,
            [("academy", "academy", "academy", "academy")]
        )

    def test_decode_from_iterator(self):
        self.assertEqual(b"abcdef", mnemonicode.mndecode(iter([
            ("bogart", "atlas", "safari"),
            ("david", "albino"),
        ])))


loader = unittest.TestLoader()
suite = unittest.TestSuite((
    loader.loadTestsFromTestCase(TestBaseConversion),
    loader.loadTestsFromTestCase(TestMnemonicode),
    loader.loadTestsFromTestCase(TestEncode),
    loader.loadTestsFromTestCase(TestDecode),
))

@@ -4541,44 +4541,19 @@
 ses(
-%0A            ValueError,%0A           
+ValueError,
  mne
@@ -4577,33 +4577,24 @@
 , %5B(%22jet%22,)%5D
-%0A        
 )%0A%0A    def t
@@ -4827,43 +4827,18 @@
 ses(
-%0A            TypeError,%0A           
+TypeError,
  mne
@@ -4867,25 +4867,16 @@
 cademy%22%5D
-%0A        
 )%0A%0A    d
@@ -4937,44 +4937,19 @@
 ses(
-%0A            ValueError,%0A           
+ValueError,
  mne
@@ -4976,25 +4976,16 @@
 tuple()%5D
-%0A        
 )%0A%0A    d

trim extra whitespace