#NEWLINE# def test_fuse_getitem( ) : #NEWLINE# #INDENT# def load( * args ) : #NEWLINE# #INDENT# pass #NEWLINE# #UNINDENT# dsk = { #NEWLINE# #INDENT# 'x' : ( load , 'store' , 'part' , [ 'a' , 'b' ] ) , #NEWLINE# 'y' : ( getitem , 'x' , 'a' ) , #NEWLINE# #UNINDENT# } #NEWLINE# dsk2 = fuse_getitem ( dsk , load , 3 ) #NEWLINE# dsk2 = cull ( dsk2 , 'y' ) #NEWLINE# assert ( dsk2 == { #NEWLINE# #INDENT# 'y' : ( load , 'store' , 'part' , 'a' ) , #NEWLINE# #UNINDENT# } )