Coverage for biobb_ml/test/unitests/test_classification/test_decision_tree.py: 94%

1from biobb_common.tools import test_fixtures as fx 

2from biobb_ml.classification.decision_tree import decision_tree 

3import platform 

4 

5 

6class TestDecisionTree(): 

7 def setup_class(self): 

8 fx.test_setup(self, 'decision_tree') 

9 

10 def teardown_class(self): 

11 fx.test_teardown(self) 

12 pass 

13 

14 def test_decision_tree(self): 

15 decision_tree(properties=self.properties, **self.paths) 

16 assert fx.not_empty(self.paths['output_model_path']) 

17 if platform.system() == 'Darwin': 

18 assert fx.equal(self.paths['output_model_path'], self.paths['ref_output_model_path']) 

19 assert fx.not_empty(self.paths['output_test_table_path']) 

20 assert fx.equal(self.paths['output_test_table_path'], self.paths['ref_output_test_table_path']) 

21 assert fx.not_empty(self.paths['output_plot_path']) 

22 # assert fx.equal(self.paths['output_plot_path'], self.paths['ref_output_plot_path'])