TeamPGM/tgcrypto
2
0
Fork 0
mirror of https://github.com/TeamPGM/tgcrypto.git synced 2024-11-25 09:27:45 +00:00
Code Issues Packages Projects Releases Wiki Activity

Don't use pathlib

Browse Source
This commit is contained in:
Dan 2019-06-13 16:39:56 +02:00
parent bd20ae640b
commit f07c2ea4ab
1 changed files with 5 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
tests/cbc/test_cbc.py
View File

@ -20,7 +20,6 @@ import os
import random import random
import re import re
import unittest import unittest
from pathlib import Path
import tgcrypto import tgcrypto
@ -96,7 +95,11 @@ class TestCBC256Cryptography(unittest.TestCase):
PATTERN = r"COUNT = (\d+)\nKEY = (\w+)\nIV = (\w+)\n(PLAINTEXT|CIPHERTEXT) = (\w+)\n(PLAINTEXT|CIPHERTEXT) = (\w+)" PATTERN = r"COUNT = (\d+)\nKEY = (\w+)\nIV = (\w+)\n(PLAINTEXT|CIPHERTEXT) = (\w+)\n(PLAINTEXT|CIPHERTEXT) = (\w+)"
for path in (Path(__file__).parent / "vectors").rglob("*.rsp"): BASE_PATH = os.path.dirname(__file__) + "/vectors"
for path in os.listdir(BASE_PATH):
path = BASE_PATH + "/" + path
with open(path, "r", encoding="utf-8") as f: with open(path, "r", encoding="utf-8") as f:
for match in re.finditer(PATTERN, f.read()): for match in re.finditer(PATTERN, f.read()):
count, key, iv, plain_or_cipher, bytes1, _, bytes2 = match.groups() count, key, iv, plain_or_cipher, bytes1, _, bytes2 = match.groups()