refactor: Changed all tests to pytest

tests/core/test_mnemonic.py

@@ -23,86 +23,85 @@
 import pytest
 
 import random
-import sys
-import unittest
 
-from eth_account.hdaccount.mnemonic import Mnemonic
+from eth_account.hdaccount.mnemonic import (
+    Mnemonic,
+    ConfigurationError,
+)
 
 
-class MnemonicTest(unittest.TestCase):
+def test_failed_checksum():
+    code = (
+        "bless cloud wheel regular tiny venue bird web grief security dignity zoo"
+    )
+    mnemo = Mnemonic("english")
+    assert not mnemo.check(code)
 
-    def test_failed_checksum(self):
-        code = (
-            "bless cloud wheel regular tiny venue bird web grief security dignity zoo"
-        )
-        mnemo = Mnemonic("english")
-        self.assertFalse(mnemo.check(code))
 
-    def test_detection(self):
-        self.assertEqual("english", Mnemonic.detect_language("security"))
+def test_detection():
+    assert "english" == Mnemonic.detect_language("security")
 
-        with self.assertRaises(Exception):
-            Mnemonic.detect_language("xxxxxxx")
+    with pytest.raises(ConfigurationError):
+        Mnemonic.detect_language("xxxxxxx")
 
-    def test_utf8_nfkd(self):
-        # The same sentence in various UTF-8 forms
-        words_nfkd = u"Pr\u030ci\u0301s\u030cerne\u030c z\u030clut\u030couc\u030cky\u0301 ku\u030an\u030c u\u0301pe\u030cl d\u030ca\u0301belske\u0301 o\u0301dy za\u0301ker\u030cny\u0301 uc\u030cen\u030c be\u030cz\u030ci\u0301 pode\u0301l zo\u0301ny u\u0301lu\u030a"
-        words_nfc = u"P\u0159\xed\u0161ern\u011b \u017elu\u0165ou\u010dk\xfd k\u016f\u0148 \xfap\u011bl \u010f\xe1belsk\xe9 \xf3dy z\xe1ke\u0159n\xfd u\u010de\u0148 b\u011b\u017e\xed pod\xe9l z\xf3ny \xfal\u016f"
-        words_nfkc = u"P\u0159\xed\u0161ern\u011b \u017elu\u0165ou\u010dk\xfd k\u016f\u0148 \xfap\u011bl \u010f\xe1belsk\xe9 \xf3dy z\xe1ke\u0159n\xfd u\u010de\u0148 b\u011b\u017e\xed pod\xe9l z\xf3ny \xfal\u016f"
-        words_nfd = u"Pr\u030ci\u0301s\u030cerne\u030c z\u030clut\u030couc\u030cky\u0301 ku\u030an\u030c u\u0301pe\u030cl d\u030ca\u0301belske\u0301 o\u0301dy za\u0301ker\u030cny\u0301 uc\u030cen\u030c be\u030cz\u030ci\u0301 pode\u0301l zo\u0301ny u\u0301lu\u030a"
 
-        passphrase_nfkd = (
-            u"Neuve\u030cr\u030citelne\u030c bezpec\u030cne\u0301 hesli\u0301c\u030cko"
-        )
-        passphrase_nfc = (
-            u"Neuv\u011b\u0159iteln\u011b bezpe\u010dn\xe9 hesl\xed\u010dko"
-        )
-        passphrase_nfkc = (
-            u"Neuv\u011b\u0159iteln\u011b bezpe\u010dn\xe9 hesl\xed\u010dko"
-        )
-        passphrase_nfd = (
-            u"Neuve\u030cr\u030citelne\u030c bezpec\u030cne\u0301 hesli\u0301c\u030cko"
-        )
+def test_utf8_nfkd():
+    # The same sentence in various UTF-8 forms
+    words_nfkd = u"Pr\u030ci\u0301s\u030cerne\u030c z\u030clut\u030couc\u030cky\u0301 ku\u030an\u030c u\u0301pe\u030cl d\u030ca\u0301belske\u0301 o\u0301dy za\u0301ker\u030cny\u0301 uc\u030cen\u030c be\u030cz\u030ci\u0301 pode\u0301l zo\u0301ny u\u0301lu\u030a"
+    words_nfc = u"P\u0159\xed\u0161ern\u011b \u017elu\u0165ou\u010dk\xfd k\u016f\u0148 \xfap\u011bl \u010f\xe1belsk\xe9 \xf3dy z\xe1ke\u0159n\xfd u\u010de\u0148 b\u011b\u017e\xed pod\xe9l z\xf3ny \xfal\u016f"
+    words_nfkc = u"P\u0159\xed\u0161ern\u011b \u017elu\u0165ou\u010dk\xfd k\u016f\u0148 \xfap\u011bl \u010f\xe1belsk\xe9 \xf3dy z\xe1ke\u0159n\xfd u\u010de\u0148 b\u011b\u017e\xed pod\xe9l z\xf3ny \xfal\u016f"
+    words_nfd = u"Pr\u030ci\u0301s\u030cerne\u030c z\u030clut\u030couc\u030cky\u0301 ku\u030an\u030c u\u0301pe\u030cl d\u030ca\u0301belske\u0301 o\u0301dy za\u0301ker\u030cny\u0301 uc\u030cen\u030c be\u030cz\u030ci\u0301 pode\u0301l zo\u0301ny u\u0301lu\u030a"
 
