From 2cb5a56cff6fc9b0a4ae81d337c15bda3866b86f Mon Sep 17 00:00:00 2001 From: Christian Feldmann Date: Wed, 12 Feb 2025 13:11:44 +0100 Subject: [PATCH] black --- tests/test_pipeline.py | 4 ---- 1 file changed, 4 deletions(-) diff --git a/tests/test_pipeline.py b/tests/test_pipeline.py index dd338573..e9edf965 100644 --- a/tests/test_pipeline.py +++ b/tests/test_pipeline.py @@ -224,7 +224,6 @@ def test_fit_transform_record_remove_nones(self) -> None: # Compare with expected output (Which is the same as the output without the faulty smiles) self.assertTrue(are_equal(EXPECTED_OUTPUT, matrix)) - def test_caching(self) -> None: """Test if the caching gives the same results and is faster on the second run.""" @@ -281,7 +280,6 @@ def test_caching(self) -> None: class PipelineCompatibilityTest(unittest.TestCase): """Test if the pipeline is compatible with other sklearn functionalities.""" - def test_gridsearchcv(self) -> None: """Test if the MolPipeline can be used in sklearn's GridSearchCV.""" @@ -342,7 +340,6 @@ def test_gridsearchcv(self) -> None: for k, value in param_grid.items(): self.assertIn(grid_search_cv.best_params_[k], value) - def test_gridsearch_cache(self) -> None: """Run a short GridSearchCV and check if the caching and not caching gives the same results.""" h_params = { @@ -382,7 +379,6 @@ def test_gridsearch_cache(self) -> None: self.assertEqual(best_param_dict[True], best_param_dict[False]) self.assertTrue(np.allclose(prediction_dict[True], prediction_dict[False])) - def test_calibrated_classifier(self) -> None: """Test if the pipeline can be used with a CalibratedClassifierCV.""" smi2mol = SmilesToMol()