Coverage for biobb_pytorch / test / unitests / test_mdae / test_lrp.py: 83%

1# type: ignore 

2from biobb_common.tools import test_fixtures as fx 

3from biobb_pytorch.mdae.explainability.LRP import relevance_propagation 

4import numpy as np 

5 

6 

7class TestLRP: 

8 def setup_class(self): 

9 fx.test_setup(self, 'lrp') 

10 

11 def teardown_class(self): 

12 fx.test_teardown(self) 

13 

14 def test_lrp(self): 

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

16 

17 if 'output_results_npz_path' in self.paths and self.paths['output_results_npz_path']: 

18 assert fx.not_empty(self.paths['output_results_npz_path']) 

19 

20 results = np.load(self.paths['output_results_npz_path'], allow_pickle=True) 

21 assert 'global_importance' in results, "Results should contain global_importance scores" 

22 

23 if 'ref_output_results_npz_path' in self.paths: 

24 ref_results = np.load(self.paths['ref_output_results_npz_path'], allow_pickle=True) 

25 # Compare relevance scores structure 

26 if 'global_importance' in results and 'global_importance' in ref_results: 

27 assert results['global_importance'].shape == ref_results['global_importance'].shape, "Global importance scores should have matching shapes"