Add type ignore and minor linting

Author: c-w-feldmann

tests/test_elements/test_mol2mol/test_mol2mol_filter.py

@@ -1,9 +1,14 @@
-"""Test MolFilter, which invalidate molecules based on criteria defined in the respective filter."""
+"""Unittest for MolFilter functionality.
+
+MolFilter set flag Molecules as invalid based on the criteria defined in the filter.
+
+"""
 
 import json
 import tempfile
 import unittest
 from pathlib import Path
+from typing import TYPE_CHECKING
 
 from molpipeline import ErrorFilter, FilterReinserter, Pipeline
 from molpipeline.any2mol import SmilesToMol
@@ -19,7 +24,9 @@
 )
 from molpipeline.utils.comparison import compare_recursive
 from molpipeline.utils.json_operations import recursive_from_json, recursive_to_json
-from molpipeline.utils.molpipeline_types import FloatCountRange, IntOrIntCountRange
+
+if TYPE_CHECKING:
+    from molpipeline.utils.molpipeline_types import FloatCountRange, IntOrIntCountRange
 
 # pylint: disable=duplicate-code  # test case molecules are allowed to be duplicated
 SMILES_ANTIMONY = "[SbH6+3]"
@@ -80,15 +87,15 @@ def test_element_filter(self) -> None:
                         6: 6,
                         1: (5, 6),
                         17: (0, 1),
-                    }
+                    },
                 },
                 "result": [SMILES_BENZENE, SMILES_CHLOROBENZENE],
             },
             {"params": {"ElementFilter__add_hydrogens": False}, "result": []},
         ]
 
         for test_params in test_params_list_with_results:
-            pipeline.set_params(**test_params["params"])
+            pipeline.set_params(**test_params["params"])  # type: ignore
             filtered_smiles = pipeline.fit_transform(SMILES_LIST)
             self.assertEqual(filtered_smiles, test_params["result"])
 
@@ -99,14 +106,15 @@ def test_json_roundtrip(self) -> None:
         -----
         It is important to save the ElementFilter as a JSON file and then load it back.
         This is because json.dumps() sets the keys of the dictionary to strings.
+
         """
         element_filter = ElementFilter()
         json_object = recursive_to_json(element_filter)
         with tempfile.TemporaryDirectory() as temp_folder:
             temp_file_path = Path(temp_folder) / "test.json"
-            with open(temp_file_path, "w", encoding="UTF-8") as out_file:
+            with temp_file_path.open("w", encoding="UTF-8") as out_file:
                 json.dump(json_object, out_file)
-            with open(temp_file_path, encoding="UTF-8") as in_file:
+            with temp_file_path.open(encoding="UTF-8") as in_file:
                 loaded_json_object = json.load(in_file)
         recreated_element_filter = recursive_from_json(loaded_json_object)
 
@@ -117,7 +125,8 @@ def test_json_roundtrip(self) -> None:
             with self.subTest(param_name=param_name):
                 self.assertTrue(
                     compare_recursive(original_value, recreated_params[param_name]),
-                    f"Original: {original_value}, Recreated: {recreated_params[param_name]}",
+                    f"Original: {original_value}, "
+                    f"Recreated: {recreated_params[param_name]}",
                 )
 
 
@@ -132,6 +141,7 @@ def _create_pipeline() -> Pipeline:
         -------
         Pipeline
             Pipeline with a complex filter.
+
         """
         element_filter_1 = ElementFilter({6: 6, 1: 6})
         element_filter_2 = ElementFilter({6: 6, 1: 5, 17: 1})
@@ -140,18 +150,17 @@ def _create_pipeline() -> Pipeline:
             (
                 ("element_filter_1", element_filter_1),
                 ("element_filter_2", element_filter_2),
-            )
+            ),
         )
 
-        pipeline = Pipeline(
+        return Pipeline(
             [
                 ("Smiles2Mol", SmilesToMol()),
                 ("MultiElementFilter", multi_element_filter),
                 ("Mol2Smiles", MolToSmiles()),
                 ("ErrorFilter", ErrorFilter()),
             ],
         )
-        return pipeline
 
     def test_complex_filter(self) -> None:
         """Test if molecules are filtered correctly by allowed chemical elements."""
@@ -169,14 +178,14 @@ def test_complex_filter(self) -> None:
             {
                 "params": {
                     "MultiElementFilter__mode": "any",
-                    "MultiElementFilter__pipeline_filter_elements__element_filter_1__add_hydrogens": False,
+                    "MultiElementFilter__pipeline_filter_elements__element_filter_1__add_hydrogens": False,  # noqa: E501
                 },
                 "result": [SMILES_CHLOROBENZENE],
             },
         ]
 
         for test_params in test_params_list_with_results:
