truss 0.10.0rc1__py3-none-any.whl → 0.60.0__py3-none-any.whl

truss/tests/patch/test_calc_patch.py

@@ -1,10 +1,11 @@
+import os
 from pathlib import Path
 from typing import Any, Callable, List, Optional
 
+import pytest
 import yaml
-from truss.patch.calc_patch import calc_truss_patch, calc_unignored_paths
-from truss.patch.signature import calc_truss_signature
-from truss.server.control.patch.types import (
+from truss.base.truss_config import TrussConfig
+from truss.templates.control.control.helpers.custom_types import (
     Action,
     ConfigPatch,
     EnvVarPatch,
@@ -16,8 +17,13 @@ from truss.server.control.patch.types import (
     PythonRequirementPatch,
     SystemPackagePatch,
 )
-from truss.truss_config import TrussConfig
-from truss.truss_handle import TrussHandle
+from truss.truss_handle.patch.calc_patch import (
+    _calc_python_requirements_patches,
+    _calc_unignored_paths,
+    calc_truss_patch,
+)
+from truss.truss_handle.patch.signature import calc_truss_signature
+from truss.truss_handle.truss_handle import TrussHandle
 
 
 def test_calc_truss_patch_unsupported(custom_model_truss_dir: Path):
@@ -45,11 +51,7 @@ def test_calc_truss_patch_add_file(custom_model_truss_dir: Path):
     patch = patches[0]
     assert patch == Patch(
         type=PatchType.MODEL_CODE,
-        body=ModelCodePatch(
-            action=Action.ADD,
-            path="dummy",
-            content="content",
-        ),
+        body=ModelCodePatch(action=Action.ADD, path="dummy", content="content"),
     )
 
 
@@ -64,11 +66,7 @@ def test_calc_truss_patch_add_under_new_directory(custom_model_truss_dir: Path):
     patch = patches[0]
     assert patch == Patch(
         type=PatchType.MODEL_CODE,
-        body=ModelCodePatch(
-            action=Action.ADD,
-            path="dir/dummy",
-            content="",
-        ),
+        body=ModelCodePatch(action=Action.ADD, path="dir/dummy", content=""),
     )
 
 
@@ -81,10 +79,7 @@ def test_calc_truss_patch_remove_file(custom_model_truss_dir: Path):
     patch = patches[0]
     assert patch == Patch(
         type=PatchType.MODEL_CODE,
-        body=ModelCodePatch(
-            action=Action.REMOVE,
-            path="model.py",
-        ),
+        body=ModelCodePatch(action=Action.REMOVE, path="model.py"),
     )
 
 
@@ -118,10 +113,7 @@ def test_calc_truss_ignore_pycache(custom_model_truss_dir: Path):
     model_pycache_path.mkdir()
     (model_pycache_path / "foo.pyo").touch()
 
-    patches = calc_truss_patch(
-        custom_model_truss_dir,
-        prev_sign,
-    )
+    patches = calc_truss_patch(custom_model_truss_dir, prev_sign)
     assert len(patches) == 0
 
 
@@ -135,10 +127,7 @@ def test_calc_truss_ignore_pycache_existing(custom_model_truss_dir: Path):
     model_pycache_path.mkdir()
     (model_pycache_path / "foo.pyo").touch()
     sign = calc_truss_signature(custom_model_truss_dir)
-    patches = calc_truss_patch(
-        custom_model_truss_dir,
-        sign,
-    )
+    patches = calc_truss_patch(custom_model_truss_dir, sign)
     assert len(patches) == 0
 
 
@@ -153,19 +142,13 @@ def test_calc_truss_ignore_changes_outside_patch_relevant_dirs(
     git_dir.mkdir()
     (git_dir / "dummy").touch()
 
-    patches = calc_truss_patch(
-        custom_model_truss_dir,
-        prev_sign,
-    )
+    patches = calc_truss_patch(custom_model_truss_dir, prev_sign)
     assert len(patches) == 0
 
     # Removing should also be ignored
     new_sign = calc_truss_signature(custom_model_truss_dir)
     (git_dir / "dummy").unlink()
-    patches = calc_truss_patch(
-        custom_model_truss_dir,
-        new_sign,
-    )
+    patches = calc_truss_patch(custom_model_truss_dir, new_sign)
     assert len(patches) == 0
 
 
@@ -186,8 +169,7 @@ def test_calc_config_patches_add_python_requirement(custom_model_truss_dir: Path
         Patch(
             type=PatchType.PYTHON_REQUIREMENT,
             body=PythonRequirementPatch(
-                action=Action.ADD,
-                requirement="requests==1.0.0",
+                action=Action.ADD, requirement="requests==1.0.0"
             ),
         ),
     ]
@@ -204,11 +186,7 @@ def test_calc_truss_patch_add_package(custom_model_truss_dir: Path):
     patch = patches[0]
     assert patch == Patch(
         type=PatchType.PACKAGE,
-        body=PackagePatch(
-            action=Action.ADD,
-            path="dir/dummy",
-            content="",
-        ),
+        body=PackagePatch(action=Action.ADD, path="dir/dummy", content=""),
     )
 
 
@@ -228,10 +206,7 @@ def test_calc_truss_patch_remove_package(
     patch = patches[0]
     assert patch == Patch(
         type=PatchType.PACKAGE,
-        body=PackagePatch(
-            action=Action.REMOVE,
-            path="test_package/test.py",
-        ),
+        body=PackagePatch(action=Action.REMOVE, path="test_package/test.py"),
     )
 
 
@@ -262,6 +237,267 @@ def test_calc_truss_patch_update_package(
     )
 
 
+def test_calc_truss_patch_handles_requirements_file_name_change(
+    custom_model_truss_dir: Path,
+):
+    requirements_contents = """xformers"""
+
+    def pre_config_op(config: TrussConfig):
+        filename = "requirement.txt"
+        config.requirements.clear()
+        config.requirements_file = filename
+        with (custom_model_truss_dir / filename).open("w") as req_file:
+            req_file.write(requirements_contents)
+
+    def config_op(config: TrussConfig):
+        filename = "requirements.txt"
+        with (custom_model_truss_dir / filename).open("w") as req_file:
+            req_file.write(requirements_contents)
+        config.requirements_file = filename
+        config.requirements.clear()
+
+    patches = _apply_config_change_and_calc_patches(
+        custom_model_truss_dir, config_op=config_op, config_pre_op=pre_config_op
+    )
+    assert len(patches) == 1
+    assert patches == [
+        Patch(
+            type=PatchType.CONFIG,
+            body=ConfigPatch(
+                action=Action.UPDATE,
+                config=yaml.safe_load((custom_model_truss_dir / "config.yaml").open()),
+            ),
+        )
+    ]
+
+
+def test_calc_truss_patch_handles_requirements_comments(custom_model_truss_dir: Path):
+    def pre_config_op(config: TrussConfig):
+        requirements_contents = """xformers\n#torch==2.0.1"""
+        filename = "./requirements.txt"
+        config.requirements_file = filename
+        with (custom_model_truss_dir / filename).open("w") as req_file:
+            req_file.write(requirements_contents)
+
+    def config_op(config: TrussConfig):
+        requirements_contents = """#xformers\ntorch==2.3.1\n"""
+        filename = "requirements.txt"
+        with (custom_model_truss_dir / filename).open("w") as req_file:
+            req_file.write(requirements_contents)
+
+    patches = _apply_config_change_and_calc_patches(
+        custom_model_truss_dir, config_op=config_op, config_pre_op=pre_config_op
+    )
+    assert len(patches) == 2
+    assert patches == [
+        Patch(
+            type=PatchType.PYTHON_REQUIREMENT,
+            body=PythonRequirementPatch(action=Action.REMOVE, requirement="xformers"),
+        ),
+        Patch(
+            type=PatchType.PYTHON_REQUIREMENT,
+            body=PythonRequirementPatch(action=Action.ADD, requirement="torch==2.3.1"),
+        ),
+    ]
+
+
+def test_calc_truss_patch_handles_requirements_file_changes(
+    custom_model_truss_dir: Path,
+):
+    def pre_config_op(config: TrussConfig):
+        requirements_contents = """xformers\ntorch==2.0.1"""
+        filename = "./requirements.txt"
+        config.requirements_file = filename
+        with (custom_model_truss_dir / filename).open("w") as req_file:
+            req_file.write(requirements_contents)
+
+    def config_op(config: TrussConfig):
+        requirements_contents = """requests\ntorch==2.3.1\n"""
+        filename = "requirements.txt"
+        with (custom_model_truss_dir / filename).open("w") as req_file:
+            req_file.write(requirements_contents)
+
+    patches = _apply_config_change_and_calc_patches(
+        custom_model_truss_dir, config_op=config_op, config_pre_op=pre_config_op
+    )
+    assert len(patches) == 3
+    assert patches == [
+        # In this case, a Config Update patch is not issued. This does not cause issues on the backend
+        Patch(
+            type=PatchType.PYTHON_REQUIREMENT,
+            body=PythonRequirementPatch(action=Action.REMOVE, requirement="xformers"),
+        ),
+        Patch(
+            type=PatchType.PYTHON_REQUIREMENT,
+            body=PythonRequirementPatch(action=Action.ADD, requirement="requests"),
+        ),
+        Patch(
+            type=PatchType.PYTHON_REQUIREMENT,
+            body=PythonRequirementPatch(
+                action=Action.UPDATE, requirement="torch==2.3.1"
+            ),
+        ),
+    ]
+
+
+def test_calc_truss_patch_handles_requirements_file_changes_and_config_changes(
+    custom_model_truss_dir: Path,
+):
+    def pre_config_op(config: TrussConfig):
+        requirements_contents = """xformers\ntorch==2.0.1"""
+        filename = "./requirements.txt"
+        config.requirements_file = filename
+        with (custom_model_truss_dir / filename).open("w") as req_file:
+            req_file.write(requirements_contents)
+
+    def config_op(config: TrussConfig):
+        requirements_contents = """requests\ntorch==2.3.1"""
+        filename = "requirement.txt"
+        with (custom_model_truss_dir / filename).open("w") as req_file:
+            req_file.write(requirements_contents)
+        config.requirements_file = filename
+
+    patches = _apply_config_change_and_calc_patches(
+        custom_model_truss_dir, config_op=config_op, config_pre_op=pre_config_op
+    )
+    assert len(patches) == 4
+    assert patches == [
+        Patch(
+            type=PatchType.CONFIG,
+            body=ConfigPatch(
+                action=Action.UPDATE,
+                config=yaml.safe_load((custom_model_truss_dir / "config.yaml").open()),
+            ),
+        ),
+        Patch(
+            type=PatchType.PYTHON_REQUIREMENT,
+            body=PythonRequirementPatch(action=Action.REMOVE, requirement="xformers"),
+        ),
+        Patch(
+            type=PatchType.PYTHON_REQUIREMENT,
+            body=PythonRequirementPatch(action=Action.ADD, requirement="requests"),
+        ),
+        Patch(
+            type=PatchType.PYTHON_REQUIREMENT,
+            body=PythonRequirementPatch(
+                action=Action.UPDATE, requirement="torch==2.3.1"
+            ),
+        ),
+    ]
+
+
+def test_calc_truss_patch_handles_requirements_file_removal(
+    custom_model_truss_dir: Path,
+):
+    requirements_contents = """xformers"""
+    filename = "requirements.txt"
+
+    def pre_config_op(config: TrussConfig):
+        with (custom_model_truss_dir / filename).open("w") as req_file:
+            req_file.write(requirements_contents)
+        config.requirements_file = filename
+        config.requirements.clear()
+
+    def config_op(config: TrussConfig):
+        requirements_contents = ["xformers", "requests"]
+        config.requirements.extend(requirements_contents)
+        config.requirements_file = ""
+        os.remove(custom_model_truss_dir / filename)
+
+    patches = _apply_config_change_and_calc_patches(
+        custom_model_truss_dir, config_op=config_op, config_pre_op=pre_config_op
+    )
+    assert len(patches) == 2
+    assert patches == [
+        Patch(
+            type=PatchType.CONFIG,
+            body=ConfigPatch(
+                action=Action.UPDATE,
+                config=yaml.safe_load((custom_model_truss_dir / "config.yaml").open()),
+            ),
+        ),
+        Patch(
+            type=PatchType.PYTHON_REQUIREMENT,
+            body=PythonRequirementPatch(action=Action.ADD, requirement="requests"),
+        ),
+    ]
+
+
+def test_calc_truss_signature_raises_for_invalid_requirements_file(
+    custom_model_truss_dir: Path,
+):
+    config_path = custom_model_truss_dir / "config.yaml"
+    config = TrussConfig.from_yaml(config_path)
+    config.requirements.clear()
+    config.requirements_file = "no_exist.txt"
+    config.write_to_yaml_file(config_path)
+
+    with pytest.raises(FileNotFoundError):
+        calc_truss_signature(custom_model_truss_dir)
+
+
+def test_calc_truss_patch_handles_requirements_file_added_no_change(
+    custom_model_truss_dir: Path,
+):
+    requirements_contents = """xformers"""
+
+    def pre_config_op(config: TrussConfig):
+        config.requirements.append(requirements_contents)
+
+    def config_op(config: TrussConfig):
+        filename = "requirements.txt"
+        with (custom_model_truss_dir / filename).open("w") as req_file:
+            req_file.write(requirements_contents)
+        config.requirements_file = filename
+        config.requirements.clear()
+
+    patches = _apply_config_change_and_calc_patches(
+        custom_model_truss_dir, config_op=config_op, config_pre_op=pre_config_op
+    )
+    assert len(patches) == 1
+    assert patches == [
+        Patch(
+            type=PatchType.CONFIG,
+            body=ConfigPatch(
+                action=Action.UPDATE,
+                config=yaml.safe_load((custom_model_truss_dir / "config.yaml").open()),
+            ),
+        )
+    ]
+
+
+def test_calc_truss_patch_handles_requirements_file_added_with_changes(
+    custom_model_truss_dir: Path,
+):
+    requirements_contents = """xformers"""
+
+    def config_op(config: TrussConfig):
+        filename = "requirements.txt"
+        with (custom_model_truss_dir / filename).open("w") as req_file:
+            req_file.write(requirements_contents)
+        config.requirements_file = filename
+        config.requirements.clear()
+
+    patches = _apply_config_change_and_calc_patches(
+        custom_model_truss_dir, config_op=config_op
+    )
+
+    assert len(patches) == 2
+    assert patches == [
+        Patch(
+            type=PatchType.CONFIG,
+            body=ConfigPatch(
+                action=Action.UPDATE,
+                config=yaml.safe_load((custom_model_truss_dir / "config.yaml").open()),
+            ),
+        ),
+        Patch(
+            type=PatchType.PYTHON_REQUIREMENT,
+            body=PythonRequirementPatch(action=Action.ADD, requirement="xformers"),
+        ),
+    ]
+
+
 def test_calc_config_patches_remove_python_requirement(custom_model_truss_dir: Path):
     patches = _apply_config_change_and_calc_patches(
         custom_model_truss_dir,
@@ -279,10 +515,7 @@ def test_calc_config_patches_remove_python_requirement(custom_model_truss_dir: P
         ),
         Patch(
             type=PatchType.PYTHON_REQUIREMENT,
-            body=PythonRequirementPatch(
-                action=Action.REMOVE,
-                requirement="requests",
-            ),
+            body=PythonRequirementPatch(action=Action.REMOVE, requirement="requests"),
         ),
     ]
 
@@ -297,6 +530,39 @@ def test_calc_config_patches_update_python_requirement(custom_model_truss_dir: P
         config_op=update_requests_version,
     )
     assert len(patches) == 2
+    assert patches == [
+        Patch(
+            type=PatchType.CONFIG,
+            body=ConfigPatch(
+                action=Action.UPDATE,
+                config=yaml.safe_load((custom_model_truss_dir / "config.yaml").open()),
+            ),
+        ),
+        Patch(
+            type=PatchType.PYTHON_REQUIREMENT,
+            body=PythonRequirementPatch(
+                action=Action.UPDATE, requirement="requests==2.0.0"
+            ),
+        ),
+    ]
+
+
+def test_calc_config_patches_update_python_url_requirement_with_egg_tag(
+    custom_model_truss_dir: Path,
+):
+    def update_requests_version(config: TrussConfig):
+        config.requirements[0] = (
+            "git+https://github.com/huggingface/transformers.git#egg=transformers"
+        )
+
+    patches = _apply_config_change_and_calc_patches(
+        custom_model_truss_dir,
+        config_pre_op=lambda config: config.requirements.append(
+            "git+https://github.com/huggingface/transformers.git"
+        ),
+        config_op=update_requests_version,
+    )
+    assert len(patches) == 2
     assert patches == [
         Patch(
             type=PatchType.CONFIG,
@@ -309,33 +575,91 @@ def test_calc_config_patches_update_python_requirement(custom_model_truss_dir: P
             type=PatchType.PYTHON_REQUIREMENT,
             body=PythonRequirementPatch(
                 action=Action.UPDATE,
-                requirement="requests==2.0.0",
+                requirement="git+https://github.com/huggingface/transformers.git#egg=transformers",
             ),
         ),
     ]
 
 
-def test_calc_config_patches_add_remove_and_update_python_requirement(
+def test_calc_config_patches_update_python_url_requirement_with_revision_tag(
+    custom_model_truss_dir: Path,
+):
+    def update_requests_version(config: TrussConfig):
+        config.requirements[0] = (
+            "git+https://github.com/huggingface/transformers.git@111111"
+        )
+
+    patches = _apply_config_change_and_calc_patches(
+        custom_model_truss_dir,
+        config_pre_op=lambda config: config.requirements.append(
+            "git+https://github.com/huggingface/transformers.git@000000"
+        ),
+        config_op=update_requests_version,
+    )
+    assert len(patches) == 2
+    assert patches == [
+        Patch(
+            type=PatchType.CONFIG,
+            body=ConfigPatch(
+                action=Action.UPDATE,
+                config=yaml.safe_load((custom_model_truss_dir / "config.yaml").open()),
+            ),
+        ),
+        Patch(
+            type=PatchType.PYTHON_REQUIREMENT,
+            body=PythonRequirementPatch(
+                action=Action.UPDATE,
+                requirement="git+https://github.com/huggingface/transformers.git@111111",
+            ),
+        ),
+    ]
+
+
+def test_calc_config_patches_ignores_removal_of_url_based_requirements_without_egg_tag(
     custom_model_truss_dir: Path,
 ):
     def config_pre_op(config: TrussConfig):
         config.requirements = [
             "requests==1.0.0",
-            "jinja==4.0.0",
+            "git+https://github.com/huggingface/transformers.git@111111",
         ]
 
     def config_op(config: TrussConfig):
+        config.requirements = ["requests==1.0.0"]
+
+    patches = _apply_config_change_and_calc_patches(
+        custom_model_truss_dir, config_pre_op=config_pre_op, config_op=config_op
+    )
+    assert len(patches) == 1
+    assert patches[0] == Patch(
+        type=PatchType.CONFIG,
+        body=ConfigPatch(
+            action=Action.UPDATE,
+            config=yaml.safe_load((custom_model_truss_dir / "config.yaml").open()),
+        ),
+    )
+    patches = patches[1:]
+    patches.sort(key=lambda patch: patch.body.requirement)
+    assert patches == []
+
+
+def test_calc_config_patches_add_remove_and_update_python_requirement(
+    custom_model_truss_dir: Path,
+):
+    def config_pre_op(config: TrussConfig):
         config.requirements = [
-            "requests==2.0.0",
-            "numpy>=1.8",
+            "requests==1.0.0",
+            "jinja==4.0.0",
+            "git+https://github.com/huggingface/transformers.git@111111#egg=transformers",
         ]
 
+    def config_op(config: TrussConfig):
+        config.requirements = ["requests==2.0.0", "numpy>=1.8"]
+
     patches = _apply_config_change_and_calc_patches(
-        custom_model_truss_dir,
-        config_pre_op=config_pre_op,
-        config_op=config_op,
+        custom_model_truss_dir, config_pre_op=config_pre_op, config_op=config_op
     )
-    assert len(patches) == 4
+    assert len(patches) == 5
     assert patches[0] == Patch(
         type=PatchType.CONFIG,
         body=ConfigPatch(
@@ -350,29 +674,27 @@ def test_calc_config_patches_add_remove_and_update_python_requirement(
             type=PatchType.PYTHON_REQUIREMENT,
             body=PythonRequirementPatch(
                 action=Action.REMOVE,
-                requirement="jinja",
+                requirement="git+https://github.com/huggingface/transformers.git@111111#egg=transformers",
             ),
         ),
         Patch(
             type=PatchType.PYTHON_REQUIREMENT,
-            body=PythonRequirementPatch(
-                action=Action.ADD,
-                requirement="numpy>=1.8",
-            ),
+            body=PythonRequirementPatch(action=Action.REMOVE, requirement="jinja"),
+        ),
+        Patch(
+            type=PatchType.PYTHON_REQUIREMENT,
+            body=PythonRequirementPatch(action=Action.ADD, requirement="numpy>=1.8"),
         ),
         Patch(
             type=PatchType.PYTHON_REQUIREMENT,
             body=PythonRequirementPatch(
-                action=Action.UPDATE,
-                requirement="requests==2.0.0",
+                action=Action.UPDATE, requirement="requests==2.0.0"
             ),
         ),
     ]
 
 
-def test_calc_config_patches_add_env_var(
-    custom_model_truss_dir: Path,
-):
+def test_calc_config_patches_add_env_var(custom_model_truss_dir: Path):
     patches = _apply_config_change_and_calc_patches(
         custom_model_truss_dir,
         config_op=lambda config: config.environment_variables.update({"foo": "bar"}),
@@ -388,17 +710,12 @@ def test_calc_config_patches_add_env_var(
         ),
         Patch(
             type=PatchType.ENVIRONMENT_VARIABLE,
-            body=EnvVarPatch(
-                action=Action.ADD,
-                item={"foo": "bar"},
-            ),
+            body=EnvVarPatch(action=Action.ADD, item={"foo": "bar"}),
         ),
     ]
 
 
-def test_calc_config_patches_add_remove_env_var(
-    custom_model_truss_dir: Path,
-):
+def test_calc_config_patches_add_remove_env_var(custom_model_truss_dir: Path):
     patches = _apply_config_change_and_calc_patches(
         custom_model_truss_dir,
         config_pre_op=lambda config: config.environment_variables.update(
@@ -417,18 +734,14 @@ def test_calc_config_patches_add_remove_env_var(
         ),
         Patch(
             type=PatchType.ENVIRONMENT_VARIABLE,
-            body=EnvVarPatch(
-                action=Action.REMOVE,
-                item={"foo": "bar"},
-            ),
+            body=EnvVarPatch(action=Action.REMOVE, item={"foo": "bar"}),
         ),
     ]
 
 
 def test_calc_config_patches_add_system_package(custom_model_truss_dir: Path):
     patches = _apply_config_change_and_calc_patches(
-        custom_model_truss_dir,
-        lambda config: config.system_packages.append("curl"),
+        custom_model_truss_dir, lambda config: config.system_packages.append("curl")
     )
     assert len(patches) == 2
     assert patches == [
@@ -441,10 +754,7 @@ def test_calc_config_patches_add_system_package(custom_model_truss_dir: Path):
         ),
         Patch(
             type=PatchType.SYSTEM_PACKAGE,
-            body=SystemPackagePatch(
-                action=Action.ADD,
-                package="curl",
-            ),
+            body=SystemPackagePatch(action=Action.ADD, package="curl"),
         ),
     ]
 
@@ -466,10 +776,7 @@ def test_calc_config_patches_remove_system_package(custom_model_truss_dir: Path)
         ),
         Patch(
             type=PatchType.SYSTEM_PACKAGE,
-            body=SystemPackagePatch(
-                action=Action.REMOVE,
-                package="curl",
-            ),
+            body=SystemPackagePatch(action=Action.REMOVE, package="curl"),
         ),
     ]
 
@@ -478,21 +785,13 @@ def test_calc_config_patches_add_and_remove_system_package(
     custom_model_truss_dir: Path,
 ):
     def config_pre_op(config: TrussConfig):
-        config.system_packages = [
-            "curl",
-            "jq",
-        ]
+        config.system_packages = ["curl", "jq"]
 
     def config_op(config: TrussConfig):
-        config.system_packages = [
-            "curl",
-            "libsnd",
-        ]
+        config.system_packages = ["curl", "libsnd"]
 
     patches = _apply_config_change_and_calc_patches(
-        custom_model_truss_dir,
-        config_pre_op=config_pre_op,
-        config_op=config_op,
+        custom_model_truss_dir, config_pre_op=config_pre_op, config_op=config_op
     )
     assert len(patches) == 3
     assert patches[0] == Patch(
@@ -507,17 +806,11 @@ def test_calc_config_patches_add_and_remove_system_package(
     assert patches == [
         Patch(
             type=PatchType.SYSTEM_PACKAGE,
-            body=SystemPackagePatch(
-                action=Action.REMOVE,
-                package="jq",
-            ),
+            body=SystemPackagePatch(action=Action.REMOVE, package="jq"),
         ),
         Patch(
             type=PatchType.SYSTEM_PACKAGE,
-            body=SystemPackagePatch(
-                action=Action.ADD,
-                package="libsnd",
-            ),
+            body=SystemPackagePatch(action=Action.ADD, package="libsnd"),
         ),
     ]
 
@@ -552,9 +845,7 @@ def test_calc_config_patches_add_external_data(
         config.external_data = None
 
     patches = _apply_config_change_and_calc_patches(
-        path,
-        config_pre_op=config_pre_op,
-        config_op=config_op,
+        path, config_pre_op=config_pre_op, config_op=config_op
     )
     assert len(patches) == 2
 
@@ -583,10 +874,7 @@ def test_calc_config_patches_remove_external_data(
     def config_op(config: TrussConfig):
         config.external_data = None
 
-    patches = _apply_config_change_and_calc_patches(
-        path,
-        config_op=config_op,
-    )
+    patches = _apply_config_change_and_calc_patches(path, config_op=config_op)
     assert len(patches) == 2
     assert patches == [
         Patch(
@@ -606,11 +894,7 @@ def test_calc_config_patches_remove_external_data(
 
 
 def test_calc_unignored_paths():
-    ignore_patterns = [
-        ".mypy_cache/",
-        "venv/",
-        "*.tmp",
-    ]
+    ignore_patterns = [".mypy_cache/", "venv/", "*.tmp"]
 
     root_relative_paths = {
         ".mypy_cache/should_ignore.json",
@@ -620,18 +904,15 @@ def test_calc_unignored_paths():
         "model/model.py",
     }
 
-    unignored_paths = calc_unignored_paths(root_relative_paths, ignore_patterns)
-    assert unignored_paths == {
-        "config.yaml",
-        "model/model.py",
-    }
+    unignored_paths = _calc_unignored_paths(root_relative_paths, ignore_patterns)
+    assert unignored_paths == {"config.yaml", "model/model.py"}
 
 
 def _apply_config_change_and_calc_patches(
     custom_model_truss_dir: Path,
     config_op: Callable[[TrussConfig], Any],
     config_pre_op: Optional[Callable[[TrussConfig], Any]] = None,
-) -> Optional[List[Patch]]:
+) -> List[Patch]:
     def modify_config(op):
         config_path = custom_model_truss_dir / "config.yaml"
         config = TrussConfig.from_yaml(config_path)
@@ -644,3 +925,34 @@ def _apply_config_change_and_calc_patches(
     prev_sign = calc_truss_signature(custom_model_truss_dir)
     modify_config(config_op)
     return calc_truss_patch(custom_model_truss_dir, prev_sign)
+
+
+@pytest.mark.parametrize(
+    "desc, prev_raw_reqs, new_raw_reqs, expected_patches",
+    [
+        (
+            "handles single update",
+            ["requests==1.0.0"],
+            ["requests==2.0.0"],
+            [
+                Patch(
+                    type=PatchType.PYTHON_REQUIREMENT,
+                    body=PythonRequirementPatch(
+                        action=Action.UPDATE, requirement="requests==2.0.0"
+                    ),
+                )
+            ],
+        ),
+        (
+            "ignores url-based requirement without egg tag",
+            ["git+https://github.com/huggingface/transformers.git@000000"],
+            [],
+            [],
+        ),
+    ],
+)
+def test_calc_python_requirements_patches(
+    desc, prev_raw_reqs, new_raw_reqs, expected_patches
+):
+    actual = _calc_python_requirements_patches(prev_raw_reqs, new_raw_reqs)
+    assert actual == expected_patches, desc