Coverage for biobb_pytorch/test/unitests/test_mdae/test_train_mdae.py: 100%

1# type: ignore 

2from biobb_common.tools import test_fixtures as fx 

3from biobb_pytorch.mdae.train_mdae import trainMDAE 

4 

5 

6class TestTrainMdae: 

7 def setup_class(self): 

8 fx.test_setup(self, 'train_mdae') 

9 

10 def teardown_class(self): 

11 fx.test_teardown(self) 

12 

13 def test_train_mdae(self): 

14 trainMDAE(properties=self.properties, **self.paths) 

15 assert fx.not_empty(self.paths['output_model_pth_path']) 

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

17 assert fx.not_empty(self.paths['output_performance_npz_path']) 

18 # assert fx.equal(self.paths['output_model_pth_path'], self.paths['ref_output_model_pth_path']) 

19 # assert fx.equal(self.paths['output_train_data_npz_path'], self.paths['ref_output_train_data_npz_path']) 

20 # assert fx.equal(self.paths['output_performance_npz_path'], self.paths['ref_output_performance_npz_path'])