Skip to content

Commit a2d5e53

Browse files
committed
Merge pull request #6184 from dsm054/fix-test-reshape
TST: fix test_reshape.py
2 parents 7d0a9cd + ac179b0 commit a2d5e53

File tree

1 file changed

+6
-6
lines changed

1 file changed

+6
-6
lines changed

pandas/tests/test_reshape.py

+6-6
Original file line numberDiff line numberDiff line change
@@ -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

Comments
 (0)