File tree 2 files changed +6
-4
lines changed
2 files changed +6
-4
lines changed Original file line number Diff line number Diff line change @@ -735,10 +735,10 @@ def visit_Yield(self, node):
735
735
736
736
def visit_YieldFrom (self , node ):
737
737
with self .delimit ("(" , ")" ):
738
- self .write ("yield from" )
739
- if node .value :
740
- self . write ( " " )
741
- self .traverse (node .value )
738
+ self .write ("yield from " )
739
+ if not node .value :
740
+ raise ValueError ( "Node can't be used without a value attribute. " )
741
+ self .traverse (node .value )
742
742
743
743
def visit_Raise (self , node ):
744
744
self .fill ("raise" )
Original file line number Diff line number Diff line change @@ -278,6 +278,8 @@ def test_invalid_fstring_conversion(self):
278
278
def test_invalid_set (self ):
279
279
self .check_invalid (ast .Set (elts = []))
280
280
281
+ def test_invalid_yield_from (self ):
282
+ self .check_invalid (ast .YieldFrom (value = None ))
281
283
282
284
class DirectoryTestCase (ASTTestCase ):
283
285
"""Test roundtrip behaviour on all files in Lib and Lib/test."""
You can’t perform that action at this time.
0 commit comments