from scripttease.library.overlays.common import *


def test_python_pip():
    c = python_pip("Pillow")
    assert "pip install -y Pillow" in c.get_statement()

    c = python_pip("Pillow", upgrade=True)
    assert "--upgrade" in c.get_statement()

    c = python_pip("Pillow", venv="python")
    assert "source python/bin/activate" in c.get_statement()


def test_python_virtual_env():
    c = python_virtualenv()
    assert "virtualenv python" in c.get_statement()