def test_pipreqcheck_virtualenv_prompt(host):

    """

    Tests if Python virtual environment prompt has been set-up

    correctly.

    """

    with host.sudo("pipreqcheck"):

        prompt = host.run('bash -c "source /var/lib/pipreqcheck/virtualenv/bin/activate; printenv PS1"')

        # Chop off trailing newline if present (this is from the

        # host.run itself).

        if prompt.stdout.endswith("\n"):

            prompt_stdout = prompt.stdout[:-1]

        else:

            prompt_stdout = prompt.stdout

        assert prompt_stdout == "(pipreqcheck) "