<commit_before>class MockUtility:
    def __init__(self, utility_return):
        self.utility_return = utility_return

    def __call__(self, *args):
        return self.utility_return


class MockClassifier:
    """
    Mock classifier object for testing. The predict_proba method returns the
    object given for argument predict_proba_return.
    """
    def __init__(
            self, predict_proba_return=None, calculate_utility_return=None, predict_return=None,
            score_return=None
    ):
        self.calculate_utility_return = calculate_utility_return
        self.predict_return = predict_return
        self.predict_proba_return = predict_proba_return
        self.score_return = score_return

    def calculate_utility(self, data):
        return self.calculate_utility_return

    def predict(self, data):
        return self.predict_return

    def predict_proba(self, data):
        return self.predict_proba_return

    def score(self, data):
        return self.score_return<commit_msg>tests.mock.MockClassifier.score() arguments fixed<commit_after>class MockUtility:
    def __init__(self, utility_return):
        self.utility_return = utility_return

    def __call__(self, *args):
        return self.utility_return


class MockClassifier:
    """
    Mock classifier object for testing. The predict_proba method returns the
    object given for argument predict_proba_return.
    """
    def __init__(
            self, predict_proba_return=None, calculate_utility_return=None, predict_return=None,
            score_return=None
    ):
        self.calculate_utility_return = calculate_utility_return
        self.predict_return = predict_return
        self.predict_proba_return = predict_proba_return
        self.score_return = score_return

    def calculate_utility(self, data):
        return self.calculate_utility_return

    def predict(self, data):
        return self.predict_return

    def predict_proba(self, data):
        return self.predict_proba_return

    def score(self, X, y, sample_weight=None):
        return self.score_return
{'fraction_non_alphanumeric': 0.06525911708253358, 'fraction_numerical': 0.0, 'mean_word_length': 3.2378048780487805, 'pattern_counts': {'":': 0, '<': 3, '<?xml version=': 0, '>': 3, 'https://': 0, 'lorem ipsum': 0, 'www.': 0, 'xml': 0}, 'pii_count': 0, 'substrings_counts': 0, 'word_list_counts': {'cursed_substrings.json': 0, 'profanity_word_list.json': 0, 'sexual_word_list.json': 0, 'zh_pornsignals.json': 0}}
{'dir': 'git-commits-cleaned', 'id': '4066310', 'n_tokens_mistral': 594, 'n_tokens_neox': 581, 'n_words': 131}