diff --git a/tests/python/test_cross_platform_parity.py b/tests/python/test_cross_platform_parity.py index 1262b20af9..384dc858bc 100644 --- a/tests/python/test_cross_platform_parity.py +++ b/tests/python/test_cross_platform_parity.py @@ -475,17 +475,15 @@ class TestPinnedRocmLeafDigitParity: def test_install_ps1_pinned_reroute_requires_rocm_digit(self): text = INSTALL_PS1.read_text(encoding = "utf-8") # The pinned gfx*/rocm reroute must require a digit after rocm. - assert re.search( - r"\$_pinLeaf -like 'gfx\*' -or \$_pinLeaf -match '\^rocm\\d'", text - ), ( + assert re.search(r"\$_pinLeaf -like 'gfx\*' -or \$_pinLeaf -match '\^rocm\\d'", text), ( "install.ps1 pinned-index reroute must use -match '^rocm\\d' (not a bare " "-like 'rocm*'), so rocm-current / rocm-rel-* fall through to the verbatim " "install instead of the ROCm --default-index path" ) # The broad glob must be gone from that reroute. - assert "-like 'rocm*'" not in text, ( - "install.ps1 must not route a pinned index on a bare -like 'rocm*' glob" - ) + assert ( + "-like 'rocm*'" not in text + ), "install.ps1 must not route a pinned index on a bare -like 'rocm*' glob" def test_install_sh_repairable_requires_rocm_digit(self): text = INSTALL_SH.read_text(encoding = "utf-8") diff --git a/tests/python/test_install_python_stack.py b/tests/python/test_install_python_stack.py index d2ebf51eee..a991c09169 100644 --- a/tests/python/test_install_python_stack.py +++ b/tests/python/test_install_python_stack.py @@ -166,9 +166,15 @@ class TestPinnedIndexClearsUvEnv: def test_pinned_index_url_strips_uv_index_vars(self): cmd = [ - "uv", "pip", "install", "--force-reinstall", - "torch", "torchvision", "torchaudio", - "--index-url", "https://download.pytorch.org/whl/cu128", + "uv", + "pip", + "install", + "--force-reinstall", + "torch", + "torchvision", + "torchaudio", + "--index-url", + "https://download.pytorch.org/whl/cu128", ] with mock.patch.dict( os.environ,