started adding tests for code in uv.py module

Comfy-Org · Aug 2, 2024 · a7866ad · a7866ad
1 parent 5baac22
commit a7866ad
Show file tree

Hide file tree

Showing 4 changed files with 21 additions and 0 deletions.
diff --git a/tests/uv/mock_requirements/core_reqs.txt b/tests/uv/mock_requirements/core_reqs.txt
@@ -0,0 +1 @@
+tqdm==4.66.4
diff --git a/tests/uv/mock_requirements/x_reqs.txt b/tests/uv/mock_requirements/x_reqs.txt
@@ -0,0 +1,3 @@
+numpy>=2.0.1
+sympy<=1.10.1
+tqdm==1.0
diff --git a/tests/uv/mock_requirements/y_reqs.txt b/tests/uv/mock_requirements/y_reqs.txt
@@ -0,0 +1,3 @@
+numpy<=1.5.0
+sympy>=1.13.1
+tqdm==2.0.0
diff --git a/tests/uv/test_uv.py b/tests/uv/test_uv.py
@@ -0,0 +1,14 @@
+from pathlib import Path
+
+from comfy_cli.uv import DependencyCompiler
+
+testsDir = Path(__file__).parent.resolve()
+temp = testsDir / "temp"
+temp.mkdir(exist_ok=True)
+here = Path(__file__).resolve()
+
+depComp = DependencyCompiler(
+    cwd=temp,
+    reqFilesCore=[here / "mock_requirements/core_reqs.txt"],
+    reqFilesExt=[here / "mock_requirements/x_reqs.txt", here / "mock_requirements/y_reqs.txt"],
+)