Merge pull request #144 from crocs-muni/closes-#106

Black, Flake8, isort linters: Closes-#106

1 files changed, 10 insertions, 8 deletions

diff --git a/tests/test_cc_txt_processing.py b/tests/test_cc_txt_processing.py
index 01b2212b..cc1cb23c 100644
--- a/tests/test_cc_txt_processing.py
+++ b/tests/test_cc_txt_processing.py
@@ -1,24 +1,26 @@
-import tests.data.test_cc_heuristics
-from unittest import TestCase
+import shutil
+import tempfile
 from pathlib import Path
 from typing import ClassVar
-import tempfile
-import shutil
+from unittest import TestCase
 
+import tests.data.test_cc_heuristics
 from sec_certs.dataset.common_criteria import CCDataset
 from sec_certs.sample.common_criteria import CommonCriteriaCert
 
 
-class TestCommonCriteriaHeuristics(TestCase):
-    dataset_json_path: ClassVar[Path] = Path(tests.data.test_cc_heuristics.__path__[0]) / 'vulnerable_dataset.json'
+class TestCommonCriteriaTextProcessing(TestCase):
+    dataset_json_path: ClassVar[Path] = Path(tests.data.test_cc_heuristics.__path__[0]) / "vulnerable_dataset.json"  # type: ignore  # mypy issue #1422
     data_dir_path: ClassVar[Path] = dataset_json_path.parent
+    tmp_dir: ClassVar[tempfile.TemporaryDirectory]
+    cc_dset: CCDataset
 
     @classmethod
     def setUpClass(cls) -> None:
-        cls.tmp_dir: ClassVar[tempfile.TemporaryDirectory] = tempfile.TemporaryDirectory()
+        cls.tmp_dir = tempfile.TemporaryDirectory()
         shutil.copytree(cls.data_dir_path, cls.tmp_dir.name, dirs_exist_ok=True)
 
-        cls.cc_dset: CCDataset = CCDataset.from_json(Path(cls.tmp_dir.name) / 'vulnerable_dataset.json')
+        cls.cc_dset = CCDataset.from_json(Path(cls.tmp_dir.name) / "vulnerable_dataset.json")
         cls.cc_dset.download_all_pdfs()
         cls.cc_dset.convert_all_pdfs()