Skip to content

Commit 69a6be4

Browse files
committed
Fix Ruff check
1 parent 83bc3db commit 69a6be4

1 file changed

Lines changed: 21 additions & 33 deletions

File tree

src/boutdata/gridue_to_bout_converter/gridue_to_bout.py

Lines changed: 21 additions & 33 deletions
Original file line numberDiff line numberDiff line change
@@ -46,7 +46,7 @@ def _importBody(gridue_settings, f):
4646
if line == "\n":
4747
try:
4848
Key = next(k)
49-
except:
49+
except StopIteration:
5050
continue
5151
else:
5252
Str[Key].append(line)
@@ -67,7 +67,7 @@ def _importBody(gridue_settings, f):
6767

6868
try:
6969
vv = next(_l)
70-
except:
70+
except StopIteration:
7171
continue
7272
gridue_settings[k] = data_
7373
return gridue_settings
@@ -114,7 +114,7 @@ def _importDN(values, f):
114114
values = [int(x) for x in next(f).split()]
115115
if len(values) != len(row):
116116
raise ValueError(
117-
"Expected row with {} integers, found {}".format(len(row), len(values))
117+
f"Expected row with {len(row)} integers, found {len(values)}"
118118
)
119119
gridue_settings.update(zip(row, values))
120120

@@ -137,7 +137,7 @@ def importGridue(fname: str = "gridue") -> dict:
137137
A dict containing header and body information from the gridue file.
138138
139139
"""
140-
f = open(fname, mode="r")
140+
f = open(fname)
141141
values = [int(x) for x in next(f).split()]
142142

143143
if len(values) == 5:
@@ -168,7 +168,7 @@ def plot(GridueParams: dict, edgecolor="black", ax: object = None, show=True):
168168
z = GridueParams["zm"]
169169
Nx = len(r)
170170
Ny = len(r[0])
171-
patches = []
171+
172172
plt.figure(figsize=(6, 10))
173173
if ax is None:
174174
ax = plt.gca()
@@ -222,7 +222,6 @@ def calcGridAngle(g: dict):
222222
Bzxy = g["bz"][:, :, 0].T
223223
Bpxy = g["bpol"][:, :, 0].T
224224

225-
psi = g["psi"]
226225
rm = g["rm"]
227226
zm = g["zm"]
228227

@@ -307,7 +306,6 @@ def calcMetric(grd: dict, bpsign, verbose=False, ignore_checks=False):
307306
"""
308307

309308
Rxy = grd["Rxy"]
310-
Zxy = grd["Zxy"]
311309
Brxy = grd["Brxy"]
312310
Bzxy = grd["Bzxy"]
313311
Btxy = grd["Btxy"]
@@ -326,34 +324,32 @@ def calcMetric(grd: dict, bpsign, verbose=False, ignore_checks=False):
326324
(tanBeta, "tan(beta)"),
327325
]:
328326
print(
329-
"{} min {}, mean {}, max {}".format(
330-
name, np.amin(var), np.mean(var), np.amax(var)
331-
)
327+
f"{name} min {np.amin(var)}, mean {np.mean(var)}, max {np.amax(var)}"
332328
)
333329

334330
dphidy = hy * Btxy / (Bpxy * Rxy)
335331

336-
I = np.zeros(Rxy.shape)
332+
sinty = np.zeros(Rxy.shape)
337333

338334
g11 = (Rxy * Bpxy) ** 2
339335
g22 = 1.0 / (hy * cosBeta) ** 2
340336
g33 = (
341337
1.0 / Rxy**2
342-
+ (Rxy * Bpxy * I) ** 2
338+
+ (Rxy * Bpxy * sinty) ** 2
343339
+ (dphidy / (hy * cosBeta)) ** 2
344-
+ 2.0 * Rxy * Bpxy * I * dphidy * tanBeta / hy
340+
+ 2.0 * Rxy * Bpxy * sinty * dphidy * tanBeta / hy
345341
)
346342
g12 = Rxy * np.abs(Bpxy) * tanBeta / hy
347-
g13 = -Rxy * Bpxy * dphidy * tanBeta / hy - I * (Rxy * Bpxy) ** 2
348-
g23 = -bpsign * dphidy / (hy * cosBeta) ** 2 - Rxy * np.abs(Bpxy) * I * tanBeta / hy
343+
g13 = -Rxy * Bpxy * dphidy * tanBeta / hy - sinty * (Rxy * Bpxy) ** 2
344+
g23 = -bpsign * dphidy / (hy * cosBeta) ** 2 - Rxy * np.abs(Bpxy) * sinty * tanBeta / hy
349345

