Skip to content

Commit 9488481

Browse files
Reformat test
1 parent 3d611e2 commit 9488481

File tree

1 file changed

+9
-5
lines changed

1 file changed

+9
-5
lines changed

tests/test_struct.py

Lines changed: 9 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -216,13 +216,14 @@ class MyStruct(xo.Struct):
216216

217217
def test_kernel_namings():
218218
class MyStruct(xo.Struct):
219-
n = xo.Int32
219+
n = xo.Int32
220220
var_mult_1 = xo.Float64[:]
221221
var_mult_2 = xo.Float64[:]
222222
var_mult_3 = xo.Float64[:]
223223
var_mult_4 = xo.Float64[:]
224224

225-
_extra_c_sources = [r"""
225+
_extra_c_sources = [
226+
r"""
226227
double mul(MyStruct stru) {
227228
int32_t n = MyStruct_get_n(stru);
228229
double* var_mult_1 = MyStruct_getp1_var_mult_1(stru, 0);
@@ -257,7 +258,8 @@ class MyStruct(xo.Struct):
257258
y+= var_mult_1[tid] * var_mult_2[tid] * var_mult_3[tid] * var_mult_4[tid];
258259
}
259260
return y;
260-
}"""]
261+
}"""
262+
]
261263

262264
kernel_descriptions = {
263265
"mul": xo.Kernel(
@@ -273,14 +275,16 @@ class MyStruct(xo.Struct):
273275
c_name="mult_four",
274276
args=[xo.Arg(MyStruct, name="stru")],
275277
ret=xo.Arg(xo.Float64),
276-
)
278+
),
277279
}
278280

279281
a1 = np.arange(10.0)
280282
a2 = np.arange(10.0)
281283
a3 = np.arange(10.0)
282284
a4 = np.arange(10.0)
283-
stru = MyStruct(n=10, var_mult_1=a1, var_mult_2=a2, var_mult_3=a3, var_mult_4=a4)
285+
stru = MyStruct(
286+
n=10, var_mult_1=a1, var_mult_2=a2, var_mult_3=a3, var_mult_4=a4
287+
)
284288

285289
ctx = stru._context
286290
ctx.add_kernels(kernels=kernel_descriptions)

0 commit comments

Comments
 (0)