<s> """<STR_LIT>""" <EOL> import pandas as pd <EOL> import nose . tools as nt <EOL> from vincent . charts import ( data_type , Chart , Bar , Scatter , Line , Area , <EOL> GroupedBar , Map , Pie , Word ) <EOL> def chart_runner ( chart , scales , axes , marks ) : <EOL> """<STR_LIT>""" <EOL> for i , scale in enumerate ( scales ) : <EOL> nt . assert_dict_equal ( chart . scales [ i ] . grammar ( ) , scale ) <EOL> for i , axis in enumerate ( axes ) : <EOL> nt . assert_dict_equal ( chart . axes [ i ] . grammar ( ) , axis ) <EOL> for i , mark in enumerate ( marks ) : <EOL> nt . assert_dict_equal ( chart . marks [ i ] . grammar ( ) , mark ) <EOL> def test_data_type ( ) : <EOL> """<STR_LIT>""" <EOL> puts1 = [ <NUM_LIT:10> , <NUM_LIT:20> , <NUM_LIT:30> , <NUM_LIT> , <NUM_LIT:50> ] <EOL> puts2 = { '<STR_LIT>' : <NUM_LIT:10> , '<STR_LIT>' : <NUM_LIT:20> , '<STR_LIT>' : <NUM_LIT:30> } <EOL> gets1 = [ { '<STR_LIT>' : '<STR_LIT:data>' , '<STR_LIT>' : <NUM_LIT:0> , '<STR_LIT>' : <NUM_LIT:10> } , <EOL> { '<STR_LIT>' : '<STR_LIT:data>' , '<STR_LIT>' : <NUM_LIT:1> , '<STR_LIT>' : <NUM_LIT:20> } , <EOL> { '<STR_LIT>' : '<STR_LIT:data>' , '<STR_LIT>' : <NUM_LIT:2> , '<STR_LIT>' : <NUM_LIT:30> } , <EOL> { '<STR_LIT>' : '<STR_LIT:data>' , '<STR_LIT>' : <NUM_LIT:3> , '<STR_LIT>' : <NUM_LIT> } , <EOL> { '<STR_LIT>' : '<STR_LIT:data>' , '<STR_LIT>' : <NUM_LIT:4> , '<STR_LIT>' : <NUM_LIT:50> } ] <EOL> gets2 = [ { '<STR_LIT>' : '<STR_LIT:data>' , '<STR_LIT>' : '<STR_LIT>' , '<STR_LIT>' : <NUM_LIT:10> } , <EOL> { '<STR_LIT>' : '<STR_LIT:data>' , '<STR_LIT>' : '<STR_LIT>' , '<STR_LIT>' : <NUM_LIT:20> } , <EOL> { '<STR_LIT>' : '<STR_LIT:data>' , '<STR_LIT>' : '<STR_LIT>' , '<STR_LIT>' : <NUM_LIT:30> } ] <EOL> for ins , outs in zip ( [ puts1 , puts2 ] , [ gets1 , gets2 ] ) : <EOL> test = data_type ( ins ) <EOL> nt . assert_list_equal ( test . values , outs ) <EOL> puts = { '<STR_LIT:x>' : [ <NUM_LIT:1> , <NUM_LIT:2> , <NUM_LIT:3> ] , '<STR_LIT:y>' : [ <NUM_LIT:10> , <NUM_LIT:20> , <NUM_LIT:30> ] , '<STR_LIT:z>' : [ <NUM_LIT> , <NUM_LIT:50> , <NUM_LIT> ] } <EOL> gets = [ { '<STR_LIT>' : '<STR_LIT:y>' , '<STR_LIT>' : <NUM_LIT:1> , '<STR_LIT>' : <NUM_LIT:10> } , <EOL> { '<STR_LIT>' : '<STR_LIT:y>' , '<STR_LIT>' : <NUM_LIT:2> , '<STR_LIT>' : <NUM_LIT:20> } , <EOL> { '<STR_LIT>' : '<STR_LIT:y>' , '<STR_LIT>' : <NUM_LIT:3> , '<STR_LIT>' : <NUM_LIT:30> } , <EOL> { '<STR_LIT>' : '<STR_LIT:z>' , '<STR_LIT>' : <NUM_LIT:1> , '<STR_LIT>' : <NUM_LIT> } , <EOL> { '<STR_LIT>' : '<STR_LIT:z>' , '<STR_LIT>' : <NUM_LIT:2> , '<STR_LIT>' : <NUM_LIT:50> } , <EOL> { '<STR_LIT>' : '<STR_LIT:z>' , '<STR_LIT>' : <NUM_LIT:3> , '<STR_LIT>' : <NUM_LIT> } ] <EOL> test = data_type ( puts , iter_idx = '<STR_LIT:x>' ) <EOL> nt . assert_list_equal ( test . values , gets ) <EOL> df = pd . DataFrame ( { '<STR_LIT>' : [ <NUM_LIT:1> , <NUM_LIT:2> , <NUM_LIT:3> ] , '<STR_LIT>' : [ <NUM_LIT:4> , <NUM_LIT:5> , <NUM_LIT:6> ] } ) <EOL> series = pd . Series ( [ <NUM_LIT:1> , <NUM_LIT:2> , <NUM_LIT:3> ] , name = '<STR_LIT:test>' ) <EOL> gets1 = [ { '<STR_LIT>' : '<STR_LIT>' , '<STR_LIT>' : <NUM_LIT:0> , '<STR_LIT>' : <NUM_LIT:1> } , <EOL> { '<STR_LIT>' : '<STR_LIT>' , '<STR_LIT>' : <NUM_LIT:0> , '<STR_LIT>' : <NUM_LIT:4> } , <EOL> { '<STR_LIT>' : '<STR_LIT>' , '<STR_LIT>' : <NUM_LIT:1> , '<STR_LIT>' : <NUM_LIT:2> } , <EOL> { '<STR_LIT>' : '<STR_LIT>' , '<STR_LIT>' : <NUM_LIT:1> , '<STR_LIT>' : <NUM_LIT:5> } , <EOL> { '<STR_LIT>' : '<STR_LIT>' , '<STR_LIT>' : <NUM_LIT:2> , '<STR_LIT>' : <NUM_LIT:3> } , <EOL> { '<STR_LIT>' : '<STR_LIT>' , '<STR_LIT>' : <NUM_LIT:2> , '<STR_LIT>' : <NUM_LIT:6> } ] <EOL> gets2 = [ { '<STR_LIT>' : '<STR_LIT:test>' , '<STR_LIT>' : <NUM_LIT:0> , '<STR_LIT>' : <NUM_LIT:1> } , <EOL> { '<STR_LIT>' : '<STR_LIT:test>' , '<STR_LIT>' : <NUM_LIT:1> , '<STR_LIT>' : <NUM_LIT:2> } , <EOL> { '<STR_LIT>' : '<STR_LIT:test>' , '<STR_LIT>' : <NUM_LIT:2> , '<STR_LIT>' : <NUM_LIT:3> } ] <EOL> test_df = data_type ( df ) <EOL> test_series = data_type ( series ) <EOL> nt . assert_list_equal ( test_df . values , gets1 ) <EOL> nt . assert_list_equal ( test_series . values , gets2 ) <EOL> class BadType ( object ) : <EOL> '<STR_LIT>' <EOL> pass <EOL> test = BadType ( ) <EOL> nt . assert_raises ( ValueError , data_type , test ) <EOL> class TestChart ( object ) : <EOL> """<STR_LIT>""" <EOL> def test_init ( self ) : <EOL> chart = Chart ( [ <NUM_LIT:0> , <NUM_LIT:1> ] , width = <NUM_LIT:100> , height = <NUM_LIT:100> ) <EOL> nt . assert_equal ( chart . width , <NUM_LIT:100> ) <EOL> nt . assert_equal ( chart . height , <NUM_LIT:100> ) <EOL> padding = "<STR_LIT>" <EOL> nt . assert_equal ( chart . padding , padding ) <EOL> nt . assert_raises ( ValueError , Chart ) <EOL> nt . assert_raises ( ValueError , Chart , [ ] ) <EOL> class TestScatter ( object ) : <EOL> """<STR_LIT>""" <EOL> def test_init ( self ) : <EOL> scatter = Scatter ( [ <NUM_LIT:1> , <NUM_LIT:2> , <NUM_LIT:3> ] ) <EOL> scales = [ { '<STR_LIT>' : { '<STR_LIT:data>' : '<STR_LIT>' , '<STR_LIT>' : '<STR_LIT>' } , <EOL> '<STR_LIT:name>' : '<STR_LIT:x>' , <EOL> '<STR_LIT>' : '<STR_LIT:width>' , <EOL> '<STR_LIT:type>' : '<STR_LIT>' } , <EOL> { '<STR_LIT>' : { '<STR_LIT:data>' : '<STR_LIT>' , '<STR_LIT>' : '<STR_LIT>' } , <EOL> '<STR_LIT:name>' : '<STR_LIT:y>' , <EOL> '<STR_LIT>' : '<STR_LIT>' , <EOL> '<STR_LIT>' : True } , <EOL> { '<STR_LIT>' : { '<STR_LIT:data>' : '<STR_LIT>' , '<STR_LIT>' : '<STR_LIT>' } , <EOL> '<STR_LIT:name>' : '<STR_LIT>' , <EOL> '<STR_LIT>' : '<STR_LIT>' , <EOL> '<STR_LIT:type>' : '<STR_LIT>' } ] <EOL> axes = [ { '<STR_LIT>' : '<STR_LIT:x>' , '<STR_LIT:type>' : '<STR_LIT:x>' } , <EOL> { '<STR_LIT>' : '<STR_LIT:y>' , '<STR_LIT:type>' : '<STR_LIT:y>' } ] <EOL> marks = [ { <EOL> '<STR_LIT:type>' : '<STR_LIT>' , <EOL> '<STR_LIT>' : { <EOL> '<STR_LIT:data>' : '<STR_LIT>' , <EOL> '<STR_LIT>' : [ <EOL> { '<STR_LIT>' : [ '<STR_LIT>' ] , '<STR_LIT:type>' : '<STR_LIT>' } <EOL> ] <EOL> } , <EOL> '<STR_LIT>' : [ { <EOL> '<STR_LIT:type>' : '<STR_LIT>' , <EOL> '<STR_LIT>' : { <EOL> '<STR_LIT>' : { <EOL> '<STR_LIT>' : { '<STR_LIT>' : '<STR_LIT>' , '<STR_LIT>' : '<STR_LIT>' } , <EOL> '<STR_LIT:size>' : { '<STR_LIT:value>' : <NUM_LIT:100> } , <EOL> '<STR_LIT:x>' : { '<STR_LIT>' : '<STR_LIT>' , '<STR_LIT>' : '<STR_LIT:x>' } , <EOL> '<STR_LIT:y>' : { '<STR_LIT>' : '<STR_LIT>' , '<STR_LIT>' : '<STR_LIT:y>' } } } , <EOL> } ] <EOL> } ] <EOL> chart_runner ( scatter , scales , axes , marks ) <EOL> class TestLine ( object ) : <EOL> """<STR_LIT>""" <EOL> def test_init ( self ) : <EOL> line = Line ( [ <NUM_LIT:1> , <NUM_LIT:2> , <NUM_LIT:3> ] ) <EOL> scales = [ { '<STR_LIT>' : { '<STR_LIT:data>' : '<STR_LIT>' , '<STR_LIT>' : '<STR_LIT>' } , <EOL> '<STR_LIT:name>' : '<STR_LIT:x>' , <EOL> '<STR_LIT:type>' : '<STR_LIT>' , <EOL> '<STR_LIT>' : '<STR_LIT:width>' } , <EOL> { '<STR_LIT>' : { '<STR_LIT:data>' : '<STR_LIT>' , '<STR_LIT>' : '<STR_LIT>' } , <EOL> '<STR_LIT:name>' : '<STR_LIT:y>' , <EOL> '<STR_LIT>' : True , <EOL> '<STR_LIT>' : '<STR_LIT>' } , <EOL> { '<STR_LIT>' : { '<STR_LIT:data>' : '<STR_LIT>' , '<STR_LIT>' : '<STR_LIT>' } , <EOL> '<STR_LIT:name>' : '<STR_LIT>' , <EOL> '<STR_LIT>' : '<STR_LIT>' , <EOL> '<STR_LIT:type>' : '<STR_LIT>' } ] <EOL> axes = [ { '<STR_LIT>' : '<STR_LIT:x>' , '<STR_LIT:type>' : '<STR_LIT:x>' } , <EOL> { '<STR_LIT>' : '<STR_LIT:y>' , '<STR_LIT:type>' : '<STR_LIT:y>' } ] <EOL> marks = [ { <EOL> '<STR_LIT:type>' : '<STR_LIT>' , <EOL> '<STR_LIT>' : { <EOL> '<STR_LIT:data>' : '<STR_LIT>' , <EOL> '<STR_LIT>' : [ <EOL> { '<STR_LIT>' : [ '<STR_LIT>' ] , '<STR_LIT:type>' : '<STR_LIT>' } <EOL> ] <EOL> } , <EOL> '<STR_LIT>' : [ { <EOL> '<STR_LIT:type>' : '<STR_LIT>' , <EOL> '<STR_LIT>' : { <EOL> '<STR_LIT>' : { <EOL> '<STR_LIT>' : { '<STR_LIT>' : '<STR_LIT>' , '<STR_LIT>' : '<STR_LIT>' } , <EOL> '<STR_LIT>' : { '<STR_LIT:value>' : <NUM_LIT:2> } , <EOL> '<STR_LIT:x>' : { '<STR_LIT>' : '<STR_LIT>' , '<STR_LIT>' : '<STR_LIT:x>' } , <EOL> '<STR_LIT:y>' : { '<STR_LIT>' : '<STR_LIT>' , '<STR_LIT>' : '<STR_LIT:y>' } <EOL> } <EOL> } <EOL> } ] <EOL> } ] <EOL> chart_runner ( line , scales , axes , marks ) <EOL> class TestArea ( object ) : <EOL> """<STR_LIT>""" <EOL> def test_init ( self ) : <EOL> area = Area ( [ <NUM_LIT:1> , <NUM_LIT:2> , <NUM_LIT:3> ] ) <EOL> stacked_area = Area ( { '<STR_LIT:x>' : [ <NUM_LIT:1> , <NUM_LIT:2> , <NUM_LIT:3> ] , '<STR_LIT:y>' : [ <NUM_LIT:4> , <NUM_LIT:5> , <NUM_LIT:6> ] , '<STR_LIT:z>' : [ <NUM_LIT:7> , <NUM_LIT:8> , <NUM_LIT:9> ] } , <EOL> iter_idx = '<STR_LIT:x>' ) <EOL> datas = [ <EOL> { '<STR_LIT:name>' : '<STR_LIT>' , <EOL> '<STR_LIT>' : [ <EOL> { '<STR_LIT>' : '<STR_LIT:y>' , '<STR_LIT>' : <NUM_LIT:1> , '<STR_LIT>' : <NUM_LIT:4> } , <EOL> { '<STR_LIT>' : '<STR_LIT:y>' , '<STR_LIT>' : <NUM_LIT:2> , '<STR_LIT>' : <NUM_LIT:5> } , <EOL> { '<STR_LIT>' : '<STR_LIT:y>' , '<STR_LIT>' : <NUM_LIT:3> , '<STR_LIT>' : <NUM_LIT:6> } , <EOL> { '<STR_LIT>' : '<STR_LIT:z>' , '<STR_LIT>' : <NUM_LIT:1> , '<STR_LIT>' : <NUM_LIT:7> } , <EOL> { '<STR_LIT>' : '<STR_LIT:z>' , '<STR_LIT>' : <NUM_LIT:2> , '<STR_LIT>' : <NUM_LIT:8> } , <EOL> { '<STR_LIT>' : '<STR_LIT:z>' , '<STR_LIT>' : <NUM_LIT:3> , '<STR_LIT>' : <NUM_LIT:9> } ] } , <EOL> { '<STR_LIT:name>' : '<STR_LIT>' , <EOL> '<STR_LIT:source>' : '<STR_LIT>' , <EOL> '<STR_LIT>' : [ <EOL> { '<STR_LIT:type>' : '<STR_LIT>' , '<STR_LIT>' : [ '<STR_LIT>' ] } , <EOL> { '<STR_LIT:type>' : '<STR_LIT>' , '<STR_LIT:value>' : '<STR_LIT>' } ] } <EOL> ] <EOL> for i , data in enumerate ( datas ) : <EOL> nt . assert_dict_equal ( stacked_area . data [ i ] . grammar ( ) , data ) <EOL> scales = [ { '<STR_LIT>' : { '<STR_LIT:data>' : '<STR_LIT>' , '<STR_LIT>' : '<STR_LIT>' } , <EOL> '<STR_LIT:name>' : '<STR_LIT:x>' , <EOL> '<STR_LIT>' : '<STR_LIT:width>' , <EOL> '<STR_LIT>' : False , <EOL> '<STR_LIT:type>' : '<STR_LIT>' } , <EOL> { '<STR_LIT>' : { '<STR_LIT:data>' : '<STR_LIT>' , '<STR_LIT>' : '<STR_LIT>' } , <EOL> '<STR_LIT:name>' : '<STR_LIT:y>' , <EOL> '<STR_LIT>' : True , <EOL> '<STR_LIT>' : '<STR_LIT>' } , <EOL> { '<STR_LIT>' : { '<STR_LIT:data>' : '<STR_LIT>' , '<STR_LIT>' : '<STR_LIT>' } , <EOL> '<STR_LIT:name>' : '<STR_LIT>' , <EOL> '<STR_LIT>' : '<STR_LIT>' , <EOL> '<STR_LIT:type>' : '<STR_LIT>' } ] <EOL> axes = [ { '<STR_LIT>' : '<STR_LIT:x>' , '<STR_LIT:type>' : '<STR_LIT:x>' } , <EOL> { '<STR_LIT>' : '<STR_LIT:y>' , '<STR_LIT:type>' : '<STR_LIT:y>' } ] <EOL> marks = [ { <EOL> '<STR_LIT:type>' : '<STR_LIT>' , <EOL> '<STR_LIT>' : { <EOL> '<STR_LIT:data>' : '<STR_LIT>' , <EOL> '<STR_LIT>' : [ <EOL> { '<STR_LIT:type>' : '<STR_LIT>' , '<STR_LIT>' : [ '<STR_LIT>' ] } , <EOL> { '<STR_LIT:type>' : '<STR_LIT>' , '<STR_LIT>' : '<STR_LIT>' , <EOL> '<STR_LIT>' : '<STR_LIT>' } ] <EOL> } , <EOL> '<STR_LIT>' : [ { <EOL> '<STR_LIT:type>' : '<STR_LIT>' , <EOL> '<STR_LIT>' : { <EOL> '<STR_LIT>' : { <EOL> '<STR_LIT:x>' : { '<STR_LIT>' : '<STR_LIT>' , '<STR_LIT>' : '<STR_LIT:x>' } , <EOL> '<STR_LIT:y>' : { '<STR_LIT>' : '<STR_LIT:y>' , '<STR_LIT>' : '<STR_LIT:y>' } , <EOL> '<STR_LIT>' : { '<STR_LIT>' : '<STR_LIT>' , '<STR_LIT>' : '<STR_LIT:y>' } , <EOL> '<STR_LIT>' : { '<STR_LIT>' : '<STR_LIT>' , '<STR_LIT>' : '<STR_LIT>' } , <EOL> '<STR_LIT>' : { '<STR_LIT:value>' : '<STR_LIT>' } <EOL> } <EOL> } <EOL> } ] <EOL> } ] <EOL> chart_runner ( area , scales , axes , marks ) <EOL> chart_runner ( stacked_area , scales , axes , marks ) <EOL> class TestBar ( object ) : <EOL> """<STR_LIT>""" <EOL> def test_init ( self ) : <EOL> bar = Bar ( [ <NUM_LIT:1> , <NUM_LIT:2> , <NUM_LIT:3> ] ) <EOL> stacked_bar = Bar ( { '<STR_LIT:x>' : [ <NUM_LIT:1> , <NUM_LIT:2> , <NUM_LIT:3> ] , '<STR_LIT:y>' : [ <NUM_LIT:4> , <NUM_LIT:5> , <NUM_LIT:6> ] , '<STR_LIT:z>' : [ <NUM_LIT:7> , <NUM_LIT:8> , <NUM_LIT:9> ] } , <EOL> iter_idx = '<STR_LIT:x>' ) <EOL> datas = [ <EOL> { '<STR_LIT:name>' : '<STR_LIT>' , <EOL> '<STR_LIT>' : [ <EOL> { '<STR_LIT>' : '<STR_LIT:y>' , '<STR_LIT>' : <NUM_LIT:1> , '<STR_LIT>' : <NUM_LIT:4> } , <EOL> { '<STR_LIT>' : '<STR_LIT:y>' , '<STR_LIT>' : <NUM_LIT:2> , '<STR_LIT>' : <NUM_LIT:5> } , <EOL> { '<STR_LIT>' : '<STR_LIT:y>' , '<STR_LIT>' : <NUM_LIT:3> , '<STR_LIT>' : <NUM_LIT:6> } , <EOL> { '<STR_LIT>' : '<STR_LIT:z>' , '<STR_LIT>' : <NUM_LIT:1> , '<STR_LIT>' : <NUM_LIT:7> } , <EOL> { '<STR_LIT>' : '<STR_LIT:z>' , '<STR_LIT>' : <NUM_LIT:2> , '<STR_LIT>' : <NUM_LIT:8> } , <EOL> { '<STR_LIT>' : '<STR_LIT:z>' , '<STR_LIT>' : <NUM_LIT:3> , '<STR_LIT>' : <NUM_LIT:9> } ] } , <EOL> { '<STR_LIT:name>' : '<STR_LIT>' , <EOL> '<STR_LIT:source>' : '<STR_LIT>' , <EOL> '<STR_LIT>' : [ <EOL> { '<STR_LIT:type>' : '<STR_LIT>' , '<STR_LIT>' : [ '<STR_LIT>' ] } , <EOL> { '<STR_LIT:type>' : '<STR_LIT>' , '<STR_LIT:value>' : '<STR_LIT>' } ] } <EOL> ] <EOL> for i , data in enumerate ( datas ) : <EOL> nt . assert_dict_equal ( stacked_bar . data [ i ] . grammar ( ) , data ) <EOL> scales = [ { '<STR_LIT>' : { '<STR_LIT:data>' : '<STR_LIT>' , '<STR_LIT>' : '<STR_LIT>' } , <EOL> '<STR_LIT:name>' : '<STR_LIT:x>' , <EOL> '<STR_LIT>' : '<STR_LIT:width>' , <EOL> '<STR_LIT>' : False , <EOL> '<STR_LIT:type>' : '<STR_LIT>' } , <EOL> { '<STR_LIT>' : { '<STR_LIT:data>' : '<STR_LIT>' , '<STR_LIT>' : '<STR_LIT>' } , <EOL> '<STR_LIT:name>' : '<STR_LIT:y>' , <EOL> '<STR_LIT>' : True , <EOL> '<STR_LIT>' : '<STR_LIT>' } , <EOL> { '<STR_LIT>' : { '<STR_LIT:data>' : '<STR_LIT>' , '<STR_LIT>' : '<STR_LIT>' } , <EOL> '<STR_LIT:name>' : '<STR_LIT>' , <EOL> '<STR_LIT>' : '<STR_LIT>' , <EOL> '<STR_LIT:type>' : '<STR_LIT>' } ] <EOL> axes = [ { '<STR_LIT>' : '<STR_LIT:x>' , '<STR_LIT:type>' : '<STR_LIT:x>' } , <EOL> { '<STR_LIT>' : '<STR_LIT:y>' , '<STR_LIT:type>' : '<STR_LIT:y>' } ] <EOL> marks = [ { <EOL> '<STR_LIT:type>' : '<STR_LIT>' , <EOL> '<STR_LIT>' : { <EOL> '<STR_LIT:data>' : '<STR_LIT>' , <EOL> '<STR_LIT>' : [ <EOL> { '<STR_LIT:type>' : '<STR_LIT>' , '<STR_LIT>' : [ '<STR_LIT>' ] } , <EOL> { '<STR_LIT:type>' : '<STR_LIT>' , '<STR_LIT>' : '<STR_LIT>' , <EOL> '<STR_LIT>' : '<STR_LIT>' } ] <EOL> } , <EOL> '<STR_LIT>' : [ { <EOL> '<STR_LIT:type>' : '<STR_LIT>' , <EOL> '<STR_LIT>' : { <EOL> '<STR_LIT>' : { <EOL> '<STR_LIT:x>' : { '<STR_LIT>' : '<STR_LIT>' , '<STR_LIT>' : '<STR_LIT:x>' } , <EOL> '<STR_LIT:width>' : { '<STR_LIT>' : True , '<STR_LIT>' : - <NUM_LIT:1> , '<STR_LIT>' : '<STR_LIT:x>' } , <EOL> '<STR_LIT:y>' : { '<STR_LIT>' : '<STR_LIT:y>' , '<STR_LIT>' : '<STR_LIT:y>' } , <EOL> '<STR_LIT>' : { '<STR_LIT>' : '<STR_LIT>' , '<STR_LIT>' : '<STR_LIT:y>' } , <EOL> '<STR_LIT>' : { '<STR_LIT>' : '<STR_LIT>' , '<STR_LIT>' : '<STR_LIT>' } <EOL> } <EOL> } <EOL> } ] <EOL> } ] <EOL> chart_runner ( bar , scales , axes , marks ) <EOL> chart_runner ( stacked_bar , scales , axes , marks ) <EOL> class TestGroupedBar ( object ) : <EOL> """<STR_LIT>""" <EOL> def test_init ( self ) : <EOL> farm_1 = { '<STR_LIT>' : <NUM_LIT:10> , '<STR_LIT>' : <NUM_LIT:32> , '<STR_LIT>' : <NUM_LIT> } <EOL> farm_2 = { '<STR_LIT>' : <NUM_LIT:15> , '<STR_LIT>' : <NUM_LIT> , '<STR_LIT>' : <NUM_LIT> } <EOL> data = [ farm_1 , farm_2 ] <EOL> index = [ '<STR_LIT>' , '<STR_LIT>' ] <EOL> df = pd . DataFrame ( data , index = index ) <EOL> group = GroupedBar ( df ) <EOL> datas = [ { <EOL> '<STR_LIT:name>' : '<STR_LIT>' , <EOL> '<STR_LIT>' : [ <EOL> { '<STR_LIT>' : '<STR_LIT>' , '<STR_LIT>' : '<STR_LIT>' , '<STR_LIT>' : <NUM_LIT:10> } , <EOL> { '<STR_LIT>' : '<STR_LIT>' , '<STR_LIT>' : '<STR_LIT>' , '<STR_LIT>' : <NUM_LIT:32> } , <EOL> { '<STR_LIT>' : '<STR_LIT>' , '<STR_LIT>' : '<STR_LIT>' , '<STR_LIT>' : <NUM_LIT> } , <EOL> { '<STR_LIT>' : '<STR_LIT>' , '<STR_LIT>' : '<STR_LIT>' , '<STR_LIT>' : <NUM_LIT:15> } , <EOL> { '<STR_LIT>' : '<STR_LIT>' , '<STR_LIT>' : '<STR_LIT>' , '<STR_LIT>' : <NUM_LIT> } , <EOL> { '<STR_LIT>' : '<STR_LIT>' , '<STR_LIT>' : '<STR_LIT>' , '<STR_LIT>' : <NUM_LIT> } <EOL> ] <EOL> } ] <EOL> for i , data in enumerate ( datas ) : <EOL> nt . assert_dict_equal ( group . data [ i ] . grammar ( ) , data ) <EOL> scales = [ { '<STR_LIT>' : { '<STR_LIT:data>' : '<STR_LIT>' , '<STR_LIT>' : '<STR_LIT>' } , <EOL> '<STR_LIT:name>' : '<STR_LIT:x>' , <EOL> '<STR_LIT>' : <NUM_LIT> , <EOL> '<STR_LIT>' : '<STR_LIT:width>' , <EOL> '<STR_LIT:type>' : '<STR_LIT>' } , <EOL> { '<STR_LIT>' : { '<STR_LIT:data>' : '<STR_LIT>' , '<STR_LIT>' : '<STR_LIT>' } , <EOL> '<STR_LIT:name>' : '<STR_LIT:y>' , <EOL> '<STR_LIT>' : True , <EOL> '<STR_LIT>' : '<STR_LIT>' } , <EOL> { '<STR_LIT>' : { '<STR_LIT:data>' : '<STR_LIT>' , '<STR_LIT>' : '<STR_LIT>' } , <EOL> '<STR_LIT:name>' : '<STR_LIT>' , <EOL> '<STR_LIT>' : '<STR_LIT>' , <EOL> '<STR_LIT:type>' : '<STR_LIT>' } ] <EOL> axes = [ { '<STR_LIT>' : '<STR_LIT:x>' , '<STR_LIT:type>' : '<STR_LIT:x>' } , <EOL> { '<STR_LIT>' : '<STR_LIT:y>' , '<STR_LIT:type>' : '<STR_LIT:y>' } ] <EOL> marks = [ { <EOL> '<STR_LIT:type>' : '<STR_LIT>' , <EOL> '<STR_LIT>' : { <EOL> '<STR_LIT:data>' : '<STR_LIT>' , <EOL> '<STR_LIT>' : [ { '<STR_LIT>' : [ '<STR_LIT>' ] , '<STR_LIT:type>' : '<STR_LIT>' } ] <EOL> } , <EOL> '<STR_LIT>' : [ { <EOL> '<STR_LIT:type>' : '<STR_LIT>' , <EOL> '<STR_LIT>' : { <EOL> '<STR_LIT>' : { <EOL> '<STR_LIT>' : { '<STR_LIT>' : '<STR_LIT>' , '<STR_LIT>' : '<STR_LIT>' } , <EOL> '<STR_LIT:width>' : { '<STR_LIT>' : True , '<STR_LIT>' : - <NUM_LIT:1> , '<STR_LIT>' : '<STR_LIT>' } , <EOL> '<STR_LIT:x>' : { '<STR_LIT>' : '<STR_LIT>' , '<STR_LIT>' : '<STR_LIT>' } , <EOL> '<STR_LIT:y>' : { '<STR_LIT>' : '<STR_LIT>' , '<STR_LIT>' : '<STR_LIT:y>' } , <EOL> '<STR_LIT>' : { '<STR_LIT>' : '<STR_LIT:y>' , '<STR_LIT:value>' : <NUM_LIT:0> } } <EOL> } <EOL> } ] , <EOL> '<STR_LIT>' : { <EOL> '<STR_LIT>' : { <EOL> '<STR_LIT:width>' : { '<STR_LIT>' : True , '<STR_LIT>' : '<STR_LIT:x>' } , <EOL> '<STR_LIT:x>' : { '<STR_LIT>' : '<STR_LIT:key>' , '<STR_LIT>' : '<STR_LIT:x>' } <EOL> } <EOL> } , <EOL> '<STR_LIT>' : [ { <EOL> '<STR_LIT>' : { '<STR_LIT>' : '<STR_LIT>' } , <EOL> '<STR_LIT:name>' : '<STR_LIT>' , <EOL> '<STR_LIT>' : '<STR_LIT:width>' , <EOL> '<STR_LIT:type>' : '<STR_LIT>' <EOL> } ] <EOL> } ] <EOL> chart_runner ( group , scales , axes , marks ) <EOL> class TestPie ( object ) : <EOL> """<STR_LIT>""" <EOL> def test_init ( self ) : <EOL> pie = Pie ( [ <NUM_LIT:12> , <NUM_LIT> , <NUM_LIT> ] ) <EOL> axes = [ ] <EOL> scales = [ { <EOL> "<STR_LIT>" : { "<STR_LIT:data>" : "<STR_LIT>" , "<STR_LIT>" : "<STR_LIT>" } , <EOL> "<STR_LIT:name>" : "<STR_LIT>" , <EOL> "<STR_LIT>" : "<STR_LIT>" , <EOL> "<STR_LIT:type>" : "<STR_LIT>" <EOL> } ] <EOL> marks = [ { <EOL> "<STR_LIT:type>" : "<STR_LIT>" , <EOL> "<STR_LIT>" : { <EOL> "<STR_LIT:data>" : "<STR_LIT>" , <EOL> "<STR_LIT>" : [ { "<STR_LIT:type>" : "<STR_LIT>" , "<STR_LIT:value>" : "<STR_LIT>" } ] <EOL> } , <EOL> "<STR_LIT>" : { <EOL> "<STR_LIT>" : { <EOL> "<STR_LIT:x>" : { "<STR_LIT>" : "<STR_LIT:width>" , "<STR_LIT>" : <NUM_LIT:0.5> } , <EOL> "<STR_LIT:y>" : { "<STR_LIT>" : "<STR_LIT>" , "<STR_LIT>" : <NUM_LIT:0.5> } , <EOL> "<STR_LIT>" : { "<STR_LIT>" : "<STR_LIT>" } , <EOL> "<STR_LIT>" : { "<STR_LIT:value>" : <NUM_LIT:0> } , <EOL> "<STR_LIT>" : { "<STR_LIT:value>" : <NUM_LIT> } , <EOL> "<STR_LIT>" : { "<STR_LIT>" : "<STR_LIT>" } , <EOL> "<STR_LIT>" : { "<STR_LIT:value>" : "<STR_LIT>" } , <EOL> "<STR_LIT>" : { "<STR_LIT>" : "<STR_LIT>" , "<STR_LIT>" : "<STR_LIT>" } <EOL> } <EOL> } <EOL> } ] <EOL> chart_runner ( pie , scales , axes , marks ) <EOL> class TestMaps ( object ) : <EOL> """<STR_LIT>""" <EOL> def setup ( self ) : <EOL> self . df = pd . DataFrame ( { '<STR_LIT>' : [ <NUM_LIT:1> , <NUM_LIT:2> , <NUM_LIT:3> ] , '<STR_LIT>' : [ <NUM_LIT:4> , <NUM_LIT:5> , <NUM_LIT:6> ] , <EOL> '<STR_LIT>' : [ <NUM_LIT:7> , <NUM_LIT:8> , <NUM_LIT:9> ] } ) <EOL> self . series = pd . Series ( [ <NUM_LIT:1> , <NUM_LIT:2> , <NUM_LIT:3> ] , name = '<STR_LIT:test>' ) <EOL> self . geo_data = [ { <EOL> '<STR_LIT:name>' : '<STR_LIT>' , <EOL> '<STR_LIT:url>' : '<STR_LIT>' , <EOL> '<STR_LIT>' : '<STR_LIT>' <EOL> } , <EOL> { <EOL> '<STR_LIT:name>' : '<STR_LIT>' , <EOL> '<STR_LIT:url>' : '<STR_LIT>' , <EOL> '<STR_LIT>' : '<STR_LIT>' <EOL> } ] <EOL> def test_simple ( self ) : <EOL> map_ = Map ( geo_data = [ self . geo_data [ <NUM_LIT:0> ] ] , projection = '<STR_LIT>' , <EOL> center = [ <NUM_LIT:10> , <NUM_LIT:20> ] , scale = <NUM_LIT:200> , rotate = <NUM_LIT:10> ) <EOL> axes = [ ] <EOL> scales = [ ] <EOL> datas = [ { '<STR_LIT:name>' : '<STR_LIT>' , <EOL> '<STR_LIT:url>' : '<STR_LIT>' , <EOL> '<STR_LIT>' : { '<STR_LIT>' : '<STR_LIT>' , '<STR_LIT:type>' : '<STR_LIT>' } , <EOL> '<STR_LIT>' : [ { <EOL> '<STR_LIT>' : [ <NUM_LIT:10> , <NUM_LIT:20> ] , <EOL> '<STR_LIT>' : '<STR_LIT>' , <EOL> '<STR_LIT>' : <NUM_LIT:10> , <EOL> '<STR_LIT>' : <NUM_LIT:200> , <EOL> '<STR_LIT>' : [ <NUM_LIT> , <NUM_LIT> ] , <EOL> '<STR_LIT:type>' : '<STR_LIT>' , <EOL> '<STR_LIT:value>' : '<STR_LIT:data>' <EOL> } ] <EOL> } ] <EOL> marks = [ { '<STR_LIT:type>' : '<STR_LIT:path>' , <EOL> '<STR_LIT>' : { '<STR_LIT:data>' : '<STR_LIT>' } , <EOL> '<STR_LIT>' : { <EOL> '<STR_LIT>' : { <EOL> '<STR_LIT:path>' : { '<STR_LIT>' : '<STR_LIT:path>' } , <EOL> '<STR_LIT>' : { '<STR_LIT:value>' : '<STR_LIT>' } <EOL> } , <EOL> '<STR_LIT>' : { <EOL> '<STR_LIT>' : { '<STR_LIT:value>' : '<STR_LIT>' } <EOL> } <EOL> } <EOL> } ] <EOL> chart_runner ( map_ , scales , axes , marks ) <EOL> for i , data in enumerate ( datas ) : <EOL> nt . assert_dict_equal ( map_ . data [ i ] . grammar ( ) , data ) <EOL> def test_binding ( self ) : <EOL> with nt . assert_raises ( ValueError ) as err : <EOL> map_df = Map ( data = self . df , geo_data = self . geo_data , <EOL> projection = '<STR_LIT>' , center = [ <NUM_LIT:10> , <NUM_LIT:20> ] , scale = <NUM_LIT:200> , <EOL> rotate = <NUM_LIT:10> , map_key = { '<STR_LIT>' : '<STR_LIT>' } ) <EOL> nt . assert_equal ( <EOL> err . exception . args [ <NUM_LIT:0> ] , <EOL> '<STR_LIT>' <EOL> ) <EOL> map_df = Map ( data = self . df , geo_data = self . geo_data , <EOL> projection = '<STR_LIT>' , <EOL> center = [ <NUM_LIT:10> , <NUM_LIT:20> ] , scale = <NUM_LIT:200> , rotate = <NUM_LIT:10> , <EOL> map_key = { '<STR_LIT>' : '<STR_LIT>' } , data_key = '<STR_LIT>' , <EOL> data_bind = '<STR_LIT>' ) <EOL> axes = [ ] <EOL> scales = [ { <EOL> '<STR_LIT:name>' : '<STR_LIT>' , <EOL> '<STR_LIT:type>' : '<STR_LIT>' , <EOL> '<STR_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> } ] <EOL> datas = [ { '<STR_LIT:name>' : '<STR_LIT>' , <EOL> '<STR_LIT:url>' : '<STR_LIT>' , <EOL> '<STR_LIT>' : { '<STR_LIT>' : '<STR_LIT>' , '<STR_LIT:type>' : '<STR_LIT>' } , <EOL> '<STR_LIT>' : [ { <EOL> '<STR_LIT>' : [ <NUM_LIT:10> , <NUM_LIT:20> ] , <EOL> '<STR_LIT>' : '<STR_LIT>' , <EOL> '<STR_LIT>' : <NUM_LIT:10> , <EOL> '<STR_LIT>' : <NUM_LIT:200> , <EOL> '<STR_LIT>' : [ <NUM_LIT> , <NUM_LIT> ] , <EOL> '<STR_LIT:type>' : '<STR_LIT>' , <EOL> '<STR_LIT:value>' : '<STR_LIT:data>' <EOL> } ] , <EOL> } , <EOL> { '<STR_LIT:name>' : '<STR_LIT>' , <EOL> '<STR_LIT>' : [ <EOL> { '<STR_LIT:x>' : <NUM_LIT:1> , '<STR_LIT:y>' : <NUM_LIT:4> } , <EOL> { '<STR_LIT:x>' : <NUM_LIT:2> , '<STR_LIT:y>' : <NUM_LIT:5> } , <EOL> { '<STR_LIT:x>' : <NUM_LIT:3> , '<STR_LIT:y>' : <NUM_LIT:6> } <EOL> ] <EOL> } , <EOL> { '<STR_LIT:name>' : '<STR_LIT>' , <EOL> '<STR_LIT:url>' : '<STR_LIT>' , <EOL> '<STR_LIT>' : { '<STR_LIT>' : '<STR_LIT>' , '<STR_LIT:type>' : '<STR_LIT>' } , <EOL> '<STR_LIT>' : [ <EOL> { <EOL> '<STR_LIT>' : '<STR_LIT:value>' , <EOL> '<STR_LIT:default>' : '<STR_LIT>' , <EOL> '<STR_LIT:key>' : '<STR_LIT>' , <EOL> '<STR_LIT:type>' : '<STR_LIT>' , <EOL> '<STR_LIT>' : '<STR_LIT>' , <EOL> '<STR_LIT>' : '<STR_LIT>' <EOL> } , <EOL> { <EOL> '<STR_LIT:test>' : "<STR_LIT>" , <EOL> '<STR_LIT:type>' : '<STR_LIT>' <EOL> } , <EOL> { <EOL> '<STR_LIT>' : [ <NUM_LIT:10> , <NUM_LIT:20> ] , <EOL> '<STR_LIT>' : '<STR_LIT>' , <EOL> '<STR_LIT>' : <NUM_LIT:10> , <EOL> '<STR_LIT>' : <NUM_LIT:200> , <EOL> '<STR_LIT>' : [ <NUM_LIT> , <NUM_LIT> ] , <EOL> '<STR_LIT:type>' : '<STR_LIT>' , <EOL> '<STR_LIT:value>' : '<STR_LIT:data>' <EOL> } ] <EOL> } <EOL> ] <EOL> marks = [ { '<STR_LIT:type>' : '<STR_LIT:path>' , <EOL> '<STR_LIT>' : { '<STR_LIT:data>' : '<STR_LIT>' } , <EOL> '<STR_LIT>' : { <EOL> '<STR_LIT>' : { <EOL> '<STR_LIT:path>' : { '<STR_LIT>' : '<STR_LIT:path>' } , <EOL> '<STR_LIT>' : { '<STR_LIT:value>' : '<STR_LIT>' } <EOL> } , <EOL> '<STR_LIT>' : { <EOL> '<STR_LIT>' : { '<STR_LIT:value>' : '<STR_LIT>' } <EOL> } <EOL> } <EOL> } , <EOL> { '<STR_LIT:type>' : '<STR_LIT:path>' , <EOL> '<STR_LIT>' : { '<STR_LIT:data>' : '<STR_LIT>' } , <EOL> '<STR_LIT>' : { <EOL> '<STR_LIT>' : { <EOL> '<STR_LIT:path>' : { '<STR_LIT>' : '<STR_LIT:path>' } , <EOL> '<STR_LIT>' : { '<STR_LIT:value>' : '<STR_LIT>' } <EOL> } , <EOL> '<STR_LIT>' : { <EOL> '<STR_LIT>' : { '<STR_LIT>' : '<STR_LIT>' , '<STR_LIT>' : '<STR_LIT>' } <EOL> } <EOL> } <EOL> } ] <EOL> chart_runner ( map_df , scales , axes , marks ) <EOL> for i , data in enumerate ( datas ) : <EOL> nt . assert_dict_equal ( map_df . data [ i ] . grammar ( ) , data ) <EOL> map_df . rebind ( column = '<STR_LIT>' , brew = '<STR_LIT>' ) <EOL> rebound = { '<STR_LIT:name>' : '<STR_LIT>' , <EOL> '<STR_LIT>' : [ <EOL> { '<STR_LIT:x>' : <NUM_LIT:1> , '<STR_LIT:y>' : <NUM_LIT:7> } , <EOL> { '<STR_LIT:x>' : <NUM_LIT:2> , '<STR_LIT:y>' : <NUM_LIT:8> } , <EOL> { '<STR_LIT:x>' : <NUM_LIT:3> , '<STR_LIT:y>' : <NUM_LIT:9> } <EOL> ] <EOL> } <EOL> new_scale = { <EOL> '<STR_LIT:name>' : '<STR_LIT>' , <EOL> '<STR_LIT:type>' : '<STR_LIT>' , <EOL> '<STR_LIT>' : [ <NUM_LIT> , <NUM_LIT> ] , <EOL> '<STR_LIT>' : [ <EOL> "<STR_LIT>" , <EOL> "<STR_LIT>" , <EOL> "<STR_LIT>" , <EOL> "<STR_LIT>" , <EOL> "<STR_LIT>" , <EOL> "<STR_LIT>" , <EOL> "<STR_LIT>" , <EOL> "<STR_LIT>" , <EOL> "<STR_LIT>" <EOL> ] <EOL> } <EOL> assert map_df . data [ '<STR_LIT>' ] . grammar ( ) == rebound <EOL> assert map_df . scales [ '<STR_LIT>' ] . grammar ( ) == new_scale <EOL> class TestWord ( object ) : <EOL> """<STR_LIT>""" <EOL> def test_init ( self ) : <EOL> d = { '<STR_LIT>' : <NUM_LIT:32> , '<STR_LIT>' : <NUM_LIT> , '<STR_LIT>' : <NUM_LIT> , '<STR_LIT>' : <NUM_LIT:12> , <EOL> '<STR_LIT>' : <NUM_LIT> , '<STR_LIT>' : <NUM_LIT> , '<STR_LIT>' : <NUM_LIT:20> , '<STR_LIT>' : <NUM_LIT> , <EOL> '<STR_LIT>' : <NUM_LIT:10> , '<STR_LIT>' : <NUM_LIT> , '<STR_LIT>' : <NUM_LIT> , '<STR_LIT>' : <NUM_LIT:10> } <EOL> word = Word ( d ) <EOL> axes = [ ] <EOL> scales = [ { '<STR_LIT>' : { '<STR_LIT:data>' : '<STR_LIT>' , '<STR_LIT>' : '<STR_LIT>' } , <EOL> '<STR_LIT:name>' : '<STR_LIT>' , <EOL> '<STR_LIT>' : '<STR_LIT>' , <EOL> '<STR_LIT:type>' : '<STR_LIT>' } ] <EOL> marks = [ { <EOL> '<STR_LIT:type>' : '<STR_LIT:text>' , <EOL> '<STR_LIT>' : { '<STR_LIT:data>' : '<STR_LIT>' } , <EOL> '<STR_LIT>' : { <EOL> '<STR_LIT>' : { <EOL> '<STR_LIT>' : { '<STR_LIT:value>' : '<STR_LIT>' } , <EOL> '<STR_LIT>' : { '<STR_LIT>' : '<STR_LIT>' } , <EOL> '<STR_LIT>' : { '<STR_LIT:value>' : '<STR_LIT>' } , <EOL> '<STR_LIT>' : { '<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:text>' : { '<STR_LIT>' : '<STR_LIT>' } , <EOL> '<STR_LIT:x>' : { '<STR_LIT>' : '<STR_LIT:x>' } , <EOL> '<STR_LIT:y>' : { '<STR_LIT>' : '<STR_LIT:y>' } <EOL> } <EOL> } <EOL> } ] <EOL> chart_runner ( word , scales , axes , marks ) </s>