350346
J = hy / Bpxy
351347

352-
g_11 = 1.0 / (Rxy * Bpxy * cosBeta) ** 2 + (I * Rxy) ** 2
348+
g_11 = 1.0 / (Rxy * Bpxy * cosBeta) ** 2 + (sinty * Rxy) ** 2
353349
g_22 = hy**2 + (dphidy * Rxy) ** 2
354350
g_33 = Rxy**2
355-
g_12 = bpsign * I * dphidy * Rxy**2 - hy * tanBeta / (Rxy * np.abs(Bpxy))
356-
g_13 = I * Rxy**2
351+
g_12 = bpsign * sinty * dphidy * Rxy**2 - hy * tanBeta / (Rxy * np.abs(Bpxy))
352+
g_13 = sinty * Rxy**2
357353
g_23 = bpsign * dphidy * Rxy**2
358354

359355
Jcheck = (
@@ -379,9 +375,7 @@ def calcMetric(grd: dict, bpsign, verbose=False, ignore_checks=False):
379375
(rel_error, "(J - Jcheck)/J"),
380376
]:
381377
print(
382-
"{} min {}, mean {}, max {}".format(
383-
name, np.amin(var), np.mean(var), np.amax(var)
384-
)
378+
f"{name} min {np.amin(var)}, mean {np.mean(var)}, max {np.amax(var)}"
385379
)
386380

387381
if np.max(np.abs(rel_error)) > 1e-6:
@@ -422,7 +416,7 @@ def calcMetric(grd: dict, bpsign, verbose=False, ignore_checks=False):
422416
curl_bOverB_z = (
423417
curl_bOverB_zetahat / Rxy
424418
- Btxy * hy / (Bpxy * Rxy) * curl_bOverB_y
425-
- I * curl_bOverB_x
419+
- sinty * curl_bOverB_x
426420
)
427421

428422
bxcvx = Bxy / 2.0 * curl_bOverB_x
@@ -436,9 +430,7 @@ def calcMetric(grd: dict, bpsign, verbose=False, ignore_checks=False):
436430
(bxcvz, "bxcvz"),
437431
]:
438432
print(
439-
"{} min {}, mean {}, max {}".format(
440-
name, np.amin(var), np.mean(var), np.amax(var)
441-
)
433+
f"{name} min {np.amin(var)}, mean {np.mean(var)}, max {np.amax(var)}"
442434
)
443435

444436
return {
@@ -460,7 +452,7 @@ def calcMetric(grd: dict, bpsign, verbose=False, ignore_checks=False):
460452
# Jacobian
461453
"J": J,
462454
# Integrated shear
463-
"sinty": I,
455+
"sinty": sinty,
464456
# Curvature
465457
"curl_bOverB_x": curl_bOverB_x,
466458
"curl_bOverB_y": curl_bOverB_y,
@@ -485,8 +477,8 @@ def calcRZCurvature(g: dict):
485477
Z = g["zm"]
486478

487479
dBzetadR, dBzetadZ = calcRZderivs(R, Z, Bzeta)
488-
dBRdR, dBRdZ = calcRZderivs(R, Z, BR)
489-
dBZdR, dBZdZ = calcRZderivs(R, Z, BZ)
480+
_, dBRdZ = calcRZderivs(R, Z, BR)
481+
dBZdR, _ = calcRZderivs(R, Z, BZ)
490482
dB2dR, dB2dZ = calcRZderivs(R, Z, B2)
491483

492484
# Select point at centre of cell
@@ -869,11 +861,7 @@ def Convert_grids(
869861

870862
if verbose:
871863
print(
872-
"Safety factor: min {}, mean {}, max {}".format(
873-
np.amin(ShiftAngle[:ixseps1]) / (2 * np.pi),
874-
np.mean(ShiftAngle[:ixseps1]) / (2 * np.pi),
875-
np.amax(ShiftAngle[:ixseps1]) / (2 * np.pi),
876-
)
864+
f"Safety factor: min {np.amin(ShiftAngle[:ixseps1]) / (2 * np.pi)}, mean {np.mean(ShiftAngle[:ixseps1]) / (2 * np.pi)}, max {np.amax(ShiftAngle[:ixseps1]) / (2 * np.pi)}"
877865
)
878866

879867
if plotting:

0 commit comments

Comments
 (0)