@@ -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