Skip to content

Commit 65148e4

Browse files
committed
test: remove unrelated GPU path tests, keep only AMD/ROCm tests
1 parent 9f58304 commit 65148e4

1 file changed

Lines changed: 0 additions & 75 deletions

File tree

tests/comfy_cli/test_install_python_resolution.py

Lines changed: 0 additions & 75 deletions
Original file line numberDiff line numberDiff line change
@@ -160,78 +160,3 @@ def test_amd_linux_rocm_version_controls_url(self, tmp_path, rocm_version, expec
160160
cmd = _get_torch_install_cmd(mock_run.call_args_list)
161161
assert expected_url in cmd
162162

163-
def test_nvidia_linux_no_extra_index(self, tmp_path):
164-
repo_dir = str(tmp_path)
165-
(tmp_path / "requirements.txt").write_text("some-package\n")
166-
167-
with patch("comfy_cli.command.install.subprocess.run", return_value=MagicMock(returncode=0)) as mock_run:
168-
install.pip_install_comfyui_dependencies(
169-
repo_dir,
170-
gpu=GPU_OPTION.NVIDIA,
171-
plat=constants.OS.LINUX,
172-
cuda_version=constants.CUDAVersion.v12_6,
173-
skip_torch_or_directml=False,
174-
skip_requirement=False,
175-
python="/usr/bin/python",
176-
)
177-
178-
cmd = _get_torch_install_cmd(mock_run.call_args_list)
179-
assert "--extra-index-url" not in cmd
180-
assert "--index-url" not in cmd
181-
182-
def test_nvidia_windows_uses_extra_index_url(self, tmp_path):
183-
repo_dir = str(tmp_path)
184-
(tmp_path / "requirements.txt").write_text("some-package\n")
185-
186-
with patch("comfy_cli.command.install.subprocess.run", return_value=MagicMock(returncode=0)) as mock_run:
187-
install.pip_install_comfyui_dependencies(
188-
repo_dir,
189-
gpu=GPU_OPTION.NVIDIA,
190-
plat=constants.OS.WINDOWS,
191-
cuda_version=constants.CUDAVersion.v12_6,
192-
skip_torch_or_directml=False,
193-
skip_requirement=False,
194-
python="/usr/bin/python",
195-
)
196-
197-
cmd = _get_torch_install_cmd(mock_run.call_args_list)
198-
assert "--extra-index-url" in cmd
199-
assert "https://download.pytorch.org/whl/cu126" in cmd
200-
201-
def test_cpu_uses_extra_index_url(self, tmp_path):
202-
repo_dir = str(tmp_path)
203-
(tmp_path / "requirements.txt").write_text("some-package\n")
204-
205-
with patch("comfy_cli.command.install.subprocess.run", return_value=MagicMock(returncode=0)) as mock_run:
206-
install.pip_install_comfyui_dependencies(
207-
repo_dir,
208-
gpu=None,
209-
plat=constants.OS.LINUX,
210-
cuda_version=constants.CUDAVersion.v12_6,
211-
skip_torch_or_directml=False,
212-
skip_requirement=False,
213-
python="/usr/bin/python",
214-
)
215-
216-
cmd = _get_torch_install_cmd(mock_run.call_args_list)
217-
assert "--extra-index-url" in cmd
218-
assert "https://download.pytorch.org/whl/cpu" in cmd
219-
220-
def test_amd_windows_installs_directml_not_rocm(self, tmp_path):
221-
repo_dir = str(tmp_path)
222-
(tmp_path / "requirements.txt").write_text("some-package\n")
223-
224-
with patch("comfy_cli.command.install.subprocess.run", return_value=MagicMock(returncode=0)) as mock_run:
225-
install.pip_install_comfyui_dependencies(
226-
repo_dir,
227-
gpu=GPU_OPTION.AMD,
228-
plat=constants.OS.WINDOWS,
229-
cuda_version=constants.CUDAVersion.v12_6,
230-
skip_torch_or_directml=False,
231-
skip_requirement=False,
232-
python="/usr/bin/python",
233-
)
234-
235-
cmds = [c[0][0] for c in mock_run.call_args_list]
236-
assert any("torch-directml" in c for c in cmds)
237-
assert not any("--index-url" in c for c in cmds)

0 commit comments

Comments
 (0)