thefuck/tests/utils.py

from pathlib import Path
from thefuck import types
from thefuck.conf import DEFAULT_PRIORITY


def Command(script='', stdout='', stderr=''):
    return types.Command(script, stdout, stderr)


class Rule(types.Rule):
    def __init__(self, name='', match=lambda *_: True,
                 get_new_command=lambda *_: '',
                 enabled_by_default=True,
                 side_effect=None,
                 priority=DEFAULT_PRIORITY,
                 requires_output=True):
        super(Rule, self).__init__(name, match, get_new_command,
                                   enabled_by_default, side_effect,
                                   priority, requires_output)


class CorrectedCommand(types.CorrectedCommand):
    def __init__(self, script='', side_effect=None, priority=DEFAULT_PRIORITY):
        super(CorrectedCommand, self).__init__(
            script, side_effect, priority)


root = Path(__file__).parent.parent.resolve()