<s> import nose <EOL> from pandas import DataFrame , Series <EOL> from pandas . core . sparse import SparseDataFrame <EOL> import pandas as pd <EOL> from numpy import nan <EOL> import numpy as np <EOL> from pandas . util . testing import assert_frame_equal <EOL> from pandas . core . reshape import ( melt , lreshape , get_dummies , wide_to_long ) <EOL> import pandas . util . testing as tm <EOL> from pandas . compat import range , u <EOL> _multiprocess_can_split_ = True <EOL> class TestMelt ( tm . TestCase ) : <EOL> def setUp ( self ) : <EOL> self . df = tm . makeTimeDataFrame ( ) [ : <NUM_LIT:10> ] <EOL> self . df [ '<STR_LIT>' ] = ( self . df [ '<STR_LIT:A>' ] > <NUM_LIT:0> ) . astype ( np . int64 ) <EOL> self . df [ '<STR_LIT>' ] = ( self . df [ '<STR_LIT:B>' ] > <NUM_LIT:0> ) . astype ( np . int64 ) <EOL> self . var_name = '<STR_LIT>' <EOL> self . value_name = '<STR_LIT>' <EOL> self . df1 = pd . DataFrame ( [ [ <NUM_LIT> , - <NUM_LIT> , <NUM_LIT> <EOL> ] , [ - <NUM_LIT> , <NUM_LIT> , - <NUM_LIT> ] , <EOL> [ - <NUM_LIT> , <NUM_LIT> , - <NUM_LIT> ] ] ) <EOL> self . df1 . columns = [ list ( '<STR_LIT>' ) , list ( '<STR_LIT:abc>' ) ] <EOL> self . df1 . columns . names = [ '<STR_LIT>' , '<STR_LIT>' ] <EOL> def test_default_col_names ( self ) : <EOL> result = melt ( self . df ) <EOL> self . assertEqual ( result . columns . tolist ( ) , [ '<STR_LIT>' , '<STR_LIT:value>' ] ) <EOL> result1 = melt ( self . df , id_vars = [ '<STR_LIT>' ] ) <EOL> self . assertEqual ( result1 . columns . tolist ( ) , [ '<STR_LIT>' , '<STR_LIT>' , '<STR_LIT:value>' <EOL> ] ) <EOL> result2 = melt ( self . df , id_vars = [ '<STR_LIT>' , '<STR_LIT>' ] ) <EOL> self . assertEqual ( result2 . columns . tolist ( ) , [ '<STR_LIT>' , '<STR_LIT>' , '<STR_LIT>' , <EOL> '<STR_LIT:value>' ] ) <EOL> def test_value_vars ( self ) : <EOL> result3 = melt ( self . df , id_vars = [ '<STR_LIT>' , '<STR_LIT>' ] , value_vars = '<STR_LIT:A>' ) <EOL> self . assertEqual ( len ( result3 ) , <NUM_LIT:10> ) <EOL> result4 = melt ( self . df , id_vars = [ '<STR_LIT>' , '<STR_LIT>' ] , value_vars = [ '<STR_LIT:A>' , '<STR_LIT:B>' ] ) <EOL> expected4 = DataFrame ( { '<STR_LIT>' : self . df [ '<STR_LIT>' ] . tolist ( ) * <NUM_LIT:2> , <EOL> '<STR_LIT>' : self . df [ '<STR_LIT>' ] . tolist ( ) * <NUM_LIT:2> , <EOL> '<STR_LIT>' : [ '<STR_LIT:A>' ] * <NUM_LIT:10> + [ '<STR_LIT:B>' ] * <NUM_LIT:10> , <EOL> '<STR_LIT:value>' : ( self . df [ '<STR_LIT:A>' ] . tolist ( ) + <EOL> self . df [ '<STR_LIT:B>' ] . tolist ( ) ) } , <EOL> columns = [ '<STR_LIT>' , '<STR_LIT>' , '<STR_LIT>' , '<STR_LIT:value>' ] ) <EOL> tm . assert_frame_equal ( result4 , expected4 ) <EOL> def test_custom_var_name ( self ) : <EOL> result5 = melt ( self . df , var_name = self . var_name ) <EOL> self . assertEqual ( result5 . columns . tolist ( ) , [ '<STR_LIT>' , '<STR_LIT:value>' ] ) <EOL> result6 = melt ( self . df , id_vars = [ '<STR_LIT>' ] , var_name = self . var_name ) <EOL> self . assertEqual ( result6 . columns . tolist ( ) , [ '<STR_LIT>' , '<STR_LIT>' , '<STR_LIT:value>' ] ) <EOL> result7 = melt ( self . df , id_vars = [ '<STR_LIT>' , '<STR_LIT>' ] , var_name = self . var_name ) <EOL> self . assertEqual ( result7 . columns . tolist ( ) , [ '<STR_LIT>' , '<STR_LIT>' , '<STR_LIT>' , <EOL> '<STR_LIT:value>' ] ) <EOL> result8 = melt ( self . df , id_vars = [ '<STR_LIT>' , '<STR_LIT>' ] , value_vars = '<STR_LIT:A>' , <EOL> var_name = self . var_name ) <EOL> self . assertEqual ( result8 . columns . tolist ( ) , [ '<STR_LIT>' , '<STR_LIT>' , '<STR_LIT>' , <EOL> '<STR_LIT:value>' ] ) <EOL> result9 = melt ( self . df , id_vars = [ '<STR_LIT>' , '<STR_LIT>' ] , value_vars = [ '<STR_LIT:A>' , '<STR_LIT:B>' ] , <EOL> var_name = self . var_name ) <EOL> expected9 = DataFrame ( { '<STR_LIT>' : self . df [ '<STR_LIT>' ] . tolist ( ) * <NUM_LIT:2> , <EOL> '<STR_LIT>' : self . df [ '<STR_LIT>' ] . tolist ( ) * <NUM_LIT:2> , <EOL> self . var_name : [ '<STR_LIT:A>' ] * <NUM_LIT:10> + [ '<STR_LIT:B>' ] * <NUM_LIT:10> , <EOL> '<STR_LIT:value>' : ( self . df [ '<STR_LIT:A>' ] . tolist ( ) + <EOL> self . df [ '<STR_LIT:B>' ] . tolist ( ) ) } , <EOL> columns = [ '<STR_LIT>' , '<STR_LIT>' , self . var_name , '<STR_LIT:value>' ] ) <EOL> tm . assert_frame_equal ( result9 , expected9 ) <EOL> def test_custom_value_name ( self ) : <EOL> result10 = melt ( self . df , value_name = self . value_name ) <EOL> self . assertEqual ( result10 . columns . tolist ( ) , [ '<STR_LIT>' , '<STR_LIT>' ] ) <EOL> result11 = melt ( self . df , id_vars = [ '<STR_LIT>' ] , value_name = self . value_name ) <EOL> self . assertEqual ( result11 . columns . tolist ( ) , [ '<STR_LIT>' , '<STR_LIT>' , '<STR_LIT>' ] ) <EOL> result12 = melt ( self . df , id_vars = [ '<STR_LIT>' , '<STR_LIT>' ] , <EOL> value_name = self . value_name ) <EOL> self . assertEqual ( result12 . columns . tolist ( ) , [ '<STR_LIT>' , '<STR_LIT>' , '<STR_LIT>' , <EOL> '<STR_LIT>' ] ) <EOL> result13 = melt ( self . df , id_vars = [ '<STR_LIT>' , '<STR_LIT>' ] , value_vars = '<STR_LIT:A>' , <EOL> value_name = self . value_name ) <EOL> self . assertEqual ( result13 . columns . tolist ( ) , [ '<STR_LIT>' , '<STR_LIT>' , '<STR_LIT>' , <EOL> '<STR_LIT>' ] ) <EOL> result14 = melt ( self . df , id_vars = [ '<STR_LIT>' , '<STR_LIT>' ] , value_vars = [ '<STR_LIT:A>' , '<STR_LIT:B>' ] , <EOL> value_name = self . value_name ) <EOL> expected14 = DataFrame ( { '<STR_LIT>' : self . df [ '<STR_LIT>' ] . tolist ( ) * <NUM_LIT:2> , <EOL> '<STR_LIT>' : self . df [ '<STR_LIT>' ] . tolist ( ) * <NUM_LIT:2> , <EOL> '<STR_LIT>' : [ '<STR_LIT:A>' ] * <NUM_LIT:10> + [ '<STR_LIT:B>' ] * <NUM_LIT:10> , <EOL> self . value_name : ( self . df [ '<STR_LIT:A>' ] . tolist ( ) + <EOL> self . df [ '<STR_LIT:B>' ] . tolist ( ) ) } , <EOL> columns = [ '<STR_LIT>' , '<STR_LIT>' , '<STR_LIT>' , <EOL> self . value_name ] ) <EOL> tm . assert_frame_equal ( result14 , expected14 ) <EOL> def test_custom_var_and_value_name ( self ) : <EOL> result15 = melt ( self . df , var_name = self . var_name , <EOL> value_name = self . value_name ) <EOL> self . assertEqual ( result15 . columns . tolist ( ) , [ '<STR_LIT>' , '<STR_LIT>' ] ) <EOL> result16 = melt ( self . df , id_vars = [ '<STR_LIT>' ] , var_name = self . var_name , <EOL> value_name = self . value_name ) <EOL> self . assertEqual ( result16 . columns . tolist ( ) , [ '<STR_LIT>' , '<STR_LIT>' , '<STR_LIT>' ] ) <EOL> result17 = melt ( self . df , id_vars = [ '<STR_LIT>' , '<STR_LIT>' ] , <EOL> var_name = self . var_name , value_name = self . value_name ) <EOL> self . assertEqual ( result17 . columns . tolist ( ) , [ '<STR_LIT>' , '<STR_LIT>' , '<STR_LIT>' , '<STR_LIT>' <EOL> ] ) <EOL> result18 = melt ( self . df , id_vars = [ '<STR_LIT>' , '<STR_LIT>' ] , value_vars = '<STR_LIT:A>' , <EOL> var_name = self . var_name , value_name = self . value_name ) <EOL> self . assertEqual ( result18 . columns . tolist ( ) , [ '<STR_LIT>' , '<STR_LIT>' , '<STR_LIT>' , '<STR_LIT>' <EOL> ] ) <EOL> result19 = melt ( self . df , id_vars = [ '<STR_LIT>' , '<STR_LIT>' ] , value_vars = [ '<STR_LIT:A>' , '<STR_LIT:B>' ] , <EOL> var_name = self . var_name , value_name = self . value_name ) <EOL> expected19 = DataFrame ( { '<STR_LIT>' : self . df [ '<STR_LIT>' ] . tolist ( ) * <NUM_LIT:2> , <EOL> '<STR_LIT>' : self . df [ '<STR_LIT>' ] . tolist ( ) * <NUM_LIT:2> , <EOL> self . var_name : [ '<STR_LIT:A>' ] * <NUM_LIT:10> + [ '<STR_LIT:B>' ] * <NUM_LIT:10> , <EOL> self . value_name : ( self . df [ '<STR_LIT:A>' ] . tolist ( ) + <EOL> self . df [ '<STR_LIT:B>' ] . tolist ( ) ) } , <EOL> columns = [ '<STR_LIT>' , '<STR_LIT>' , self . var_name , <EOL> self . value_name ] ) <EOL> tm . assert_frame_equal ( result19 , expected19 ) <EOL> df20 = self . df . copy ( ) <EOL> df20 . columns . name = '<STR_LIT:foo>' <EOL> result20 = melt ( df20 ) <EOL> self . assertEqual ( result20 . columns . tolist ( ) , [ '<STR_LIT:foo>' , '<STR_LIT:value>' ] ) <EOL> def test_col_level ( self ) : <EOL> res1 = melt ( self . df1 , col_level = <NUM_LIT:0> ) <EOL> res2 = melt ( self . df1 , col_level = '<STR_LIT>' ) <EOL> self . assertEqual ( res1 . columns . tolist ( ) , [ '<STR_LIT>' , '<STR_LIT:value>' ] ) <EOL> self . assertEqual ( res2 . columns . tolist ( ) , [ '<STR_LIT>' , '<STR_LIT:value>' ] ) <EOL> def test_multiindex ( self ) : <EOL> res = pd . melt ( self . df1 ) <EOL> self . assertEqual ( res . columns . tolist ( ) , [ '<STR_LIT>' , '<STR_LIT>' , '<STR_LIT:value>' ] ) <EOL> class TestGetDummies ( tm . TestCase ) : <EOL> sparse = False <EOL> def setUp ( self ) : <EOL> self . df = DataFrame ( { '<STR_LIT:A>' : [ '<STR_LIT:a>' , '<STR_LIT:b>' , '<STR_LIT:a>' ] , <EOL> '<STR_LIT:B>' : [ '<STR_LIT:b>' , '<STR_LIT:b>' , '<STR_LIT:c>' ] , <EOL> '<STR_LIT:C>' : [ <NUM_LIT:1> , <NUM_LIT:2> , <NUM_LIT:3> ] } ) <EOL> def test_basic ( self ) : <EOL> s_list = list ( '<STR_LIT:abc>' ) <EOL> s_series = Series ( s_list ) <EOL> s_series_index = Series ( s_list , list ( '<STR_LIT>' ) ) <EOL> expected = DataFrame ( { '<STR_LIT:a>' : { <NUM_LIT:0> : <NUM_LIT:1.0> , <EOL> <NUM_LIT:1> : <NUM_LIT:0.0> , <EOL> <NUM_LIT:2> : <NUM_LIT:0.0> } , <EOL> '<STR_LIT:b>' : { <NUM_LIT:0> : <NUM_LIT:0.0> , <EOL> <NUM_LIT:1> : <NUM_LIT:1.0> , <EOL> <NUM_LIT:2> : <NUM_LIT:0.0> } , <EOL> '<STR_LIT:c>' : { <NUM_LIT:0> : <NUM_LIT:0.0> , <EOL> <NUM_LIT:1> : <NUM_LIT:0.0> , <EOL> <NUM_LIT:2> : <NUM_LIT:1.0> } } ) <EOL> assert_frame_equal ( get_dummies ( s_list , sparse = self . sparse ) , expected ) <EOL> assert_frame_equal ( get_dummies ( s_series , sparse = self . sparse ) , expected ) <EOL> expected . index = list ( '<STR_LIT>' ) <EOL> assert_frame_equal ( <EOL> get_dummies ( s_series_index , sparse = self . sparse ) , expected ) <EOL> def test_basic_types ( self ) : <EOL> s_list = list ( '<STR_LIT:abc>' ) <EOL> s_series = Series ( s_list ) <EOL> s_df = DataFrame ( { '<STR_LIT:a>' : [ <NUM_LIT:0> , <NUM_LIT:1> , <NUM_LIT:0> , <NUM_LIT:1> , <NUM_LIT:2> ] , <EOL> '<STR_LIT:b>' : [ '<STR_LIT:A>' , '<STR_LIT:A>' , '<STR_LIT:B>' , '<STR_LIT:C>' , '<STR_LIT:C>' ] , <EOL> '<STR_LIT:c>' : [ <NUM_LIT:2> , <NUM_LIT:3> , <NUM_LIT:3> , <NUM_LIT:3> , <NUM_LIT:2> ] } ) <EOL> if not self . sparse : <EOL> exp_df_type = DataFrame <EOL> exp_blk_type = pd . core . internals . FloatBlock <EOL> else : <EOL> exp_df_type = SparseDataFrame <EOL> exp_blk_type = pd . core . internals . SparseBlock <EOL> self . assertEqual ( <EOL> type ( get_dummies ( s_list , sparse = self . sparse ) ) , exp_df_type ) <EOL> self . assertEqual ( <EOL> type ( get_dummies ( s_series , sparse = self . sparse ) ) , exp_df_type ) <EOL> r = get_dummies ( s_df , sparse = self . sparse , columns = s_df . columns ) <EOL> self . assertEqual ( type ( r ) , exp_df_type ) <EOL> r = get_dummies ( s_df , sparse = self . sparse , columns = [ '<STR_LIT:a>' ] ) <EOL> self . assertEqual ( type ( r [ [ '<STR_LIT>' ] ] . _data . blocks [ <NUM_LIT:0> ] ) , exp_blk_type ) <EOL> self . assertEqual ( type ( r [ [ '<STR_LIT>' ] ] . _data . blocks [ <NUM_LIT:0> ] ) , exp_blk_type ) <EOL> self . assertEqual ( type ( r [ [ '<STR_LIT>' ] ] . _data . blocks [ <NUM_LIT:0> ] ) , exp_blk_type ) <EOL> def test_just_na ( self ) : <EOL> just_na_list = [ np . nan ] <EOL> just_na_series = Series ( just_na_list ) <EOL> just_na_series_index = Series ( just_na_list , index = [ '<STR_LIT:A>' ] ) <EOL> res_list = get_dummies ( just_na_list , sparse = self . sparse ) <EOL> res_series = get_dummies ( just_na_series , sparse = self . sparse ) <EOL> res_series_index = get_dummies ( just_na_series_index , <EOL> sparse = self . sparse ) <EOL> self . assertEqual ( res_list . empty , True ) <EOL> self . assertEqual ( res_series . empty , True ) <EOL> self . assertEqual ( res_series_index . empty , True ) <EOL> self . assertEqual ( res_list . index . tolist ( ) , [ <NUM_LIT:0> ] ) <EOL> self . assertEqual ( res_series . index . tolist ( ) , [ <NUM_LIT:0> ] ) <EOL> self . assertEqual ( res_series_index . index . tolist ( ) , [ '<STR_LIT:A>' ] ) <EOL> def test_include_na ( self ) : <EOL> s = [ '<STR_LIT:a>' , '<STR_LIT:b>' , np . nan ] <EOL> res = get_dummies ( s , sparse = self . sparse ) <EOL> exp = DataFrame ( { '<STR_LIT:a>' : { <NUM_LIT:0> : <NUM_LIT:1.0> , <EOL> <NUM_LIT:1> : <NUM_LIT:0.0> , <EOL> <NUM_LIT:2> : <NUM_LIT:0.0> } , <EOL> '<STR_LIT:b>' : { <NUM_LIT:0> : <NUM_LIT:0.0> , <EOL> <NUM_LIT:1> : <NUM_LIT:1.0> , <EOL> <NUM_LIT:2> : <NUM_LIT:0.0> } } ) <EOL> assert_frame_equal ( res , exp ) <EOL> res_na = get_dummies ( s , dummy_na = True , sparse = self . sparse ) <EOL> exp_na = DataFrame ( { nan : { <NUM_LIT:0> : <NUM_LIT:0.0> , <EOL> <NUM_LIT:1> : <NUM_LIT:0.0> , <EOL> <NUM_LIT:2> : <NUM_LIT:1.0> } , <EOL> '<STR_LIT:a>' : { <NUM_LIT:0> : <NUM_LIT:1.0> , <EOL> <NUM_LIT:1> : <NUM_LIT:0.0> , <EOL> <NUM_LIT:2> : <NUM_LIT:0.0> } , <EOL> '<STR_LIT:b>' : { <NUM_LIT:0> : <NUM_LIT:0.0> , <EOL> <NUM_LIT:1> : <NUM_LIT:1.0> , <EOL> <NUM_LIT:2> : <NUM_LIT:0.0> } } ) . reindex_axis ( <EOL> [ '<STR_LIT:a>' , '<STR_LIT:b>' , nan ] , <NUM_LIT:1> ) <EOL> exp_na . columns = res_na . columns <EOL> assert_frame_equal ( res_na , exp_na ) <EOL> res_just_na = get_dummies ( [ nan ] , dummy_na = True , sparse = self . sparse ) <EOL> exp_just_na = DataFrame ( Series ( <NUM_LIT:1.0> , index = [ <NUM_LIT:0> ] ) , columns = [ nan ] ) <EOL> tm . assert_numpy_array_equal ( res_just_na . values , exp_just_na . values ) <EOL> def test_unicode ( self <EOL> ) : <EOL> import unicodedata <EOL> e = '<STR_LIT:e>' <EOL> eacute = unicodedata . lookup ( '<STR_LIT>' ) <EOL> s = [ e , eacute , eacute ] <EOL> res = get_dummies ( s , prefix = '<STR_LIT>' , sparse = self . sparse ) <EOL> exp = DataFrame ( { '<STR_LIT>' : { <NUM_LIT:0> : <NUM_LIT:1.0> , <EOL> <NUM_LIT:1> : <NUM_LIT:0.0> , <EOL> <NUM_LIT:2> : <NUM_LIT:0.0> } , <EOL> u ( '<STR_LIT>' ) % eacute : { <NUM_LIT:0> : <NUM_LIT:0.0> , <EOL> <NUM_LIT:1> : <NUM_LIT:1.0> , <EOL> <NUM_LIT:2> : <NUM_LIT:1.0> } } ) <EOL> assert_frame_equal ( res , exp ) <EOL> def test_dataframe_dummies_all_obj ( self ) : <EOL> df = self . df [ [ '<STR_LIT:A>' , '<STR_LIT:B>' ] ] <EOL> result = get_dummies ( df , sparse = self . sparse ) <EOL> expected = DataFrame ( { '<STR_LIT>' : [ <NUM_LIT:1.> , <NUM_LIT:0> , <NUM_LIT:1> ] , <EOL> '<STR_LIT>' : [ <NUM_LIT:0.> , <NUM_LIT:1> , <NUM_LIT:0> ] , <EOL> '<STR_LIT>' : [ <NUM_LIT:1.> , <NUM_LIT:1> , <NUM_LIT:0> ] , <EOL> '<STR_LIT>' : [ <NUM_LIT:0.> , <NUM_LIT:0> , <NUM_LIT:1> ] } ) <EOL> assert_frame_equal ( result , expected ) <EOL> def test_dataframe_dummies_mix_default ( self ) : <EOL> df = self . df <EOL> result = get_dummies ( df , sparse = self . sparse ) <EOL> expected = DataFrame ( { '<STR_LIT:C>' : [ <NUM_LIT:1> , <NUM_LIT:2> , <NUM_LIT:3> ] , <EOL> '<STR_LIT>' : [ <NUM_LIT:1.> , <NUM_LIT:0> , <NUM_LIT:1> ] , <EOL> '<STR_LIT>' : [ <NUM_LIT:0.> , <NUM_LIT:1> , <NUM_LIT:0> ] , <EOL> '<STR_LIT>' : [ <NUM_LIT:1.> , <NUM_LIT:1> , <NUM_LIT:0> ] , <EOL> '<STR_LIT>' : [ <NUM_LIT:0.> , <NUM_LIT:0> , <NUM_LIT:1> ] } ) <EOL> expected = expected [ [ '<STR_LIT:C>' , '<STR_LIT>' , '<STR_LIT>' , '<STR_LIT>' , '<STR_LIT>' ] ] <EOL> assert_frame_equal ( result , expected ) <EOL> def test_dataframe_dummies_prefix_list ( self ) : <EOL> prefixes = [ '<STR_LIT>' , '<STR_LIT>' ] <EOL> df = DataFrame ( { '<STR_LIT:A>' : [ '<STR_LIT:a>' , '<STR_LIT:b>' , '<STR_LIT:a>' ] , <EOL> '<STR_LIT:B>' : [ '<STR_LIT:b>' , '<STR_LIT:b>' , '<STR_LIT:c>' ] , <EOL> '<STR_LIT:C>' : [ <NUM_LIT:1> , <NUM_LIT:2> , <NUM_LIT:3> ] } ) <EOL> result = get_dummies ( df , prefix = prefixes , sparse = self . sparse ) <EOL> expected = DataFrame ( { '<STR_LIT:C>' : [ <NUM_LIT:1> , <NUM_LIT:2> , <NUM_LIT:3> ] , <EOL> '<STR_LIT>' : [ <NUM_LIT:1.> , <NUM_LIT:0> , <NUM_LIT:1> ] , <EOL> '<STR_LIT>' : [ <NUM_LIT:0.> , <NUM_LIT:1> , <NUM_LIT:0> ] , <EOL> '<STR_LIT>' : [ <NUM_LIT:1.> , <NUM_LIT:1> , <NUM_LIT:0> ] , <EOL> '<STR_LIT>' : [ <NUM_LIT:0.> , <NUM_LIT:0> , <NUM_LIT:1> ] } ) <EOL> expected = expected [ [ '<STR_LIT:C>' , '<STR_LIT>' , '<STR_LIT>' , '<STR_LIT>' , <EOL> '<STR_LIT>' ] ] <EOL> assert_frame_equal ( result , expected ) <EOL> def test_dataframe_dummies_prefix_str ( self ) : <EOL> df = self . df <EOL> result = get_dummies ( df , prefix = '<STR_LIT>' , sparse = self . sparse ) <EOL> expected = DataFrame ( [ [ <NUM_LIT:1> , <NUM_LIT:1.> , <NUM_LIT:0.> , <NUM_LIT:1.> , <NUM_LIT:0.> ] , <EOL> [ <NUM_LIT:2> , <NUM_LIT:0.> , <NUM_LIT:1.> , <NUM_LIT:1.> , <NUM_LIT:0.> ] , <EOL> [ <NUM_LIT:3> , <NUM_LIT:1.> , <NUM_LIT:0.> , <NUM_LIT:0.> , <NUM_LIT:1.> ] ] , <EOL> columns = [ '<STR_LIT:C>' , '<STR_LIT>' , '<STR_LIT>' , '<STR_LIT>' , '<STR_LIT>' ] ) <EOL> assert_frame_equal ( result , expected ) <EOL> def test_dataframe_dummies_subset ( self ) : <EOL> df = self . df <EOL> result = get_dummies ( df , prefix = [ '<STR_LIT>' ] , columns = [ '<STR_LIT:A>' ] , <EOL> sparse = self . sparse ) <EOL> expected = DataFrame ( { '<STR_LIT>' : [ <NUM_LIT:1.> , <NUM_LIT:0> , <NUM_LIT:1> ] , <EOL> '<STR_LIT>' : [ <NUM_LIT:0.> , <NUM_LIT:1> , <NUM_LIT:0> ] , <EOL> '<STR_LIT:B>' : [ '<STR_LIT:b>' , '<STR_LIT:b>' , '<STR_LIT:c>' ] , <EOL> '<STR_LIT:C>' : [ <NUM_LIT:1> , <NUM_LIT:2> , <NUM_LIT:3> ] } ) <EOL> assert_frame_equal ( result , expected ) <EOL> def test_dataframe_dummies_prefix_sep ( self ) : <EOL> df = self . df <EOL> result = get_dummies ( df , prefix_sep = '<STR_LIT:..>' , sparse = self . sparse ) <EOL> expected = DataFrame ( { '<STR_LIT:C>' : [ <NUM_LIT:1> , <NUM_LIT:2> , <NUM_LIT:3> ] , <EOL> '<STR_LIT>' : [ <NUM_LIT:1.> , <NUM_LIT:0> , <NUM_LIT:1> ] , <EOL> '<STR_LIT>' : [ <NUM_LIT:0.> , <NUM_LIT:1> , <NUM_LIT:0> ] , <EOL> '<STR_LIT>' : [ <NUM_LIT:1.> , <NUM_LIT:1> , <NUM_LIT:0> ] , <EOL> '<STR_LIT>' : [ <NUM_LIT:0.> , <NUM_LIT:0> , <NUM_LIT:1> ] } ) <EOL> expected = expected [ [ '<STR_LIT:C>' , '<STR_LIT>' , '<STR_LIT>' , '<STR_LIT>' , '<STR_LIT>' ] ] <EOL> assert_frame_equal ( result , expected ) <EOL> result = get_dummies ( df , prefix_sep = [ '<STR_LIT:..>' , '<STR_LIT>' ] , sparse = self . sparse ) <EOL> expected = expected . rename ( columns = { '<STR_LIT>' : '<STR_LIT>' , '<STR_LIT>' : '<STR_LIT>' } ) <EOL> assert_frame_equal ( result , expected ) <EOL> result = get_dummies ( df , prefix_sep = { '<STR_LIT:A>' : '<STR_LIT:..>' , <EOL> '<STR_LIT:B>' : '<STR_LIT>' } , sparse = self . sparse ) <EOL> assert_frame_equal ( result , expected ) <EOL> def test_dataframe_dummies_prefix_bad_length ( self ) : <EOL> with tm . assertRaises ( ValueError ) : <EOL> get_dummies ( self . df , prefix = [ '<STR_LIT>' ] , sparse = self . sparse ) <EOL> def test_dataframe_dummies_prefix_sep_bad_length ( self ) : <EOL> with tm . assertRaises ( ValueError ) : <EOL> get_dummies ( self . df , prefix_sep = [ '<STR_LIT>' ] , sparse = self . sparse ) <EOL> def test_dataframe_dummies_prefix_dict ( self ) : <EOL> prefixes = { '<STR_LIT:A>' : '<STR_LIT>' , '<STR_LIT:B>' : '<STR_LIT>' } <EOL> df = DataFrame ( { '<STR_LIT:A>' : [ '<STR_LIT:a>' , '<STR_LIT:b>' , '<STR_LIT:a>' ] , <EOL> '<STR_LIT:B>' : [ '<STR_LIT:b>' , '<STR_LIT:b>' , '<STR_LIT:c>' ] , <EOL> '<STR_LIT:C>' : [ <NUM_LIT:1> , <NUM_LIT:2> , <NUM_LIT:3> ] } ) <EOL> result = get_dummies ( df , prefix = prefixes , sparse = self . sparse ) <EOL> expected = DataFrame ( { '<STR_LIT>' : [ <NUM_LIT:1.> , <NUM_LIT:0> , <NUM_LIT:1> ] , <EOL> '<STR_LIT>' : [ <NUM_LIT:0.> , <NUM_LIT:1> , <NUM_LIT:0> ] , <EOL> '<STR_LIT>' : [ <NUM_LIT:1.> , <NUM_LIT:1> , <NUM_LIT:0> ] , <EOL> '<STR_LIT>' : [ <NUM_LIT:0.> , <NUM_LIT:0> , <NUM_LIT:1> ] , <EOL> '<STR_LIT:C>' : [ <NUM_LIT:1> , <NUM_LIT:2> , <NUM_LIT:3> ] } ) <EOL> assert_frame_equal ( result , expected ) <EOL> def test_dataframe_dummies_with_na ( self ) : <EOL> df = self . df <EOL> df . loc [ <NUM_LIT:3> , : ] = [ np . nan , np . nan , np . nan ] <EOL> result = get_dummies ( df , dummy_na = True , sparse = self . sparse ) <EOL> expected = DataFrame ( { '<STR_LIT:C>' : [ <NUM_LIT:1> , <NUM_LIT:2> , <NUM_LIT:3> , np . nan ] , <EOL> '<STR_LIT>' : [ <NUM_LIT:1.> , <NUM_LIT:0> , <NUM_LIT:1> , <NUM_LIT:0> ] , <EOL> '<STR_LIT>' : [ <NUM_LIT:0.> , <NUM_LIT:1> , <NUM_LIT:0> , <NUM_LIT:0> ] , <EOL> '<STR_LIT>' : [ <NUM_LIT:0.> , <NUM_LIT:0> , <NUM_LIT:0> , <NUM_LIT:1> ] , <EOL> '<STR_LIT>' : [ <NUM_LIT:1.> , <NUM_LIT:1> , <NUM_LIT:0> , <NUM_LIT:0> ] , <EOL> '<STR_LIT>' : [ <NUM_LIT:0.> , <NUM_LIT:0> , <NUM_LIT:1> , <NUM_LIT:0> ] , <EOL> '<STR_LIT>' : [ <NUM_LIT:0.> , <NUM_LIT:0> , <NUM_LIT:0> , <NUM_LIT:1> ] } ) <EOL> expected = expected [ [ '<STR_LIT:C>' , '<STR_LIT>' , '<STR_LIT>' , '<STR_LIT>' , <EOL> '<STR_LIT>' , '<STR_LIT>' , '<STR_LIT>' ] ] <EOL> assert_frame_equal ( result , expected ) <EOL> result = get_dummies ( df , dummy_na = False , sparse = self . sparse ) <EOL> expected = expected [ [ '<STR_LIT:C>' , '<STR_LIT>' , '<STR_LIT>' , '<STR_LIT>' , '<STR_LIT>' ] ] <EOL> assert_frame_equal ( result , expected ) <EOL> def test_dataframe_dummies_with_categorical ( self ) : <EOL> df = self . df <EOL> df [ '<STR_LIT>' ] = pd . Categorical ( [ '<STR_LIT:x>' , '<STR_LIT:y>' , '<STR_LIT:y>' ] ) <EOL> result = get_dummies ( df , sparse = self . sparse ) <EOL> expected = DataFrame ( { '<STR_LIT:C>' : [ <NUM_LIT:1> , <NUM_LIT:2> , <NUM_LIT:3> ] , <EOL> '<STR_LIT>' : [ <NUM_LIT:1.> , <NUM_LIT:0> , <NUM_LIT:1> ] , <EOL> '<STR_LIT>' : [ <NUM_LIT:0.> , <NUM_LIT:1> , <NUM_LIT:0> ] , <EOL> '<STR_LIT>' : [ <NUM_LIT:1.> , <NUM_LIT:1> , <NUM_LIT:0> ] , <EOL> '<STR_LIT>' : [ <NUM_LIT:0.> , <NUM_LIT:0> , <NUM_LIT:1> ] , <EOL> '<STR_LIT>' : [ <NUM_LIT:1.> , <NUM_LIT:0> , <NUM_LIT:0> ] , <EOL> '<STR_LIT>' : [ <NUM_LIT:0.> , <NUM_LIT:1> , <NUM_LIT:1> ] } ) <EOL> expected = expected [ [ '<STR_LIT:C>' , '<STR_LIT>' , '<STR_LIT>' , '<STR_LIT>' , '<STR_LIT>' , <EOL> '<STR_LIT>' , '<STR_LIT>' ] ] <EOL> assert_frame_equal ( result , expected ) <EOL> def test_basic_drop_first ( self ) : <EOL> s_list = list ( '<STR_LIT:abc>' ) <EOL> s_series = Series ( s_list ) <EOL> s_series_index = Series ( s_list , list ( '<STR_LIT>' ) ) <EOL> expected = DataFrame ( { '<STR_LIT:b>' : { <NUM_LIT:0> : <NUM_LIT:0.0> , <EOL> <NUM_LIT:1> : <NUM_LIT:1.0> , <EOL> <NUM_LIT:2> : <NUM_LIT:0.0> } , <EOL> '<STR_LIT:c>' : { <NUM_LIT:0> : <NUM_LIT:0.0> , <EOL> <NUM_LIT:1> : <NUM_LIT:0.0> , <EOL> <NUM_LIT:2> : <NUM_LIT:1.0> } } ) <EOL> result = get_dummies ( s_list , sparse = self . sparse , drop_first = True ) <EOL> assert_frame_equal ( result , expected ) <EOL> result = get_dummies ( s_series , sparse = self . sparse , drop_first = True ) <EOL> assert_frame_equal ( result , expected ) <EOL> expected . index = list ( '<STR_LIT>' ) <EOL> result = get_dummies ( s_series_index , sparse = self . sparse , <EOL> drop_first = True ) <EOL> assert_frame_equal ( result , expected ) <EOL> def test_basic_drop_first_one_level ( self ) : <EOL> s_list = list ( '<STR_LIT>' ) <EOL> s_series = Series ( s_list ) <EOL> s_series_index = Series ( s_list , list ( '<STR_LIT>' ) ) <EOL> expected = DataFrame ( index = np . arange ( <NUM_LIT:3> ) ) <EOL> result = get_dummies ( s_list , sparse = self . sparse , drop_first = True ) <EOL> assert_frame_equal ( result , expected ) <EOL> result = get_dummies ( s_series , sparse = self . sparse , drop_first = True ) <EOL> assert_frame_equal ( result , expected ) <EOL> expected = DataFrame ( index = list ( '<STR_LIT>' ) ) <EOL> result = get_dummies ( s_series_index , sparse = self . sparse , <EOL> drop_first = True ) <EOL> assert_frame_equal ( result , expected ) <EOL> def test_basic_drop_first_NA ( self ) : <EOL> s_NA = [ '<STR_LIT:a>' , '<STR_LIT:b>' , np . nan ] <EOL> res = get_dummies ( s_NA , sparse = self . sparse , drop_first = True ) <EOL> exp = DataFrame ( { '<STR_LIT:b>' : { <NUM_LIT:0> : <NUM_LIT:0.0> , <EOL> <NUM_LIT:1> : <NUM_LIT:1.0> , <EOL> <NUM_LIT:2> : <NUM_LIT:0.0> } } ) <EOL> assert_frame_equal ( res , exp ) <EOL> res_na = get_dummies ( s_NA , dummy_na = True , sparse = self . sparse , <EOL> drop_first = True ) <EOL> exp_na = DataFrame ( { '<STR_LIT:b>' : { <NUM_LIT:0> : <NUM_LIT:0.0> , <EOL> <NUM_LIT:1> : <NUM_LIT:1.0> , <EOL> <NUM_LIT:2> : <NUM_LIT:0.0> } , <EOL> nan : { <NUM_LIT:0> : <NUM_LIT:0.0> , <EOL> <NUM_LIT:1> : <NUM_LIT:0.0> , <EOL> <NUM_LIT:2> : <NUM_LIT:1.0> } } ) . reindex_axis ( <EOL> [ '<STR_LIT:b>' , nan ] , <NUM_LIT:1> ) <EOL> assert_frame_equal ( res_na , exp_na ) <EOL> res_just_na = get_dummies ( [ nan ] , dummy_na = True , sparse = self . sparse , <EOL> drop_first = True ) <EOL> exp_just_na = DataFrame ( index = np . arange ( <NUM_LIT:1> ) ) <EOL> assert_frame_equal ( res_just_na , exp_just_na ) <EOL> def test_dataframe_dummies_drop_first ( self ) : <EOL> df = self . df [ [ '<STR_LIT:A>' , '<STR_LIT:B>' ] ] <EOL> result = get_dummies ( df , sparse = self . sparse , drop_first = True ) <EOL> expected = DataFrame ( { '<STR_LIT>' : [ <NUM_LIT:0.> , <NUM_LIT:1> , <NUM_LIT:0> ] , <EOL> '<STR_LIT>' : [ <NUM_LIT:0.> , <NUM_LIT:0> , <NUM_LIT:1> ] } ) <EOL> assert_frame_equal ( result , expected ) <EOL> def test_dataframe_dummies_drop_first_with_categorical ( self ) : <EOL> df = self . df <EOL> df [ '<STR_LIT>' ] = pd . Categorical ( [ '<STR_LIT:x>' , '<STR_LIT:y>' , '<STR_LIT:y>' ] ) <EOL> result = get_dummies ( df , sparse = self . sparse , drop_first = True ) <EOL> expected = DataFrame ( { '<STR_LIT:C>' : [ <NUM_LIT:1> , <NUM_LIT:2> , <NUM_LIT:3> ] , <EOL> '<STR_LIT>' : [ <NUM_LIT:0.> , <NUM_LIT:1> , <NUM_LIT:0> ] , <EOL> '<STR_LIT>' : [ <NUM_LIT:0.> , <NUM_LIT:0> , <NUM_LIT:1> ] , <EOL> '<STR_LIT>' : [ <NUM_LIT:0.> , <NUM_LIT:1> , <NUM_LIT:1> ] } ) <EOL> expected = expected [ [ '<STR_LIT:C>' , '<STR_LIT>' , '<STR_LIT>' , '<STR_LIT>' ] ] <EOL> assert_frame_equal ( result , expected ) <EOL> def test_dataframe_dummies_drop_first_with_na ( self ) : <EOL> df = self . df <EOL> df . loc [ <NUM_LIT:3> , : ] = [ np . nan , np . nan , np . nan ] <EOL> result = get_dummies ( df , dummy_na = True , sparse = self . sparse , <EOL> drop_first = True ) <EOL> expected = DataFrame ( { '<STR_LIT:C>' : [ <NUM_LIT:1> , <NUM_LIT:2> , <NUM_LIT:3> , np . nan ] , <EOL> '<STR_LIT>' : [ <NUM_LIT:0.> , <NUM_LIT:1> , <NUM_LIT:0> , <NUM_LIT:0> ] , <EOL> '<STR_LIT>' : [ <NUM_LIT:0.> , <NUM_LIT:0> , <NUM_LIT:0> , <NUM_LIT:1> ] , <EOL> '<STR_LIT>' : [ <NUM_LIT:0.> , <NUM_LIT:0> , <NUM_LIT:1> , <NUM_LIT:0> ] , <EOL> '<STR_LIT>' : [ <NUM_LIT:0.> , <NUM_LIT:0> , <NUM_LIT:0> , <NUM_LIT:1> ] } ) <EOL> expected = expected [ [ '<STR_LIT:C>' , '<STR_LIT>' , '<STR_LIT>' , '<STR_LIT>' , '<STR_LIT>' ] ] <EOL> assert_frame_equal ( result , expected ) <EOL> result = get_dummies ( df , dummy_na = False , sparse = self . sparse , <EOL> drop_first = True ) <EOL> expected = expected [ [ '<STR_LIT:C>' , '<STR_LIT>' , '<STR_LIT>' ] ] <EOL> assert_frame_equal ( result , expected ) <EOL> class TestGetDummiesSparse ( TestGetDummies ) : <EOL> sparse = True <EOL> class TestLreshape ( tm . TestCase ) : <EOL> def test_pairs ( self ) : <EOL> data = { '<STR_LIT>' : [ '<STR_LIT>' , '<STR_LIT>' , '<STR_LIT>' , '<STR_LIT>' , <EOL> '<STR_LIT>' ] , <EOL> '<STR_LIT>' : [ <NUM_LIT> , <NUM_LIT> , <NUM_LIT> , <NUM_LIT> , <NUM_LIT> ] , <EOL> '<STR_LIT:id>' : [ <NUM_LIT> , <NUM_LIT> , <NUM_LIT> , <NUM_LIT> , <NUM_LIT> ] , <EOL> '<STR_LIT>' : [ '<STR_LIT>' , '<STR_LIT>' , '<STR_LIT>' , '<STR_LIT>' , '<STR_LIT>' ] , <EOL> '<STR_LIT>' : [ '<STR_LIT>' , '<STR_LIT>' , '<STR_LIT>' , <EOL> '<STR_LIT>' , '<STR_LIT>' ] , <EOL> '<STR_LIT>' : <EOL> [ '<STR_LIT>' , nan , '<STR_LIT>' , '<STR_LIT>' , '<STR_LIT>' ] , <EOL> '<STR_LIT>' : [ '<STR_LIT>' , nan , nan , '<STR_LIT>' , '<STR_LIT>' ] , <EOL> '<STR_LIT>' : [ <NUM_LIT> , <NUM_LIT> , <NUM_LIT> , <NUM_LIT> , <NUM_LIT> ] , <EOL> '<STR_LIT>' : [ <NUM_LIT> , nan , <NUM_LIT> , <NUM_LIT> , <NUM_LIT> ] , <EOL> '<STR_LIT>' : [ <NUM_LIT> , nan , nan , <NUM_LIT> , <NUM_LIT> ] } <EOL> df = DataFrame ( data ) <EOL> spec = { '<STR_LIT>' : [ '<STR_LIT>' % i for i in range ( <NUM_LIT:1> , <NUM_LIT:4> ) ] , <EOL> '<STR_LIT>' : [ '<STR_LIT>' % i for i in range ( <NUM_LIT:1> , <NUM_LIT:4> ) ] } <EOL> result = lreshape ( df , spec ) <EOL> exp_data = { '<STR_LIT>' : <EOL> [ '<STR_LIT>' , '<STR_LIT>' , '<STR_LIT>' , '<STR_LIT>' , <EOL> '<STR_LIT>' , '<STR_LIT>' , '<STR_LIT>' , '<STR_LIT>' , <EOL> '<STR_LIT>' , '<STR_LIT>' , '<STR_LIT>' , '<STR_LIT>' ] , <EOL> '<STR_LIT>' : [ <NUM_LIT> , <NUM_LIT> , <NUM_LIT> , <NUM_LIT> , <NUM_LIT> , <NUM_LIT> , <NUM_LIT> , <NUM_LIT> , <EOL> <NUM_LIT> , <NUM_LIT> , <NUM_LIT> , <NUM_LIT> ] , <EOL> '<STR_LIT:id>' : [ <NUM_LIT> , <NUM_LIT> , <NUM_LIT> , <NUM_LIT> , <NUM_LIT> , <NUM_LIT> , <NUM_LIT> , <NUM_LIT> , <NUM_LIT> , <NUM_LIT> , <EOL> <NUM_LIT> , <NUM_LIT> ] , <EOL> '<STR_LIT>' : [ '<STR_LIT>' , '<STR_LIT>' , '<STR_LIT>' , '<STR_LIT>' , '<STR_LIT>' , <EOL> '<STR_LIT>' , '<STR_LIT>' , '<STR_LIT>' , '<STR_LIT>' , '<STR_LIT>' , <EOL> '<STR_LIT>' , '<STR_LIT>' ] , <EOL> '<STR_LIT>' : [ '<STR_LIT>' , '<STR_LIT>' , '<STR_LIT>' , <EOL> '<STR_LIT>' , '<STR_LIT>' , '<STR_LIT>' , <EOL> '<STR_LIT>' , '<STR_LIT>' , '<STR_LIT>' , <EOL> '<STR_LIT>' , '<STR_LIT>' , '<STR_LIT>' ] , <EOL> '<STR_LIT>' : [ <NUM_LIT> , <NUM_LIT> , <NUM_LIT> , <NUM_LIT> , <NUM_LIT> , <NUM_LIT> , <EOL> <NUM_LIT> , <NUM_LIT> , <NUM_LIT> , <NUM_LIT> , <NUM_LIT> , <NUM_LIT> ] } <EOL> exp = DataFrame ( exp_data , columns = result . columns ) <EOL> tm . assert_frame_equal ( result , exp ) <EOL> result = lreshape ( df , spec , dropna = False ) <EOL> exp_data = { '<STR_LIT>' : <EOL> [ '<STR_LIT>' , '<STR_LIT>' , '<STR_LIT>' , '<STR_LIT>' , <EOL> '<STR_LIT>' , '<STR_LIT>' , '<STR_LIT>' , '<STR_LIT>' , <EOL> '<STR_LIT>' , '<STR_LIT>' , '<STR_LIT>' , '<STR_LIT>' , <EOL> '<STR_LIT>' , '<STR_LIT>' , '<STR_LIT>' ] , <EOL> '<STR_LIT>' : [ <NUM_LIT> , <NUM_LIT> , <NUM_LIT> , <NUM_LIT> , <NUM_LIT> , <NUM_LIT> , <NUM_LIT> , <NUM_LIT> , <EOL> <NUM_LIT> , <NUM_LIT> , <NUM_LIT> , <NUM_LIT> , <NUM_LIT> , <NUM_LIT> , <NUM_LIT> ] , <EOL> '<STR_LIT:id>' : [ <NUM_LIT> , <NUM_LIT> , <NUM_LIT> , <NUM_LIT> , <NUM_LIT> , <NUM_LIT> , <NUM_LIT> , <NUM_LIT> , <NUM_LIT> , <NUM_LIT> , <EOL> <NUM_LIT> , <NUM_LIT> , <NUM_LIT> , <NUM_LIT> , <NUM_LIT> ] , <EOL> '<STR_LIT>' : [ '<STR_LIT>' , '<STR_LIT>' , '<STR_LIT>' , '<STR_LIT>' , '<STR_LIT>' , <EOL> '<STR_LIT>' , '<STR_LIT>' , '<STR_LIT>' , '<STR_LIT>' , '<STR_LIT>' , <EOL> '<STR_LIT>' , '<STR_LIT>' , '<STR_LIT>' , '<STR_LIT>' , '<STR_LIT>' ] , <EOL> '<STR_LIT>' : [ '<STR_LIT>' , '<STR_LIT>' , '<STR_LIT>' , <EOL> '<STR_LIT>' , '<STR_LIT>' , '<STR_LIT>' , nan , <EOL> '<STR_LIT>' , '<STR_LIT>' , '<STR_LIT>' , <EOL> '<STR_LIT>' , nan , nan , '<STR_LIT>' , <EOL> '<STR_LIT>' ] , <EOL> '<STR_LIT>' : [ <NUM_LIT> , <NUM_LIT> , <NUM_LIT> , <NUM_LIT> , <NUM_LIT> , <NUM_LIT> , nan , <EOL> <NUM_LIT> , <NUM_LIT> , <NUM_LIT> , <NUM_LIT> , nan , nan , <NUM_LIT> , <EOL> <NUM_LIT> ] } <EOL> exp = DataFrame ( exp_data , columns = result . columns ) <EOL> tm . assert_frame_equal ( result , exp ) <EOL> spec = { '<STR_LIT>' : [ '<STR_LIT>' % i for i in range ( <NUM_LIT:1> , <NUM_LIT:3> ) ] , <EOL> '<STR_LIT>' : [ '<STR_LIT>' % i for i in range ( <NUM_LIT:1> , <NUM_LIT:4> ) ] } <EOL> self . assertRaises ( ValueError , lreshape , df , spec ) <EOL> class TestWideToLong ( tm . TestCase ) : <EOL> def test_simple ( self ) : <EOL> np . random . seed ( <NUM_LIT> ) <EOL> x = np . random . randn ( <NUM_LIT:3> ) <EOL> df = pd . DataFrame ( { "<STR_LIT>" : { <NUM_LIT:0> : "<STR_LIT:a>" , <EOL> <NUM_LIT:1> : "<STR_LIT:b>" , <EOL> <NUM_LIT:2> : "<STR_LIT:c>" } , <EOL> "<STR_LIT>" : { <NUM_LIT:0> : "<STR_LIT:d>" , <EOL> <NUM_LIT:1> : "<STR_LIT:e>" , <EOL> <NUM_LIT:2> : "<STR_LIT:f>" } , <EOL> "<STR_LIT>" : { <NUM_LIT:0> : <NUM_LIT> , <EOL> <NUM_LIT:1> : <NUM_LIT> , <EOL> <NUM_LIT:2> : <NUM_LIT> } , <EOL> "<STR_LIT>" : { <NUM_LIT:0> : <NUM_LIT> , <EOL> <NUM_LIT:1> : <NUM_LIT> , <EOL> <NUM_LIT:2> : <NUM_LIT> } , <EOL> "<STR_LIT:X>" : dict ( zip ( <EOL> range ( <NUM_LIT:3> ) , x ) ) } ) <EOL> df [ "<STR_LIT:id>" ] = df . index <EOL> exp_data = { "<STR_LIT:X>" : x . tolist ( ) + x . tolist ( ) , <EOL> "<STR_LIT:A>" : [ '<STR_LIT:a>' , '<STR_LIT:b>' , '<STR_LIT:c>' , '<STR_LIT:d>' , '<STR_LIT:e>' , '<STR_LIT:f>' ] , <EOL> "<STR_LIT:B>" : [ <NUM_LIT> , <NUM_LIT> , <NUM_LIT> , <NUM_LIT> , <NUM_LIT> , <NUM_LIT:0.1> ] , <EOL> "<STR_LIT>" : [ <NUM_LIT> , <NUM_LIT> , <NUM_LIT> , <NUM_LIT> , <NUM_LIT> , <NUM_LIT> ] , <EOL> "<STR_LIT:id>" : [ <NUM_LIT:0> , <NUM_LIT:1> , <NUM_LIT:2> , <NUM_LIT:0> , <NUM_LIT:1> , <NUM_LIT:2> ] } <EOL> exp_frame = DataFrame ( exp_data ) <EOL> exp_frame = exp_frame . set_index ( [ '<STR_LIT:id>' , '<STR_LIT>' ] ) [ [ "<STR_LIT:X>" , "<STR_LIT:A>" , "<STR_LIT:B>" ] ] <EOL> long_frame = wide_to_long ( df , [ "<STR_LIT:A>" , "<STR_LIT:B>" ] , i = "<STR_LIT:id>" , j = "<STR_LIT>" ) <EOL> tm . assert_frame_equal ( long_frame , exp_frame ) <EOL> def test_stubs ( self ) : <EOL> df = pd . DataFrame ( [ [ <NUM_LIT:0> , <NUM_LIT:1> , <NUM_LIT:2> , <NUM_LIT:3> , <NUM_LIT:8> ] , [ <NUM_LIT:4> , <NUM_LIT:5> , <NUM_LIT:6> , <NUM_LIT:7> , <NUM_LIT:9> ] ] ) <EOL> df . columns = [ '<STR_LIT:id>' , '<STR_LIT>' , '<STR_LIT>' , '<STR_LIT>' , '<STR_LIT>' ] <EOL> stubs = [ '<STR_LIT>' , '<STR_LIT>' ] <EOL> df_long = pd . wide_to_long ( df , stubs , i = '<STR_LIT:id>' , j = '<STR_LIT>' ) <EOL> self . assertEqual ( stubs , [ '<STR_LIT>' , '<STR_LIT>' ] ) <EOL> if __name__ == '<STR_LIT:__main__>' : <EOL> nose . runmodule ( argv = [ __file__ , '<STR_LIT>' , '<STR_LIT>' , '<STR_LIT>' , '<STR_LIT>' ] , <EOL> exit = False ) </s>
