<s> from __future__ import ( absolute_import , division , print_function , <EOL> unicode_literals ) <EOL> from . import get_assert_same_ggplot , cleanup , assert_same_elements <EOL> assert_same_ggplot = get_assert_same_ggplot ( __file__ ) <EOL> from nose . tools import ( assert_true , assert_raises , assert_is , <EOL> assert_is_not , assert_equal ) <EOL> from ggplot import * <EOL> import six <EOL> import pandas as pd <EOL> from ggplot . components import assign_visual_mapping <EOL> from ggplot . utils . exceptions import GgplotError <EOL> def get_test_df ( ) : <EOL> df = pd . DataFrame ( { <EOL> '<STR_LIT>' : [ <NUM_LIT:1> , <NUM_LIT:3> , <NUM_LIT:5> ] , <EOL> '<STR_LIT>' : [ <NUM_LIT:2> , <NUM_LIT> , <NUM_LIT:7> ] , <EOL> '<STR_LIT>' : [ <NUM_LIT:1> , <NUM_LIT:4> , <NUM_LIT:6> ] , <EOL> '<STR_LIT>' : [ <NUM_LIT:5> , <NUM_LIT:5> , <NUM_LIT:9> ] , <EOL> '<STR_LIT>' : [ '<STR_LIT>' , '<STR_LIT>' , '<STR_LIT>' ] , <EOL> '<STR_LIT>' : [ '<STR_LIT>' , '<STR_LIT>' , '<STR_LIT>' ] , <EOL> '<STR_LIT>' : [ <NUM_LIT:0.1> , <NUM_LIT:0.5> , <NUM_LIT> ] , <EOL> '<STR_LIT>' : [ '<STR_LIT>' , '<STR_LIT>' , '<STR_LIT>' ] } ) <EOL> return df <EOL> def test_legend_structure ( ) : <EOL> df = get_test_df ( ) <EOL> gg = ggplot ( df , aes ( xmin = '<STR_LIT>' , xmax = '<STR_LIT>' , ymin = '<STR_LIT>' , ymax = '<STR_LIT>' , <EOL> colour = '<STR_LIT>' , fill = '<STR_LIT>' , alpha = '<STR_LIT>' , <EOL> linetype = '<STR_LIT>' ) ) <EOL> new_df , legend = assign_visual_mapping ( df , gg . aesthetics , gg ) <EOL> for aesthetic in ( '<STR_LIT>' , '<STR_LIT>' , '<STR_LIT>' , '<STR_LIT>' ) : <EOL> assert ( aesthetic in legend ) <EOL> assert ( '<STR_LIT:size>' not in legend ) <EOL> assert ( '<STR_LIT>' not in legend ) <EOL> assert ( legend [ '<STR_LIT>' ] [ '<STR_LIT>' ] == '<STR_LIT>' ) <EOL> assert ( legend [ '<STR_LIT>' ] [ '<STR_LIT>' ] == '<STR_LIT>' ) <EOL> assert ( legend [ '<STR_LIT>' ] [ '<STR_LIT>' ] == '<STR_LIT>' ) <EOL> assert ( legend [ '<STR_LIT>' ] [ '<STR_LIT>' ] == '<STR_LIT>' ) <EOL> assert ( legend [ '<STR_LIT>' ] [ '<STR_LIT>' ] == '<STR_LIT>' ) <EOL> assert ( legend [ '<STR_LIT>' ] [ '<STR_LIT>' ] == '<STR_LIT>' ) <EOL> assert ( legend [ '<STR_LIT>' ] [ '<STR_LIT>' ] == '<STR_LIT>' ) <EOL> assert ( legend [ '<STR_LIT>' ] [ '<STR_LIT>' ] == '<STR_LIT>' ) <EOL> df2 = pd . DataFrame . copy ( df ) <EOL> df2 [ '<STR_LIT>' ] = [ <NUM_LIT> , <NUM_LIT> , <NUM_LIT> ] <EOL> gg = ggplot ( df2 , aes ( xmin = '<STR_LIT>' , xmax = '<STR_LIT>' , ymin = '<STR_LIT>' , ymax = '<STR_LIT>' , <EOL> colour = '<STR_LIT>' , fill = '<STR_LIT>' , alpha = '<STR_LIT>' , <EOL> linetype = '<STR_LIT>' ) ) <EOL> new_df , legend = assign_visual_mapping ( df2 , gg . aesthetics , gg ) <EOL> assert ( legend [ '<STR_LIT>' ] [ '<STR_LIT>' ] == '<STR_LIT>' ) <EOL> df3 = pd . DataFrame ( { <EOL> '<STR_LIT>' : [ <NUM_LIT:1> , <NUM_LIT:3> , <NUM_LIT:5> , <NUM_LIT:8> , <NUM_LIT:2> , <NUM_LIT:1> , <NUM_LIT:4> , <NUM_LIT:7> , <NUM_LIT:9> ] , <EOL> '<STR_LIT>' : [ <NUM_LIT:2> , <NUM_LIT> , <NUM_LIT:7> , <NUM_LIT:12> , <NUM_LIT:3> , <NUM_LIT:2> , <NUM_LIT:6> , <NUM_LIT:8> , <NUM_LIT:10> ] , <EOL> '<STR_LIT>' : [ <NUM_LIT:1> , <NUM_LIT:4> , <NUM_LIT:6> , <NUM_LIT:0> , <NUM_LIT:0> , <NUM_LIT:0> , <NUM_LIT:0> , <NUM_LIT:0> , <NUM_LIT:0> ] , <EOL> '<STR_LIT>' : [ <NUM_LIT:5> , <NUM_LIT:5> , <NUM_LIT:9> , <NUM_LIT:1> , <NUM_LIT:1> , <NUM_LIT:1> , <NUM_LIT:1> , <NUM_LIT:1> , <NUM_LIT:1> ] , <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>' , '<STR_LIT>' , '<STR_LIT>' , '<STR_LIT>' , '<STR_LIT>' , <EOL> '<STR_LIT>' , '<STR_LIT>' , '<STR_LIT>' , '<STR_LIT>' ] , <EOL> '<STR_LIT>' : [ <NUM_LIT:0.1> , <NUM_LIT> , <NUM_LIT> , <NUM_LIT:0.5> , <NUM_LIT> , <NUM_LIT> , <NUM_LIT> , <NUM_LIT> , <NUM_LIT> ] , <EOL> '<STR_LIT>' : [ '<STR_LIT>' , '<STR_LIT>' , '<STR_LIT>' , '<STR_LIT>' , '<STR_LIT>' , '<STR_LIT>' , <EOL> '<STR_LIT>' , '<STR_LIT>' , '<STR_LIT>' ] <EOL> } ) <EOL> gg = ggplot ( df2 , aes ( xmin = '<STR_LIT>' , xmax = '<STR_LIT>' , ymin = '<STR_LIT>' , ymax = '<STR_LIT>' , <EOL> colour = '<STR_LIT>' , fill = '<STR_LIT>' , alpha = '<STR_LIT>' , <EOL> linetype = '<STR_LIT>' ) ) <EOL> new_df , legend = assign_visual_mapping ( df3 , gg . aesthetics , gg ) <EOL> assert ( legend [ '<STR_LIT>' ] [ '<STR_LIT>' ] == '<STR_LIT>' ) <EOL> gg = ggplot ( df3 , aes ( size = "<STR_LIT>" ) ) <EOL> assert_raises ( GgplotError , assign_visual_mapping , df3 , gg . aesthetics , gg ) <EOL> gg = ggplot ( df3 , aes ( alpha = "<STR_LIT>" ) ) <EOL> assert_raises ( GgplotError , assign_visual_mapping , df3 , gg . aesthetics , gg ) <EOL> @ cleanup <EOL> def test_alpha_rect ( ) : <EOL> df = get_test_df ( ) <EOL> p = ggplot ( df , aes ( xmin = '<STR_LIT>' , xmax = '<STR_LIT>' , ymin = '<STR_LIT>' , ymax = '<STR_LIT>' , <EOL> colour = '<STR_LIT>' , fill = '<STR_LIT>' , alpha = '<STR_LIT>' , <EOL> linetype = '<STR_LIT>' ) ) <EOL> p += geom_rect ( size = <NUM_LIT:5> ) <EOL> assert_same_ggplot ( p , "<STR_LIT>" ) <EOL> @ cleanup <EOL> def test_alpha ( ) : <EOL> diamonds [ "<STR_LIT:test>" ] = diamonds [ "<STR_LIT>" ] . map ( len ) <EOL> p = ggplot ( diamonds [ : : <NUM_LIT:50> ] , aes ( x = '<STR_LIT>' , y = '<STR_LIT>' , colour = '<STR_LIT:test>' , <EOL> size = '<STR_LIT:test>' , alpha = '<STR_LIT:test>' ) ) <EOL> p = p + geom_point ( ) + ggtitle ( "<STR_LIT>" ) <EOL> p = p + xlab ( "<STR_LIT>" ) + ylab ( "<STR_LIT>" ) <EOL> assert_same_ggplot ( p , "<STR_LIT>" ) <EOL> @ cleanup <EOL> def test_linetype ( ) : <EOL> meat_lng = pd . melt ( meat [ [ '<STR_LIT:date>' , '<STR_LIT>' , '<STR_LIT>' , '<STR_LIT>' ] ] , id_vars = '<STR_LIT:date>' ) <EOL> p = ggplot ( aes ( x = '<STR_LIT:date>' , y = '<STR_LIT:value>' , colour = '<STR_LIT>' , <EOL> linetype = '<STR_LIT>' , shape = '<STR_LIT>' ) , data = meat_lng ) + geom_line ( ) + geom_point ( ) + ylim ( <NUM_LIT:0> , <NUM_LIT> ) <EOL> assert_same_ggplot ( p , "<STR_LIT>" ) <EOL> @ cleanup <EOL> def test_shape_alpha ( ) : <EOL> diamonds [ "<STR_LIT:test>" ] = diamonds [ "<STR_LIT>" ] . map ( len ) <EOL> df = diamonds [ : : <NUM_LIT:50> ] <EOL> p = ggplot ( df , aes ( x = '<STR_LIT>' , y = '<STR_LIT>' , colour = '<STR_LIT:test>' , size = '<STR_LIT:test>' , <EOL> alpha = '<STR_LIT:test>' , shape = '<STR_LIT>' ) ) + geom_point ( ) <EOL> assert_same_ggplot ( p , "<STR_LIT>" ) </s>
