/usr/share/pyshared/pandas/tests/test_reshape.py is in python-pandas 0.7.0-1.
This file is owned by root:root, with mode 0o644.
The actual contents of the file can be viewed below.
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 | from pandas import DataFrame
import numpy as np
from pandas.core.reshape import melt, convert_dummies
import pandas.util.testing as tm
def test_melt():
df = tm.makeTimeDataFrame()[:10]
df['id1'] = (df['A'] > 0).astype(int)
df['id2'] = (df['B'] > 0).astype(int)
molten1 = melt(df)
molten2 = melt(df, id_vars=['id1'])
molten3 = melt(df, id_vars=['id1', 'id2'])
def test_convert_dummies():
df = DataFrame({'A' : ['foo', 'bar', 'foo', 'bar',
'foo', 'bar', 'foo', 'foo'],
'B' : ['one', 'one', 'two', 'three',
'two', 'two', 'one', 'three'],
'C' : np.random.randn(8),
'D' : np.random.randn(8)})
result = convert_dummies(df, ['A', 'B'])
result2 = convert_dummies(df, ['A', 'B'], prefix_sep='.')
expected = DataFrame({'A_foo' : [1, 0, 1, 0, 1, 0, 1, 1],
'A_bar' : [0, 1, 0, 1, 0, 1, 0, 0],
'B_one' : [1, 1, 0, 0, 0, 0, 1, 0],
'B_two' : [0, 0, 1, 0, 1, 1, 0, 0],
'B_three' : [0, 0, 0, 1, 0, 0, 0, 1],
'C' : df['C'].values,
'D' : df['D'].values},
columns=result.columns, dtype=float)
expected2 = expected.rename(columns=lambda x: x.replace('_', '.'))
tm.assert_frame_equal(result, expected)
tm.assert_frame_equal(result2, expected2)
if __name__ == '__main__':
import nose
nose.runmodule(argv=[__file__,'-vvs','-x','--pdb', '--pdb-failure'],
exit=False)
|