3dc324d3f1
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.
17 lines
355 B
Python
17 lines
355 B
Python
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))
|