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.
29 lines
530 B
Python
29 lines
530 B
Python
try:
|
|
from ucryptolib import aes
|
|
except ImportError:
|
|
print("SKIP")
|
|
raise SystemExit
|
|
|
|
|
|
def _new(k, ctr_initial):
|
|
return aes(k, 6, ctr_initial)
|
|
|
|
|
|
try:
|
|
_new(b"x" * 16, b"x" * 16)
|
|
except ValueError as e:
|
|
# is CTR support disabled?
|
|
if e.args[0] == "mode":
|
|
print("SKIP")
|
|
raise SystemExit
|
|
raise e
|
|
|
|
crypto = _new(b"1234" * 4, b"5678" * 4)
|
|
enc = crypto.encrypt(b"a")
|
|
print(enc)
|
|
enc += crypto.encrypt(b"b" * 1000)
|
|
print(enc)
|
|
|
|
crypto = _new(b"1234" * 4, b"5678" * 4)
|
|
print(crypto.decrypt(enc))
|