def test_data_parameter_missing(self):
with self.assertRaises(TypeError):
plot.linesGraph(y_label='Y',
title='T',
lines_color=['r'],
alpha_values=[0.5],
areas=None,
x_label='X')
def test_data_parameter_index_wrong_type(self):
df = pd.DataFrame(index=range(3), columns=['A1'], data=[1, 2, 3])
with self.assertRaises(TypeError):
plot.linesGraph(data=df,
y_label='Y',
title='T',
lines_color=['r'],
alpha_values=[0.5],
areas=None,
x_label='X')
def test_areas_parameter_missing(self):
df = pd.DataFrame(index=pd.DatetimeIndex(
['2018-01-01', '2018-01-02', '2018-01-03']),
columns=['A1'],
data=[1, 2, 3])
with self.assertRaises(TypeError):
plot.linesGraph(data=df,
y_label='Y',
title='T',
lines_color=['r'],
alpha_values=[0.5],
x_label='X')
def test_x_label_parameter_missing(self):
df = pd.DataFrame(index=pd.DatetimeIndex(
['2018-01-01', '2018-01-02', '2018-01-03']),
columns=['A1'],
data=[1, 2, 3])
self.assertIsNotNone(
plot.linesGraph(data=df,
y_label='Y',
title='T',
lines_color=['r'],
alpha_values=[0.5],
areas=None))
plt.close('all')
def test_input_parameter_x_label(self):
df = pd.DataFrame(index=pd.DatetimeIndex(
['2018-01-01', '2018-01-02', '2018-01-03']),
columns=['A1'],
data=[1, 2, 3])
self.assertEqual(
plot.linesGraph(data=df,
y_label='Y',
title='T',
lines_color=['r'],
alpha_values=[0.5],
areas=None,
x_label='X').gca().get_xlabel(), 'X')
plt.close('all')
def test_data_parameter_three_dataframes(self):
df = pd.DataFrame(index=pd.DatetimeIndex(
['2018-01-01', '2018-01-02', '2018-01-03']),
columns=['A1', 'A2'],
data=[[1, 1], [2, 2], [3, 3]])
self.assertIsNotNone(
plot.linesGraph(data=[df, df, df],
y_label='Y',
title='T',
lines_color=['r', 'g', 'b'],
alpha_values=[0.5, 0.5, 0.5],
areas=None,
x_label='X'))
plt.close('all')
def test_input_parameter_areas(self):
df = pd.DataFrame(index=pd.DatetimeIndex(
['2018-01-01', '2018-01-02', '2018-01-03']),
columns=['A1', 'A2'],
data=[[1, 1], [2, 2], [3, 3]])
self.assertIsNotNone(
plot.linesGraph(data=[df, df],
y_label='Y',
title='T',
lines_color=['red', 'green', 'blue', 'black'],
alpha_values=[0.5, 0.5, 0.5, 0.5],
areas=[{
'x': 'ti_index',
'y1': [0, 'ti_data', 'A1'],
'y2': [1, 'ti_data', 'A2'],
'color': 'lightblue'
}],
x_label='X'))
plt.close('all')
