-
Notifications
You must be signed in to change notification settings - Fork 1.6k
/
test_python.py
39 lines (33 loc) · 1.18 KB
/
test_python.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
from textwrap import dedent
from prefect.software.pip import PipRequirement
from prefect.software.python import PythonEnvironment
class TestPythonEnvironment:
def test_init(self):
reqs = PythonEnvironment(pip_requirements=["foo", "bar>=2"])
assert reqs.pip_requirements == [
PipRequirement("foo"),
PipRequirement("bar>=2"),
]
def test_from_requirements_file(self, tmp_path):
reqs_file = tmp_path / "requirements.txt"
reqs_file.write_text(
dedent(
"""
foo
bar>=2
"""
)
)
reqs = PythonEnvironment.from_file(reqs_file)
assert reqs.pip_requirements == [
PipRequirement("foo"),
PipRequirement("bar>=2"),
]
def test_install_commands(self):
reqs = PythonEnvironment(pip_requirements=["foo", "bar>=2"])
commands = reqs.install_commands()
assert commands == [["pip", "install", "foo", "bar>=2"]]
def test_install_commands_empty(self):
reqs = PythonEnvironment(pip_requirements=[])
commands = reqs.install_commands()
assert commands == []