#NEWLINE# def test_get_merged_dataset_data_returns_specified_columns( self ) : #NEWLINE# #INDENT# data = MergedDataset ( [ ( 'NSE/OIL' , { #NEWLINE# #INDENT# 'column_index' : [ 1 , 2 ] , #NEWLINE# #UNINDENT# } ) , ( 'SINGLE/COLUMN' , { #NEWLINE# #INDENT# 'column_index' : [ 1 ] , #NEWLINE# #UNINDENT# } ) , 'GOOG/NASDAQ_MSFT' ] ) . data ( ) #NEWLINE# actual = data . to_pandas ( ) . columns . tolist ( ) #NEWLINE# expected = [ six . u ( 'NSE/OIL - column.1' ) , six . u ( 'NSE/OIL - column.2' ) , six . u ( 'SINGLE/COLUMN - column.1' ) , six . u ( 'GOOG/NASDAQ_MSFT - column.1' ) , six . u ( 'GOOG/NASDAQ_MSFT - column.2' ) , six . u ( 'GOOG/NASDAQ_MSFT - column.3' ) ] #NEWLINE# self . assertItemsEqual ( actual , expected )