temp_fp = tempfile.NamedTemporaryFile(delete=False)

temp_fp.write(contents)

temp_fp.close()

@classmethod

def setUpClass(cls):

cls.data = {}

file_contents = dedent('''

"X Values"

1

2

3

4

5

6

7

8

9

''')

cls.data['int'] = create_temp_csv(file_contents)

file_contents = dedent('''

"X Values","Y Values"

1,1

2,4

3,9

4,16

5,25

6,36

7,49

8,64

9,81

''')

cls.data['int-int'] = create_temp_csv(file_contents)

file_contents = dedent('''

"X Values","Y Values"

8,64

9,81

1,1

3,9

5,25

2,4

4,16

6,36

7,49

''')

cls.data['int-int-unordered'] = create_temp_csv(file_contents)

file_contents = dedent('''

"X Values","Y Values"

2011-01-01,1

2011-01-02,4

2011-01-03,9

2011-01-04,16

2011-01-05,25

2011-01-06,36

2011-01-07,49

2011-01-08,64

2011-01-09,81

''')

cls.data['date-int'] = create_temp_csv(file_contents)

file_contents = dedent('''

"X Values","Y Values","Z Values"

1,1,2011-01-01

4,8,2011-01-02

9,27,2011-01-03

16,64,2011-01-04

25,125,2011-01-05

36,216,2011-01-06

49,343,2011-01-07

64,512,2011-01-08

81,729,2011-01-09

''')

cls.data['int-int-date'] = create_temp_csv(file_contents)

file_contents = dedent('''

"X Values","Y Values","Z Values"

1,1,1

4,8,12

9,27,36

16,64,80

25,125,150

36,216,252

49,343,392

64,512,576

81,729,810

''')

cls.data['int-int-int'] = create_temp_csv(file_contents)

file_contents = dedent('''

animal

dog

cat

dog

dog

dog

cat

cat

human

dog

tiger

tiger

coati

''')

cls.data['animals'] = create_temp_csv(file_contents)

file_contents = dedent('''

product_name,quantity,year

E,235,2004

D,154,2001

A,200,2000

B,300,2000

C,345,2001

D,156,2004

E,233,2001

C,250,2000

D,245,2000

E,190,2000

B,286,2001

D,254,2003

A,249,2004

B,288,2004

C,347,2004

D,144,2002

E,256,2002

A,339,2003

B,386,2003

C,243,2003

C,195,2005

D,231,2005

E,234,2005

''')

cls.data['bar-data'] = create_temp_csv(file_contents)

@classmethod

def tearDownClass(cls):

for filename in cls.data.values():

os.remove(filename)

def test_01_2_columns_with_only_data_no_scatter(self):

image_filename = get_filename_from_frame(inspect.currentframe())

my_plot = Plotter(self.data['int-int'])

my_plot.linear(title='Hello, world')

my_plot.save(image_filename)

def test_02_2_columns_with_only_data_scatter(self):

image_filename = get_filename_from_frame(inspect.currentframe())

my_plot = Plotter(self.data['int-int'])

my_plot.scatter(x_column='X Values')

my_plot.save(image_filename)

def test_02_2_columns_with_unordered_data_scatter(self):

image_filename = get_filename_from_frame(inspect.currentframe())

my_plot = Plotter(self.data['int-int-unordered'])

my_plot.scatter(x_column='X Values')

my_plot.save(image_filename)

def test_03_2_columns_with_one_as_date_no_scatter(self):

image_filename = get_filename_from_frame(inspect.currentframe())

my_plot = Plotter(self.data['date-int'])

my_plot.linear()

my_plot.save(image_filename)

def test_04_2_columns_with_one_as_date_scatter(self):

image_filename = get_filename_from_frame(inspect.currentframe())

my_plot = Plotter(self.data['date-int'])

my_plot.scatter(x_column='X Values')

my_plot.save(image_filename)

def test_05_3_columns_with_one_as_date_scatter_no_labels(self):

image_filename = get_filename_from_frame(inspect.currentframe())

my_plot = Plotter(self.data['date-int'])

my_plot.scatter(x_column='X Values', labels=False)

my_plot.save(image_filename)

def test_06_3_columns_with_one_as_date_and_x_as_int_scatter(self):

image_filename = get_filename_from_frame(inspect.currentframe())

my_plot = Plotter(self.data['int-int-date'])

my_plot.scatter(x_column='Y Values')

my_plot.save(image_filename)

def test_06_scatter_with_x_column_as_string(self):

image_filename = get_filename_from_frame(inspect.currentframe())

my_plot = Plotter(self.data['int-int-date'])

my_plot.scatter(x_column='Y Values')

