circuitpython/tests/extmod/ucryptolib_aes256_cbc.py

try:
    from Crypto.Cipher import AES

    aes = AES.new
except ImportError:
    try:
        from ucryptolib import aes
    except ImportError:
        print("SKIP")
        raise SystemExit

crypto = aes(b"1234" * 8, 2, b"5678" * 4)
enc = crypto.encrypt(bytes(range(32)))
print(enc)
crypto = aes(b"1234" * 8, 2, b"5678" * 4)
print(crypto.decrypt(enc))
tests/extmod/ucryptolib*: Add tests for ucryptolib module. 2018-01-07 08:14:36 -05:00			`try:`
			`from Crypto.Cipher import AES`
tests: Format all Python code with black, except tests in basics subdir. This adds the Python files in the tests/ directory to be formatted with ./tools/codeformat.py. The basics/ subdirectory is excluded for now so we aren't changing too much at once. In a few places `# fmt: off`/`# fmt: on` was used where the code had special formatting for readability or where the test was actually testing the specific formatting. 2020-03-22 22:26:08 -04:00
tests/extmod/ucryptolib*: Add tests for ucryptolib module. 2018-01-07 08:14:36 -05:00			`aes = AES.new`
			`except ImportError:`
			`try:`
			`from ucryptolib import aes`
			`except ImportError:`
			`print("SKIP")`
			`raise SystemExit`

			`crypto = aes(b"1234" * 8, 2, b"5678" * 4)`
			`enc = crypto.encrypt(bytes(range(32)))`
			`print(enc)`
			`crypto = aes(b"1234" * 8, 2, b"5678" * 4)`
			`print(crypto.decrypt(enc))`