Make black happy

Author: freddieknets

tests/test_buffer.py

@@ -95,10 +95,12 @@ def test_free_simple(test_context):
         ch.free_string(offset)
         ch.check()
 
+
 @for_all_test_contexts
 def test_free(test_context):
     class CheckFree(xo.Struct):
         a = xo.Float64
+
     ch = CheckFree(a=5, _context=test_context)
     assert ch._buffer.capacity == 8
     assert ch._buffer.chunks == []
@@ -107,17 +109,17 @@ class CheckFree(xo.Struct):
     with pytest.raises(ValueError, match="Cannot free outside of buffer"):
         ch._buffer.free(0, 10)
     with pytest.raises(ValueError, match="Cannot free outside of buffer"):
-        ch._buffer.free(7,2)
-    ch._buffer.free(0,4)
+        ch._buffer.free(7, 2)
+    ch._buffer.free(0, 4)
     assert len(ch._buffer.chunks) == 1
     assert ch._buffer.chunks[0].start == 0
     assert ch._buffer.chunks[0].end == 4
-    ch._buffer.free(0,4)  # Does nothing
-    ch._buffer.free(2,4)  # Increases free chunk
+    ch._buffer.free(0, 4)  # Does nothing
+    ch._buffer.free(2, 4)  # Increases free chunk
     assert len(ch._buffer.chunks) == 1
     assert ch._buffer.chunks[0].start == 0
     assert ch._buffer.chunks[0].end == 6
-    ch._buffer.free(7,1)
+    ch._buffer.free(7, 1)
     assert len(ch._buffer.chunks) == 2
     assert ch._buffer.chunks[0].start == 0
     assert ch._buffer.chunks[0].end == 6

tests/test_hybrid_class.py

@@ -338,16 +338,16 @@ class TD(xo.HybridClass):
             "_a": xo.Float64[3],
             "_b": xo.Int64,
         }
-        _skip_in_to_dict = ['_a', '_b']
-        _store_in_to_dict = ['a', 'b']
-        _extra_fields_in_to_dict = ['c']
+        _skip_in_to_dict = ["_a", "_b"]
+        _store_in_to_dict = ["a", "b"]
+        _extra_fields_in_to_dict = ["c"]
 
         def __init__(self, **kwargs):
-            if '_xobject' in kwargs and kwargs['_xobject'] is not None:
+            if "_xobject" in kwargs and kwargs["_xobject"] is not None:
                 self._initialize(**kwargs)
                 return
-            kwargs['_a'] = kwargs.pop('a', [1., 2., 3.])
-            kwargs['_b'] = kwargs.pop('b', 0)
+            kwargs["_a"] = kwargs.pop("a", [1.0, 2.0, 3.0])
+            kwargs["_b"] = kwargs.pop("b", 0)
             self._initialize(**kwargs)
 
         def _initialize(self, **kwargs):
@@ -380,7 +380,7 @@ def c(self):
 
     # Verify that to_dict has all fields, including python-only ones,
     # for custom initialisation
-    td2 = TD(a=[8,9,10], b=40, c=20)
+    td2 = TD(a=[8, 9, 10], b=40, c=20)
     td2_dict = td2.to_dict()
     assert td2_dict.pop("__class__") == "TD"
     assert all(td2_dict.pop("a") == [8, 9, 10])
@@ -402,7 +402,7 @@ def c(self):
     assert td4.c == td2.c
 
     # Verify that move works correctly
-    td3.move(_context=xo.ContextCpu(omp_num_threads='auto'))
+    td3.move(_context=xo.ContextCpu(omp_num_threads="auto"))
     assert all(td3.a == td2.a)
     assert td3.b == td2.b
     assert td3.c == td2.c

xobjects/context_cupy.py

@@ -457,7 +457,11 @@ def build_kernels(
 
         extra_include_paths = self.get_installed_c_source_paths()
         include_flags = [f"-I{path}" for path in extra_include_paths]
-        extra_compile_args = (*extra_compile_args, *include_flags, "-DXO_CONTEXT_CUDA")
+        extra_compile_args = (
+            *extra_compile_args,
+            *include_flags,
+            "-DXO_CONTEXT_CUDA"
+        )
 
         module = cupy.RawModule(
             code=specialized_source, options=extra_compile_args