<filename>tests/rules/test_go_run.py
import pytest

from thefuck.rules.go_run import get_new_command
from thefuck.rules.go_run import match
from thefuck.types import Command


@pytest.mark.parametrize(
    "command", [Command("go run foo", ""),
                Command("go run bar", "")])
def test_match(command):
    assert match(command)


@pytest.mark.parametrize(
    "command, new_command",
    [
        (Command("go run foo", ""), "go run foo.go"),
        (Command("go run bar", ""), "go run bar.go"),
    ],
)
def test_get_new_command(command, new_command):
    assert get_new_command(command) == new_command

<filename>tests/rules/test_go_run.py
import pytest

from thefuck.rules.go_run import get_new_command
from thefuck.rules.go_run import match
from thefuck.types import Command


@pytest.mark.parametrize(
    "command", [Command("go run foo", ""),
                Command("go run bar", "")])
def test_match(command):
    assert match(command)


@pytest.mark.parametrize(
    "command, new_command",
    [
        (Command("go run foo", ""), "go run foo.go"),
        (Command("go run bar", ""), "go run bar.go"),
    ],
)
def test_get_new_command(command, new_command):
    assert get_new_command(command) == new_command
