test

tests/test_helpers.py

@@ -1,4 +1,5 @@
 import json
+import subprocess
 import tempfile
 from pathlib import Path
 
@@ -64,6 +65,37 @@ def update(self, _, event: Event) -> None:
     assert e0.metadata == ModelMetadata.from_dict(md)
 
 
+def test_push_pull_chunked(tmp_path, target):
+    omlmd = Helper()
+
+    md = {
+        "name": "mnist",
+        "description": "Lorem ipsum",
+        "author": "John Doe",
+        "accuracy": 0.987,
+    }
+    temp = Path(tempfile.NamedTemporaryFile(suffix=".onnx", delete=False).name)
+    base_size = 16 * 1024 * 1024 * 3  # 48MB
+    try:
+        subprocess.run(
+            [
+                "dd",
+                "if=/dev/null",
+                f"of={temp}",
+                "bs=1",
+                "count=0",
+                f"seek={base_size}",
+            ],
+        )
+
+        omlmd.push(target, temp, **md)
+        omlmd.pull(target, tmp_path)
+        assert len(list(tmp_path.iterdir())) == 3
+        assert tmp_path.joinpath(temp.name).stat().st_size == base_size
+    finally:
+        temp.unlink()
+
+
 @pytest.mark.e2e
 def test_e2e_push_pull(tmp_path, target):
     omlmd = Helper()