feat: Unit tests for serialize_v1 for folders

.github/workflows/unit_tests.yml

@@ -30,4 +30,7 @@ jobs:
       run: |
         set -euo pipefail
         source venv/bin/activate
-        pytest -v --full-trace .
+        # NOTE: option --full-trace may be useful for troubleshooting.
+        # TODO(#68): Remove the need to create this folder.
+        mkdir testdata
+        pytest -v .

model_signing/serialize.py

@@ -119,6 +119,13 @@ def validate_signature_path(model_path: Path, sig_path: Path):
         raise ValueError(f"{sig_path} must be in the folder root")
 
 
+def is_relative_to(p: Path, path_list: [Path]) -> bool:
+    for e in path_list:
+        if p.is_relative_to(e):
+            return True
+    return False
+
+
 # TODO(): add a context "AI model"?
 class Serializer:
     @staticmethod
@@ -137,7 +144,7 @@ def _ordered_files(path: Path, ignorepaths: [Path]) -> []:
         filtered = []
         total_size = 0
         for child in children:
-            if child in ignorepaths:
+            if is_relative_to(child, ignorepaths):
                 continue
 
             # To avoid bugs where we read the link rather than its target,
@@ -287,6 +294,9 @@ def _serialize_v1(path: Path, chunk: int, shard: int, signature_path: Path,
         if not allow_symlinks and path.is_symlink():
             raise ValueError(f"{str(path)} is a symlink")
 
+        if chunk < 0:
+            raise ValueError(f"{str(chunk)} is invalid")
+
         if not path.is_file() and not path.is_dir():
             raise ValueError(f"{str(path)} is not a dir or file")