make linter happy

Author: leofang

cuda_core/tests/test_launcher.py

@@ -69,39 +69,40 @@ def test_launch_invalid_values(init_cuda):
 
 # Parametrize: (python_type, cpp_type, init_value)
 PARAMS = (
-    (bool,         'bool',          True),
-    (float, 'double',   2.718),
-    (np.bool,         'bool',          True),
-    (np.int8,         'signed char',   -42),
-    (np.int16,        'signed short',  -1234),
-    (np.int32,        'signed int',  -123456),
-    (np.int64,        'signed long long',  -123456789),
-    (np.uint8,        'unsigned char',  42),
-    (np.uint16,       'unsigned short', 1234),
-    (np.uint32,       'unsigned int', 123456),
-    (np.uint64,       'unsigned long long', 123456789),
-    (np.float32,      'float',    3.14),
-    (np.float64,      'double',   2.718),
-    (ctypes.c_bool,   'bool',          True),
-    (ctypes.c_int8,   'signed char',   -42),
-    (ctypes.c_int16,  'signed short',  -1234),
-    (ctypes.c_int32,  'signed int',  -123456),
-    (ctypes.c_int64,  'signed long long',  -123456789),
-    (ctypes.c_uint8,  'unsigned char',  42),
-    (ctypes.c_uint16, 'unsigned short', 1234),
-    (ctypes.c_uint32, 'unsigned int', 123456),
-    (ctypes.c_uint64, 'unsigned long long', 123456789),
-    (ctypes.c_float,  'float',    3.14),
-    (ctypes.c_double, 'double',   2.718),
+    (bool, "bool", True),
+    (float, "double", 2.718),
+    (np.bool, "bool", True),
+    (np.int8, "signed char", -42),
+    (np.int16, "signed short", -1234),
+    (np.int32, "signed int", -123456),
+    (np.int64, "signed long long", -123456789),
+    (np.uint8, "unsigned char", 42),
+    (np.uint16, "unsigned short", 1234),
+    (np.uint32, "unsigned int", 123456),
+    (np.uint64, "unsigned long long", 123456789),
+    (np.float32, "float", 3.14),
+    (np.float64, "double", 2.718),
+    (ctypes.c_bool, "bool", True),
+    (ctypes.c_int8, "signed char", -42),
+    (ctypes.c_int16, "signed short", -1234),
+    (ctypes.c_int32, "signed int", -123456),
+    (ctypes.c_int64, "signed long long", -123456789),
+    (ctypes.c_uint8, "unsigned char", 42),
+    (ctypes.c_uint16, "unsigned short", 1234),
+    (ctypes.c_uint32, "unsigned int", 123456),
+    (ctypes.c_uint64, "unsigned long long", 123456789),
+    (ctypes.c_float, "float", 3.14),
+    (ctypes.c_double, "double", 2.718),
 )
 if os.environ.get("CUDA_PATH"):
     PARAMS += (
-        (np.float16,      'half',    0.78),
-        (np.complex64, 'cuda::std::complex<float>', 1+2j),
-        (np.complex128, 'cuda::std::complex<double>', -3-4j),
-        (complex, 'cuda::std::complex<double>',   5-7j),
+        (np.float16, "half", 0.78),
+        (np.complex64, "cuda::std::complex<float>", 1 + 2j),
+        (np.complex128, "cuda::std::complex<double>", -3 - 4j),
+        (complex, "cuda::std::complex<double>", 5 - 7j),
     )
 
+
 @pytest.mark.parametrize("python_type, cpp_type, init_value", PARAMS)
 def test_launch_scalar_argument(python_type, cpp_type, init_value):
     dev = Device()
@@ -127,13 +128,16 @@ def test_launch_scalar_argument(python_type, cpp_type, init_value):
     # Compile and force instantiation for this type
     arch = "".join(f"{i}" for i in dev.compute_capability)
     if os.environ.get("CUDA_PATH"):
-        include_path=str(pathlib.Path(os.environ["CUDA_PATH"]) / pathlib.Path("include"))
-        code = r"""
+        include_path = str(pathlib.Path(os.environ["CUDA_PATH"]) / pathlib.Path("include"))
+        code = (
+            r"""
         #include <cuda_fp16.h>
         #include <cuda/std/complex>
-        """ + code
+        """
+            + code
+        )
     else:
-        include_path=None
+        include_path = None
     pro_opts = ProgramOptions(std="c++11", arch=f"sm_{arch}", include_path=include_path)
     prog = Program(code, code_type="c++", options=pro_opts)
     ker_name = f"write_scalar<{cpp_type}>"