my_plot.save(image_filename)

def test_07_1_column_barplot(self):

image_filename = get_filename_from_frame(inspect.currentframe())

my_plot = Plotter(self.data['int'])

my_plot.bar()

my_plot.save(image_filename)

def test_08_2_columns_barplot(self):

image_filename = get_filename_from_frame(inspect.currentframe())

my_plot = Plotter(self.data['int-int'])

my_plot.bar()

my_plot.save(image_filename)

def test_08_2_columns_barplot_with_x_rotation(self):

image_filename = get_filename_from_frame(inspect.currentframe())

my_plot = Plotter(self.data['int-int'])

my_plot.bar(x_rotation=45)

my_plot.save(image_filename)

def test_09_bar_plot_and_1_column_with_count(self):

image_filename = get_filename_from_frame(inspect.currentframe())

my_plot = Plotter(self.data['animals'])

my_plot.bar(count='animal', legends={'animal': 'Animals'})

my_plot.save(image_filename)

def test_10_2_columns_plot_with_one_as_label(self):

image_filename = get_filename_from_frame(inspect.currentframe())

my_plot = Plotter(self.data['int-int'])

my_plot.linear(x_labels='Y Values')

my_plot.save(image_filename)

def test_11_ignore_columns_linear(self):

image_filename = get_filename_from_frame(inspect.currentframe())

my_plot = Plotter(self.data['int-int-int'])

my_plot.linear(x_labels='Y Values', ignore=['X Values', 'Y Values'])

my_plot.save(image_filename)

def test_11_ignore_columns_scatter(self):

image_filename = get_filename_from_frame(inspect.currentframe())

my_plot = Plotter(self.data['int-int-int'])

my_plot.scatter(x_column='Y Values', ignore=['X Values'])

my_plot.save(image_filename)

def test_12_plot_stacked_bars(self):

image_filename = get_filename_from_frame(inspect.currentframe())

my_plot = Plotter(self.data['bar-data'])

my_plot.stacked_bar(x_column='year', y_column='quantity',

y_labels='product_name',

title='Stacked bars example')

my_plot.save(image_filename)

def test_13_use_colors_instead_of_colormap_linear(self):

image_filename = get_filename_from_frame(inspect.currentframe())

my_plot = Plotter(self.data['int-int-int'])

my_plot.linear(x_labels='Y Values', colors=['red', 'blue'])

my_plot.save(image_filename)

def test_13_use_colors_instead_of_colormap_scatter(self):

image_filename = get_filename_from_frame(inspect.currentframe())

my_plot = Plotter(self.data['int-int-int'])

my_plot.scatter(x_column='Y Values', colors=['red', 'blue'])

my_plot.save(image_filename)

def test_13_use_colors_instead_of_colormap_bar(self):

image_filename = get_filename_from_frame(inspect.currentframe())

my_plot = Plotter(self.data['int-int-int'])

my_plot.bar(colors=['red', 'blue', 'yellow'])

my_plot.save(image_filename)

def test_13_use_colors_instead_of_colormap_stacked_bar(self):

image_filename = get_filename_from_frame(inspect.currentframe())

my_plot = Plotter(self.data['bar-data'])

my_plot.stacked_bar(x_column='year', y_column='quantity',

y_labels='product_name',

title='Stacked bars example',

colors=['red', 'blue', 'yellow', 'magenta', 'gray'])

my_plot.save(image_filename)

def test_15_a_figure_with_a_lot_of_subplots(self):

image_filename = get_filename_from_frame(inspect.currentframe())

my_plot = Plotter(self.data['bar-data'], rows=2, cols=2)

my_plot.linear(ignore=['product_name', 'year'], title='Linear plot')

my_plot.scatter(x_column='quantity', title='Scatter plot')

my_plot.stacked_bar(x_column='year', y_column='quantity',

y_labels='product_name', title='Stacked bars plot',

x_rotation=45)

my_plot.bar(count='year', x_rotation=45, title='Bar plot')

my_plot.save(image_filename)

def test_should_raises_OverflowError_when_exceed_number_of_subplots(self):

my_plot = Plotter(self.data['bar-data'], rows=2, cols=1)

my_plot.linear(ignore=['product_name', 'year'])

my_plot.linear(ignore=['product_name', 'year'])

with self.assertRaises(OverflowError):

my_plot.linear(ignore=['product_name', 'year'])

def test_16_radar_plot(self):

image_filename = get_filename_from_frame(inspect.currentframe())

my_plot = Plotter(self.data['bar-data'])

my_plot.radar(axis_labels='product_name', values='quantity',

legends='year', title='Products by year')