@@ -119,22 +119,22 @@ def test_custom_var_and_value_name(self):
119
119
var_name = self .var_name , value_name = self .value_name )
120
120
self .assertEqual (result17 .columns .tolist (), ['id1' , 'id2' , 'var' , 'val' ])
121
121
122
- result18 = melt (df , id_vars = ['id1' , 'id2' ],
122
+ result18 = melt (self . df , id_vars = ['id1' , 'id2' ],
123
123
value_vars = 'A' , var_name = self .var_name , value_name = self .value_name )
124
124
self .assertEqual (result18 .columns .tolist (), ['id1' , 'id2' , 'var' , 'val' ])
125
125
126
126
result19 = melt (self .df , id_vars = ['id1' , 'id2' ],
127
127
value_vars = ['A' , 'B' ], var_name = self .var_name , value_name = self .value_name )
128
128
expected19 = DataFrame ({'id1' : self .df ['id1' ].tolist () * 2 ,
129
129
'id2' : self .df ['id2' ].tolist () * 2 ,
130
- var_name : ['A' ]* 10 + ['B' ]* 10 ,
131
- value_name : self .df ['A' ].tolist () + self .df ['B' ].tolist ()},
130
+ self . var_name : ['A' ]* 10 + ['B' ]* 10 ,
131
+ self . value_name : self .df ['A' ].tolist () + self .df ['B' ].tolist ()},
132
132
columns = ['id1' , 'id2' , self .var_name , self .value_name ])
133
133
tm .assert_frame_equal (result19 , expected19 )
134
134
135
- def test_custom_var_and_value_name ( self ):
136
- self . df .columns .name = 'foo'
137
- result20 = melt (self . df )
135
+ df20 = self . df . copy ()
136
+ df20 .columns .name = 'foo'
137
+ result20 = melt (df20 )
138
138
self .assertEqual (result20 .columns .tolist (), ['foo' , 'value' ])
139
139
140
140
def test_col_level (self ):
0 commit comments