def test_propose_many_values_no_duplicates(self, mock__check_proposals): """Test that ``propose`` method calls it's child implemented method with more than one proposals and ``allow_duplicates`` as ``False``. """ # setup instance = MagicMock() inverse_return = instance.tunable.inverse_transform.return_value inverse_return.to_dict.return_value = [1, 2] instance._propose = MagicMock(return_value=2) # run result = BaseTuner.propose(instance, 2) # assert instance._propose.assert_called_once_with(2, False) instance.tunable.inverse_transform.called_once_with(2) inverse_return.to_dict.assert_called_once_with(orient='records') assert result == [1, 2]
def test_propose_one_value_allow_duplicates(self, mock__check_proposals): """Test that ``propose`` method calls it's child implemented method with ``allow_duplicates`` as ``True``. """ # setup instance = MagicMock() inverse_return = instance.tunable.inverse_transform.return_value inverse_return.to_dict.return_value = [1] instance._propose = MagicMock(return_value=1) # run result = BaseTuner.propose(instance, 1, allow_duplicates=True) # assert instance._check_proposals.assert_not_called() instance._propose.assert_called_once_with(1, True) instance.tunable.inverse_transform.called_once_with(1) inverse_return.to_dict.assert_called_once_with(orient='records') assert result == 1