from pip._internal.vcs.mercurial import Mercurial
from tests.lib import _create_test_package


def test_get_repository_root(script):
    version_pkg_path = _create_test_package(script, vcs="hg")
    tests_path = version_pkg_path.joinpath("tests")
    tests_path.mkdir()

    root1 = Mercurial.get_repository_root(version_pkg_path)
    assert root1 == version_pkg_path

    root2 = Mercurial.get_repository_root(version_pkg_path.joinpath("tests"))
    assert root2 == version_pkg_path

from pip._internal.vcs.mercurial import Mercurial
from tests.lib import _create_test_package, need_mercurial


@need_mercurial
def test_get_repository_root(script):
    version_pkg_path = _create_test_package(script, vcs="hg")
    tests_path = version_pkg_path.joinpath("tests")
    tests_path.mkdir()

    root1 = Mercurial.get_repository_root(version_pkg_path)
    assert root1 == version_pkg_path

    root2 = Mercurial.get_repository_root(version_pkg_path.joinpath("tests"))
    assert root2 == version_pkg_path

Add marker to Mercurial test