示例#1
0
 def test_function_calculate_mask(self, mock_full, mock_down, mock_or, mock_pad, mock_ams, mock_sar):
     """test the calling of methods correctly in this function"""
     fil = mock.Mock()
     test_archive = Archive()
     test_archive.apply_instead_of_training = True
     test_archive.calculate_mask(fil)
     mock_ams.assert_called()
     mock_sar.assert_called()
     mock_or.assert_called_with('rp1', 'aout0')
     mock_pad.assert_called_with('aout0', 'sarout')
     mock_down.assert_called_with(np.array([[1]]), 'aout1', 'aout2')
     mock_full.assert_called()