Skip to content

Commit d5d26a9

Browse files
committed
style: fix lint errors
1 parent 2ccdacb commit d5d26a9

File tree

3 files changed

+24
-23
lines changed

3 files changed

+24
-23
lines changed

.flake8

+2-1
Original file line numberDiff line numberDiff line change
@@ -4,10 +4,11 @@ max-complexity = 16
44
# B = bugbear
55
# B9 = bugbear opinionated (incl line length)
66
select = C,E,F,W,B,B9
7+
# B905 `zip()` without an explicit `strict=` parameter
78
# E203: whitespace before ':' (black behaviour)
89
# E501: flake8 line length (covered by bugbear B950)
910
# W503: line break before binary operator (black behaviour)
10-
ignore = E203,E501,W503
11+
ignore = B905,E203,E501,W503
1112
classmethod-decorators =
1213
classmethod
1314
validator

pytopas/ast.py

+21-19
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,5 @@
11
"Syntax tree"
2-
# pylint: disable=too-many-lines
2+
# pylint: disable=too-many-lines,cyclic-import
33

44
from __future__ import annotations
55

@@ -744,6 +744,7 @@ def get_parser(cls):
744744

745745
@classmethod
746746
def unparse_args(cls, args: list[FormulaNode | str | None | TextNode]):
747+
"Reconstruct args"
747748
result = []
748749
for x in args:
749750
if isinstance(x, BaseNode):
@@ -763,6 +764,7 @@ def serialize(self) -> NodeSerialized:
763764

