@@ -117,13 +117,12 @@ def test_method_calls_exist_in_driver(self, driver_dir, example_path):
117117 return # Covered by test_syntax_valid
118118 driver_imports = set ()
119119 for node in ast .walk (tree ):
120- if isinstance (node , ast .ImportFrom ):
121- if node .module and any (
122- part in driver_dir .name .replace ("-" , "_" )
123- for part in (node .module .split ("." )[0 ],)
124- ):
125- for alias in node .names :
126- driver_imports .add (alias .asname or alias .name )
120+ if isinstance (node , ast .ImportFrom ) and node .module and any (
121+ part in driver_dir .name .replace ("-" , "_" )
122+ for part in (node .module .split ("." )[0 ],)
123+ ):
124+ for alias in node .names :
125+ driver_imports .add (alias .asname or alias .name )
127126
128127 if not driver_imports :
129128 pytest .skip (f"No driver import found in { example_path .name } " )
@@ -132,18 +131,17 @@ def test_method_calls_exist_in_driver(self, driver_dir, example_path):
132131 # is assigned from a driver constructor
133132 driver_vars = set ()
134133 for node in ast .walk (tree ):
135- if isinstance (node , ast .Assign ):
136- if isinstance (node .value , ast .Call ):
137- func = node .value .func
138- func_name = None
139- if isinstance (func , ast .Name ):
140- func_name = func .id
141- elif isinstance (func , ast .Attribute ):
142- func_name = func .attr
143- if func_name in driver_imports :
144- for target in node .targets :
145- if isinstance (target , ast .Name ):
146- driver_vars .add (target .id )
134+ if isinstance (node , ast .Assign ) and isinstance (node .value , ast .Call ):
135+ func = node .value .func
136+ func_name = None
137+ if isinstance (func , ast .Name ):
138+ func_name = func .id
139+ elif isinstance (func , ast .Attribute ):
140+ func_name = func .attr
141+ if func_name in driver_imports :
142+ for target in node .targets :
143+ if isinstance (target , ast .Name ):
144+ driver_vars .add (target .id )
147145
148146 if not driver_vars :
149147 pytest .skip (
0 commit comments