Run Black on all of tools/
Signed-off-by: Edward Z. Yang <ezyangfb.com>
Pull Request resolved: https://github.com/pytorch/pytorch/pull/76089
Approved by: https://github.com/albanD
diff --git a/tools/test/test_cmake.py b/tools/test/test_cmake.py
index ecbce07..2c4bead 100644
--- a/tools/test/test_cmake.py
+++ b/tools/test/test_cmake.py
@@ -9,49 +9,60 @@
import tools.setup_helpers.cmake
-T = typing.TypeVar('T')
+T = typing.TypeVar("T")
class TestCMake(unittest.TestCase):
-
- @unittest.mock.patch('multiprocessing.cpu_count')
+ @unittest.mock.patch("multiprocessing.cpu_count")
def test_build_jobs(self, mock_cpu_count: unittest.mock.MagicMock) -> None:
"""Tests that the number of build jobs comes out correctly."""
mock_cpu_count.return_value = 13
cases = [
# MAX_JOBS, USE_NINJA, IS_WINDOWS, want
- (( '8', True, False), ['-j', '8']), # noqa: E201,E241
- (( None, True, False), None), # noqa: E201,E241
- (( '7', False, False), ['-j', '7']), # noqa: E201,E241
- (( None, False, False), ['-j', '13']), # noqa: E201,E241
- (( '6', True, True), ['-j', '6']), # noqa: E201,E241
- (( None, True, True), None), # noqa: E201,E241
- (( '11', False, True), ['/p:CL_MPCount=11']), # noqa: E201,E241
- (( None, False, True), ['/p:CL_MPCount=13']), # noqa: E201,E241
+ (("8", True, False), ["-j", "8"]), # noqa: E201,E241
+ ((None, True, False), None), # noqa: E201,E241
+ (("7", False, False), ["-j", "7"]), # noqa: E201,E241
+ ((None, False, False), ["-j", "13"]), # noqa: E201,E241
+ (("6", True, True), ["-j", "6"]), # noqa: E201,E241
+ ((None, True, True), None), # noqa: E201,E241
+ (("11", False, True), ["/p:CL_MPCount=11"]), # noqa: E201,E241
+ ((None, False, True), ["/p:CL_MPCount=13"]), # noqa: E201,E241
]
for (max_jobs, use_ninja, is_windows), want in cases:
- with self.subTest(MAX_JOBS=max_jobs, USE_NINJA=use_ninja, IS_WINDOWS=is_windows):
+ with self.subTest(
+ MAX_JOBS=max_jobs, USE_NINJA=use_ninja, IS_WINDOWS=is_windows
+ ):
with contextlib.ExitStack() as stack:
- stack.enter_context(env_var('MAX_JOBS', max_jobs))
- stack.enter_context(unittest.mock.patch.object(tools.setup_helpers.cmake, 'USE_NINJA', use_ninja))
- stack.enter_context(unittest.mock.patch.object(tools.setup_helpers.cmake, 'IS_WINDOWS', is_windows))
+ stack.enter_context(env_var("MAX_JOBS", max_jobs))
+ stack.enter_context(
+ unittest.mock.patch.object(
+ tools.setup_helpers.cmake, "USE_NINJA", use_ninja
+ )
+ )
+ stack.enter_context(
+ unittest.mock.patch.object(
+ tools.setup_helpers.cmake, "IS_WINDOWS", is_windows
+ )
+ )
cmake = tools.setup_helpers.cmake.CMake()
- with unittest.mock.patch.object(cmake, 'run') as cmake_run:
+ with unittest.mock.patch.object(cmake, "run") as cmake_run:
cmake.build({})
cmake_run.assert_called_once()
- call, = cmake_run.mock_calls
+ (call,) = cmake_run.mock_calls
build_args, _ = call.args
if want is None:
- self.assertNotIn('-j', build_args)
+ self.assertNotIn("-j", build_args)
else:
self.assert_contains_sequence(build_args, want)
@staticmethod
- def assert_contains_sequence(sequence: Sequence[T], subsequence: Sequence[T]) -> None:
+ def assert_contains_sequence(
+ sequence: Sequence[T], subsequence: Sequence[T]
+ ) -> None:
"""Raises an assertion if the subsequence is not contained in the sequence."""
if len(subsequence) == 0:
return # all sequences contain the empty subsequence
@@ -63,7 +74,7 @@
assert len(candidate) == len(subsequence) # sanity check
if candidate == subsequence:
return # found it
- raise AssertionError(f'{subsequence} not found in {sequence}')
+ raise AssertionError(f"{subsequence} not found in {sequence}")
@contextlib.contextmanager
