#NEWLINE# @ pytest . mark . parametrize ( 'data,expr,ret_type,exp_type' , [ ( 1 , symbol ( 'x' , 'int' ) , 'native' , int ) , ( 1 , symbol ( 'x' , 'int' ) , 'core' , int ) , ( into ( da . core . Array , [ 1 , 2 ] , chunks = ( 10 , ) ) , symbol ( 'x' , '2 * int' ) , 'core' , pd . Series ) , ( into ( da . core . Array , [ { #NEWLINE# #INDENT# 'a' : 1 , #NEWLINE# 'b' : 2 , #NEWLINE# #UNINDENT# } , { #NEWLINE# #INDENT# 'a' : 3 , #NEWLINE# 'b' : 4 , #NEWLINE# #UNINDENT# } ] , chunks = ( 10 , 10 ) ) , symbol ( 'x' , '2 * {a: int, b: int}' ) , 'core' , pd . DataFrame ) , ( into ( da . core . Array , [ [ 1 , 2 ] , [ 3 , 4 ] ] , chunks = ( 10 , 10 ) ) , symbol ( 'x' , '2 *  2 * int' ) , 'core' , np . ndarray ) , ( [ 1 , 2 ] , symbol ( 'x' , '2 * int' ) , tuple , tuple ) ] ) #NEWLINE# def test_compute_return_type( data , expr , ret_type , exp_type ) : #NEWLINE# #INDENT# assert isinstance ( compute ( expr , data , return_type = ret_type ) , exp_type )