def test_no_cut(self) -> None:
"""Test not removing a locked package based on direct dependencies."""
tf = PackageVersion(
name="tensorflow",
version="==1.9.0",
index=Source("https://pypi.org/simple"),
develop=False,
)
context = flexmock(project=Project.from_strings(self._CASE_PIPFILE_LOCKED))
with CutLockedSieve.assigned_context(context):
sieve = CutLockedSieve()
assert list(sieve.run(p for p in [tf])) == [tf]
def test_noop_dev(self) -> None:
"""Test no operation if dependencies are not locked."""
pytest = PackageVersion(
name="pytest",
version="==5.3.1",
index=Source("https://pypi.org/simple"),
develop=False,
)
context = flexmock(project=Project.from_strings(self._CASE_PIPFILE_NOT_LOCKED))
with CutLockedSieve.assigned_context(context):
sieve = CutLockedSieve()
assert list(sieve.run(p for p in [pytest])) == [pytest]
