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()