-            pipeline.set_params(**test_params["params"])
+            pipeline.set_params(**test_params["params"])  # type: ignore
             filtered_smiles = pipeline.fit_transform(SMILES_LIST)
             self.assertEqual(filtered_smiles, test_params["result"])
 
@@ -198,7 +207,7 @@ def test_complex_filter_non_unique_names(self) -> None:
 
         with self.assertRaises(ValueError):
             ComplexFilter(
-                (("filter_1", element_filter_1), ("filter_1", element_filter_2))
+                (("filter_1", element_filter_1), ("filter_1", element_filter_2)),
             )
 
 
@@ -264,7 +273,7 @@ def test_smarts_smiles_filter(self) -> None:
             ]
 
             for test_params in test_params_list_with_results:
-                pipeline.set_params(**test_params["params"])
+                pipeline.set_params(**test_params["params"])  # type: ignore
                 filtered_smiles = pipeline.fit_transform(SMILES_LIST)
                 self.assertEqual(filtered_smiles, test_params["result"])
 
@@ -285,7 +294,11 @@ def test_smarts_smiles_filter_wrong_pattern(self) -> None:
             SmilesFilter(smiles_pats)
 
     def test_smarts_filter_parallel(self) -> None:
-        """Test if molecules are filtered correctly by allowed SMARTS patterns in parallel."""
+        """Test if molecules are filtered correctly.
+
+        This test runs the SmartsFilter in parallel.
+
+        """
         smarts_pats: dict[str, IntOrIntCountRange] = {
             "c": (4, None),
             "Cl": 1,
@@ -352,38 +365,38 @@ def test_descriptor_filter(self) -> None:
             },
             {
                 "params": {
-                    "DescriptorsFilter__filter_elements": {"NumHAcceptors": (1.99, 4)}
+                    "DescriptorsFilter__filter_elements": {"NumHAcceptors": (1.99, 4)},
                 },
                 "result": [SMILES_CL_BR],
             },
             {
                 "params": {
-                    "DescriptorsFilter__filter_elements": {"NumHAcceptors": (2.01, 4)}
+                    "DescriptorsFilter__filter_elements": {"NumHAcceptors": (2.01, 4)},
                 },
                 "result": [],
             },
             {
                 "params": {
-                    "DescriptorsFilter__filter_elements": {"NumHAcceptors": (1, 2.00)}
+                    "DescriptorsFilter__filter_elements": {"NumHAcceptors": (1, 2.00)},
                 },
                 "result": [SMILES_CL_BR],
             },
             {
                 "params": {
-                    "DescriptorsFilter__filter_elements": {"NumHAcceptors": (1, 2.01)}
+                    "DescriptorsFilter__filter_elements": {"NumHAcceptors": (1, 2.01)},
                 },
                 "result": [SMILES_CL_BR],
             },
             {
                 "params": {
-                    "DescriptorsFilter__filter_elements": {"NumHAcceptors": (1, 1.99)}
+                    "DescriptorsFilter__filter_elements": {"NumHAcceptors": (1, 1.99)},
                 },
                 "result": [],
             },
         ]
 
         for test_params in test_params_list_with_results:
-            pipeline.set_params(**test_params["params"])
+            pipeline.set_params(**test_params["params"])  # type: ignore
             filtered_smiles = pipeline.fit_transform(SMILES_LIST)
             self.assertEqual(filtered_smiles, test_params["result"])
 
@@ -409,7 +422,7 @@ def test_invalidate_mixtures(self) -> None:
                 ("mol2smi", mol2smi),
                 ("error_filter", error_filter),
                 ("error_replacer", error_replacer),
-            ]
+            ],
         )
         mols_processed = pipeline.fit_transform(mol_list)
         self.assertEqual(expected_invalidated_mol_list, mols_processed)
@@ -424,7 +437,7 @@ def test_inorganic_filter(self) -> None:
         inorganics_filter = InorganicsFilter()
         mol2smiles = MolToSmiles()
         error_filter = ErrorFilter.from_element_list(
-            [smiles2mol, inorganics_filter, mol2smiles]
+            [smiles2mol, inorganics_filter, mol2smiles],
         )
         pipeline = Pipeline(
             [