-        seed_nfkd = Mnemonic.to_seed(words_nfkd, passphrase_nfkd)
-        seed_nfc = Mnemonic.to_seed(words_nfc, passphrase_nfc)
-        seed_nfkc = Mnemonic.to_seed(words_nfkc, passphrase_nfkc)
-        seed_nfd = Mnemonic.to_seed(words_nfd, passphrase_nfd)
+    passphrase_nfkd = (
+        u"Neuve\u030cr\u030citelne\u030c bezpec\u030cne\u0301 hesli\u0301c\u030cko"
+    )
+    passphrase_nfc = (
+        u"Neuv\u011b\u0159iteln\u011b bezpe\u010dn\xe9 hesl\xed\u010dko"
+    )
+    passphrase_nfkc = (
+        u"Neuv\u011b\u0159iteln\u011b bezpe\u010dn\xe9 hesl\xed\u010dko"
+    )
+    passphrase_nfd = (
+        u"Neuve\u030cr\u030citelne\u030c bezpec\u030cne\u0301 hesli\u0301c\u030cko"
+    )
 
-        self.assertEqual(seed_nfkd, seed_nfc)
-        self.assertEqual(seed_nfkd, seed_nfkc)
-        self.assertEqual(seed_nfkd, seed_nfd)
+    seed_nfkd = Mnemonic.to_seed(words_nfkd, passphrase_nfkd)
+    seed_nfc = Mnemonic.to_seed(words_nfc, passphrase_nfc)
+    seed_nfkc = Mnemonic.to_seed(words_nfkc, passphrase_nfkc)
+    seed_nfd = Mnemonic.to_seed(words_nfd, passphrase_nfd)
 
-    def test_to_entropy(self):
-        data = [
-            bytearray((random.getrandbits(8) for _ in range(32))) for _ in range(1024)
-        ]
-        data.append(b"Lorem ipsum dolor sit amet amet.")
-        m = Mnemonic("english")
-        for d in data:
-            self.assertEqual(m.to_entropy(m.to_mnemonic(d).split()), d)
+    assert seed_nfkd == seed_nfc
+    assert seed_nfkd == seed_nfkc
+    assert seed_nfkd == seed_nfd
 
-    def test_expand_word(self):
-        m = Mnemonic("english")
-        self.assertEqual("", m.expand_word(""))
-        self.assertEqual(" ", m.expand_word(" "))
-        self.assertEqual("access", m.expand_word("access"))  # word in list
-        self.assertEqual(
-            "access", m.expand_word("acce")
-        )  # unique prefix expanded to word in list
-        self.assertEqual("acb", m.expand_word("acb"))  # not found at all
-        self.assertEqual("acc", m.expand_word("acc"))  # multi-prefix match
-        self.assertEqual("act", m.expand_word("act"))  # exact three letter match
-        self.assertEqual(
-            "action", m.expand_word("acti")
-        )  # unique prefix expanded to word in list
 
-    def test_expand(self):
-        m = Mnemonic("english")
-        self.assertEqual("access", m.expand("access"))
-        self.assertEqual(
-            "access access acb acc act action", m.expand("access acce acb acc act acti")
-        )
+def test_to_entropy():
+    data = [
+        bytearray((random.getrandbits(8) for _ in range(32))) for _ in range(1024)
+    ]
+    data.append(b"Lorem ipsum dolor sit amet amet.")
+    m = Mnemonic("english")
+    for d in data:
+        assert m.to_entropy(m.to_mnemonic(d).split()) == d
+
+
+def test_expand_word():
+    m = Mnemonic("english")
+    assert "" == m.expand_word("")
+    assert " " == m.expand_word(" ")
+    assert "access" == m.expand_word("access")  # word in list
+    assert "access" == m.expand_word("acce")  # unique prefix expanded to word in list
+    assert "acb" == m.expand_word("acb")  # not found at all
+    assert "acc" == m.expand_word("acc")  # multi-prefix match
+    assert "act" == m.expand_word("act")  # exact three letter match
+    assert "action" == m.expand_word("acti")  # unique prefix expanded to word in list
+
+
+def test_expand():
+    m = Mnemonic("english")
+    assert "access" == m.expand("access")
+    assert "access access acb acc act action" == m.expand("access acce acb acc act acti")
+
 
 @pytest.mark.parametrize("encrypted_mnemonic,decrypted_mnemonic,expected_seed", [
     (
@@ -305,11 +304,3 @@ def test_english_mnemonics(encrypted_mnemonic, decrypted_mnemonic, expected_seed
     assert m.check(decrypted_mnemonic)
     assert decrypted_mnemonic == mnemonic
     assert expected_seed == seed.hex()
-
-
-def __main__():
-    unittest.main()
-
-
-if __name__ == "__main__":
-    __main__()