@@ -119,22 +119,22 @@ def test_custom_var_and_value_name(self):
119119 var_name = self .var_name , value_name = self .value_name )
120120 self .assertEqual (result17 .columns .tolist (), ['id1' , 'id2' , 'var' , 'val' ])
121121
122- result18 = melt (df , id_vars = ['id1' , 'id2' ],
122+ result18 = melt (self . df , id_vars = ['id1' , 'id2' ],
123123 value_vars = 'A' , var_name = self .var_name , value_name = self .value_name )
124124 self .assertEqual (result18 .columns .tolist (), ['id1' , 'id2' , 'var' , 'val' ])
125125
126126 result19 = melt (self .df , id_vars = ['id1' , 'id2' ],
127127 value_vars = ['A' , 'B' ], var_name = self .var_name , value_name = self .value_name )
128128 expected19 = DataFrame ({'id1' : self .df ['id1' ].tolist () * 2 ,
129129 '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 ()},
132132 columns = ['id1' , 'id2' , self .var_name , self .value_name ])
133133 tm .assert_frame_equal (result19 , expected19 )
134134
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 )
138138 self .assertEqual (result20 .columns .tolist (), ['foo' , 'value' ])
139139
140140 def test_col_level (self ):
0 commit comments