764765
@classmethod
765766
def unserialize_args(cls, data: list[Any]):
767+
"Reconstruct args"
766768
kinds = (cls.formula_cls, cls.text_cls)
767769
return [
768770
(
@@ -1209,7 +1211,7 @@ def unserialize(cls, data: list[Any]):
12091211
if typ != cls.type:
12101212
raise ReconstructException(f"assert data[0] == {cls.type}", data)
12111213
if not isinstance(value, (int, list)):
1212-
raise ReconstructException(f"assert isinstance(data[1], int)", data)
1214+
raise ReconstructException("assert isinstance(data[1], int)", data)
12131215
return cls(
12141216
value if isinstance(value, int) else ParameterNameNode.unserialize(value)
12151217
)
@@ -1253,7 +1255,7 @@ def unparse(self) -> str:
12531255
result = self.type
12541256
if self.filename:
12551257
filename_dbl_quote_esc = self.filename.replace('"', r"\"")
1256-
result += f' "{filename_dbl_quote_esc}"'
1258+
result += f" {filename_dbl_quote_esc!r}"
12571259
if self.filename is None and self.inline_data:
12581260
in_parts = (["_xy"] if self.inline_data_xy else []) + list(
12591261
map(str, self.inline_data)
@@ -1304,10 +1306,10 @@ def unserialize(cls, data: list[Any]):
13041306
raise ReconstructException(f"assert data[0] == {cls.type}", data)
13051307
opts = data[1]
13061308
if not isinstance(opts, dict):
1307-
raise ReconstructException(f"assert isinstance(data[1], dict)", data)
1309+
raise ReconstructException("assert isinstance(data[1], dict)", data)
13081310
flags = data[2] if len(data) > 2 else []
13091311
if not isinstance(flags, list):
1310-
raise ReconstructException(f"assert isinstance(data[2], list)", data)
1312+
raise ReconstructException("assert isinstance(data[2], list)", data)
13111313

13121314
return cls(
13131315
filename=opts.get("filename"),
@@ -1349,9 +1351,9 @@ def parse_action(cls, toks: pp.ParseResults):
13491351
filament_length=toks.filament_length, # type: ignore[assigment]
13501352
sample_length=toks.sample_length, # type: ignore[assigment]
13511353
receiving_slit_length=toks.receiving_slit_length, # type: ignore[assigment]
1352-
primary_soller_angle=data.get("primary_soller_angle"), # type: ignore[assigment]
1353-
secondary_soller_angle=data.get("secondary_soller_angle"), # type: ignore[assigment]
1354-
axial_n_beta=data.get("axial_n_beta"), # type: ignore[assigment]
1354+
primary_soller_angle=data.get("primary_soller_angle"),
1355+
secondary_soller_angle=data.get("secondary_soller_angle"),
1356+
axial_n_beta=data.get("axial_n_beta"),
13551357
)
13561358

13571359
@classmethod
@@ -1396,12 +1398,12 @@ def unserialize(cls, data: list[Any]):
13961398
raise ReconstructException(f"assert data[0] == {cls.type}", data)
13971399
args = data[1]
13981400
if not isinstance(args, list):
1399-
raise ReconstructException(f"assert isinstance(data[1], list)", data)
1401+
raise ReconstructException("assert isinstance(data[1], list)", data)
14001402
if len(args) != 3:
1401-
raise ReconstructException(f"assert len(data[1]) == 3", data)
1403+
raise ReconstructException("assert len(data[1]) == 3", data)
14021404
opts = data[2] if len(data) > 2 else {}
14031405
if not isinstance(opts, dict):
1404-
raise ReconstructException(f"assert isinstance(data[2], dict)", data)
1406+
raise ReconstructException("assert isinstance(data[2], dict)", data)
14051407

14061408
return cls(
14071409
filament_length=cls.parameter_cls.unserialize(args[0]),
@@ -1428,7 +1430,7 @@ class BkgNode(BaseNode):
14281430
@classmethod
14291431
def parse_action(cls, toks: pp.ParseResults):
14301432
"Parse action for the bkg node"
1431-
return cls(params=(toks.as_list()))
1433+
return cls(params=toks.as_list())
14321434

14331435
@classmethod
14341436
def get_parser(cls):
@@ -1447,8 +1449,8 @@ def unserialize(cls, data: list[Any]):
14471449
raise ReconstructException("assert len >= 2", data)
14481450
if data[0] != cls.type:
14491451
raise ReconstructException(f"assert data[0] == {cls.type}", data)
1450-
if not all([isinstance(x, list) for x in data[1:]]):
1451-
raise ReconstructException(f"assert all of data[1:] of list type", data)
1452+
if not all(isinstance(x, list) for x in data[1:]):
1453+
raise ReconstructException("assert all of data[1:] of list type", data)
14521454
return cls(params=[cls.parameter_cls.unserialize(x) for x in data[1:]])
14531455

14541456

@@ -1461,7 +1463,7 @@ class ScaleNode(BaseNode):
14611463
@classmethod
14621464
def parse_action(cls, toks: pp.ParseResults):
14631465
"Parse action for the bkg node"
1464-
return cls(param=(toks.as_list()[0]))
1466+
return cls(param=toks.as_list()[0])
14651467

14661468
@classmethod
14671469
def get_parser(cls):
@@ -1480,7 +1482,7 @@ def unserialize(cls, data: list[Any]):
14801482
if data[0] != cls.type:
14811483
raise ReconstructException(f"assert data[0] == {cls.type}", data)
14821484
if not isinstance(data[1], list):
1483-
raise ReconstructException(f"assert isinstance(data[1], list)", data)
1485+
raise ReconstructException("assert isinstance(data[1], list)", data)
14841486
return cls(param=cls.parameter_cls.unserialize(data[1]))
14851487

14861488

@@ -1574,11 +1576,11 @@ def unserialize(cls, data: list[Any]):
15741576
if data[0] != cls.type:
15751577
raise ReconstructException(f"assert data[0] == {cls.type}", data)
15761578
if not isinstance(data[1], str):
1577-
raise ReconstructException(f"assert type of data[1] == str", data)
1579+
raise ReconstructException("assert type of data[1] == str", data)
15781580
if not isinstance(data[2], list):
1579-
raise ReconstructException(f"assert type of data[2] == list", data)
1581+
raise ReconstructException("assert type of data[2] == list", data)
15801582
if not isinstance(data[3], list):
1581-
raise ReconstructException(f"assert type of data[3] == list", data)
1583+
raise ReconstructException("assert type of data[3] == list", data)
15821584

15831585
stmts = [
15841586
(

tests/grammar/test_grammar_root.py

+1-3
Original file line numberDiff line numberDiff line change
@@ -124,9 +124,7 @@
124124
ast.RootNode(
125125
statements=[
126126
ast.ScaleNode(
127-
ast.ParameterNode(
128-
prm_value=ast.ParameterValueNode(Decimal(1))
129-
)
127+
ast.ParameterNode(prm_value=ast.ParameterValueNode(Decimal(1)))
130128
)
131129
]
132130
)

0 commit comments

Comments
 (0)