#NEWLINE# def test_pandas_dataframe_loading( self ) : #NEWLINE# #INDENT# df = pd . DataFrame ( { #NEWLINE# #INDENT# 'one' : [ 1 , 2 , 3 ] , #NEWLINE# 'two' : [ 6 , 7 , 8 ] , #NEWLINE# 'three' : [ 11 , 12 , 13 ] , #NEWLINE# 'four' : [ 17 , 18 , 19 ] , #NEWLINE# #UNINDENT# } ) #NEWLINE# get_all = [ { #NEWLINE# #INDENT# 'col' : 'four' , #NEWLINE# 'idx' : 0 , #NEWLINE# 'val' : 17 , #NEWLINE# #UNINDENT# } , { #NEWLINE# #INDENT# 'col' : 'one' , #NEWLINE# 'idx' : 0 , #NEWLINE# 'val' : 1 , #NEWLINE# #UNINDENT# } , { #NEWLINE# #INDENT# 'col' : 'three' , #NEWLINE# 'idx' : 0 , #NEWLINE# 'val' : 11 , #NEWLINE# #UNINDENT# } , { #NEWLINE# #INDENT# 'col' : 'two' , #NEWLINE# 'idx' : 0 , #NEWLINE# 'val' : 6 , #NEWLINE# #UNINDENT# } , { #NEWLINE# #INDENT# 'col' : 'four' , #NEWLINE# 'idx' : 1 , #NEWLINE# 'val' : 18 , #NEWLINE# #UNINDENT# } , { #NEWLINE# #INDENT# 'col' : 'one' , #NEWLINE# 'idx' : 1 , #NEWLINE# 'val' : 2 , #NEWLINE# #UNINDENT# } , { #NEWLINE# #INDENT# 'col' : 'three' , #NEWLINE# 'idx' : 1 , #NEWLINE# 'val' : 12 , #NEWLINE# #UNINDENT# } , { #NEWLINE# #INDENT# 'col' : 'two' , #NEWLINE# 'idx' : 1 , #NEWLINE# 'val' : 7 , #NEWLINE# #UNINDENT# } , { #NEWLINE# #INDENT# 'col' : 'four' , #NEWLINE# 'idx' : 2 , #NEWLINE# 'val' : 19 , #NEWLINE# #UNINDENT# } , { #NEWLINE# #INDENT# 'col' : 'one' , #NEWLINE# 'idx' : 2 , #NEWLINE# 'val' : 3 , #NEWLINE# #UNINDENT# } , { #NEWLINE# #INDENT# 'col' : 'three' , #NEWLINE# 'idx' : 2 , #NEWLINE# 'val' : 13 , #NEWLINE# #UNINDENT# } , { #NEWLINE# #INDENT# 'col' : 'two' , #NEWLINE# 'idx' : 2 , #NEWLINE# 'val' : 8 , #NEWLINE# #UNINDENT# } ] #NEWLINE# get1 = [ { #NEWLINE# #INDENT# 'col' : 'one' , #NEWLINE# 'idx' : 0 , #NEWLINE# 'val' : 1 , #NEWLINE# #UNINDENT# } , { #NEWLINE# #INDENT# 'col' : 'one' , #NEWLINE# 'idx' : 1 , #NEWLINE# 'val' : 2 , #NEWLINE# #UNINDENT# } , { #NEWLINE# #INDENT# 'col' : 'one' , #NEWLINE# 'idx' : 2 , #NEWLINE# 'val' : 3 , #NEWLINE# #UNINDENT# } ] #NEWLINE# get2 = [ { #NEWLINE# #INDENT# 'col' : 'one' , #NEWLINE# 'idx' : 0 , #NEWLINE# 'val' : 1 , #NEWLINE# #UNINDENT# } , { #NEWLINE# #INDENT# 'col' : 'two' , #NEWLINE# 'idx' : 0 , #NEWLINE# 'val' : 6 , #NEWLINE# #UNINDENT# } , { #NEWLINE# #INDENT# 'col' : 'one' , #NEWLINE# 'idx' : 1 , #NEWLINE# 'val' : 2 , #NEWLINE# #UNINDENT# } , { #NEWLINE# #INDENT# 'col' : 'two' , #NEWLINE# 'idx' : 1 , #NEWLINE# 'val' : 7 , #NEWLINE# #UNINDENT# } , { #NEWLINE# #INDENT# 'col' : 'one' , #NEWLINE# 'idx' : 2 , #NEWLINE# 'val' : 3 , #NEWLINE# #UNINDENT# } , { #NEWLINE# #INDENT# 'col' : 'two' , #NEWLINE# 'idx' : 2 , #NEWLINE# 'val' : 8 , #NEWLINE# #UNINDENT# } ] #NEWLINE# getkey2 = [ { #NEWLINE# #INDENT# 'col' : 'one' , #NEWLINE# 'idx' : 6 , #NEWLINE# 'val' : 1 , #NEWLINE# #UNINDENT# } , { #NEWLINE# #INDENT# 'col' : 'one' , #NEWLINE# 'idx' : 7 , #NEWLINE# 'val' : 2 , #NEWLINE# #UNINDENT# } , { #NEWLINE# #INDENT# 'col' : 'one' , #NEWLINE# 'idx' : 8 , #NEWLINE# 'val' : 3 , #NEWLINE# #UNINDENT# } ] #NEWLINE# getkey3 = [ { #NEWLINE# #INDENT# 'col' : 'one' , #NEWLINE# 'idx' : 11 , #NEWLINE# 'val' : 1 , #NEWLINE# #UNINDENT# } , { #NEWLINE# #INDENT# 'col' : 'two' , #NEWLINE# 'idx' : 11 , #NEWLINE# 'val' : 6 , #NEWLINE# #UNINDENT# } , { #NEWLINE# #INDENT# 'col' : 'one' , #NEWLINE# 'idx' : 12 , #NEWLINE# 'val' : 2 , #NEWLINE# #UNINDENT# } , { #NEWLINE# #INDENT# 'col' : 'two' , #NEWLINE# 'idx' : 12 , #NEWLINE# 'val' : 7 , #NEWLINE# #UNINDENT# } , { #NEWLINE# #INDENT# 'col' : 'one' , #NEWLINE# 'idx' : 13 , #NEWLINE# 'val' : 3 , #NEWLINE# #UNINDENT# } , { #NEWLINE# #INDENT# 'col' : 'two' , #NEWLINE# 'idx' : 13 , #NEWLINE# 'val' : 8 , #NEWLINE# #UNINDENT# } ] #NEWLINE# val_all = Data . from_pandas ( df ) #NEWLINE# val1 = Data . from_pandas ( df , columns = [ 'one' ] ) #NEWLINE# val2 = Data . from_pandas ( df , columns = [ 'one' , 'two' ] ) #NEWLINE# key2 = Data . from_pandas ( df , columns = [ 'one' ] , key_on = 'two' ) #NEWLINE# key3 = Data . from_pandas ( df , columns = [ 'one' , 'two' ] , key_on = 'three' ) #NEWLINE# nt . assert_list_equal ( val_all . values , get_all ) #NEWLINE# nt . assert_list_equal ( val1 . values , get1 ) #NEWLINE# nt . assert_list_equal ( val2 . values , get2 ) #NEWLINE# nt . assert_list_equal ( key3 . values , getkey2 ) #NEWLINE# nt . assert_list_equal ( key3 . values , getkey3 ) #NEWLINE# dataframe = pd . DataFrame ( np . random . randn ( 10 , 3 ) ) #NEWLINE# data = Data . from_pandas ( dataframe ) #NEWLINE# nt . assert_equal ( data . name , 'table' ) #NEWLINE# nt . assert_raises ( ValueError , Data . from_pandas , { #NEWLINE# #INDENT# #NEWLINE# #UNINDENT# } )