def test_summary_without_rfecv(self): # nuke the rfe_scores to test the other branch of _summarize_estimator del self.pipeline.rfe_scores summarize(self.temp_dir.name, self.pipeline) self.assertFalse('rfe_plot.pdf' in os.listdir(self.temp_dir.name)) self.assertFalse('rfe_plot.png' in os.listdir(self.temp_dir.name)) self.assertFalse('rfe_scores.tsv' in os.listdir(self.temp_dir.name))
def test_summary_without_rfecv(self): del self.pipeline.rfe_scores summarize(self.temp_dir.name, self.pipeline)
def test_summary_with_rfecv(self): summarize(self.temp_dir.name, self.pipeline) self.assertTrue('rfe_plot.pdf' in os.listdir(self.temp_dir.name)) self.assertTrue('rfe_plot.png' in os.listdir(self.temp_dir.name)) self.assertTrue('rfe_scores.tsv' in os.listdir(self.temp_dir.name))
def test_summary_with_rfecv(self): summarize(self.temp_dir.name, self.pipeline)