def test_get_fill_color(self):
''' Tests if an element from collor array is correctly fetched '''
options_from_yaml = {
'chart_options': {
'colorArray': ['red', 'yellow', 'green']
}
}
self.assertEqual(BaseChart.get_fill_color(1, options_from_yaml),
'yellow')
def test_legend_names_no_label_field(self):
''' Tests if legend names are built from series ID in the dataframe '''
options_from_yaml = {'chart_options': {'id': 'idx'}}
dataframe = pd.DataFrame([{'idx': 'A'}, {'idx': 'B'}])
self.assertEqual(
BaseChart.get_legend_names(dataframe, options_from_yaml), {
'A': 'A',
'B': 'B'
})
def test_legend_names(self):
''' Tests if legend names are built from series ID in the dataframe, with
a mirror legend name specified '''
options_from_yaml = {
'chart_options': {
'legend_field': 'lgnd',
'id': 'idx'
}
}
dataframe = pd.DataFrame([{
'idx': 'A',
'lgnd': 'A_lbl'
}, {
'idx': 'B',
'lgnd': 'B_lbl'
}])
self.assertEqual(
BaseChart.get_legend_names(dataframe, options_from_yaml), {
'A': 'A_lbl',
'B': 'B_lbl'
})
def test_tooltip(self):
''' Tests if tooltips are built correctly '''
options_from_yaml = {
'headers': [{
'text': 'Value A:',
'value': 'field_a'
}, {
'text': 'Value B:',
'value': 'field_b'
}]
}
self.assertEqual(
BaseChart.build_tooltip(options_from_yaml), '<table>'
'<tr style="text-align: left;">'
'<th style="padding: 4px; padding-right: 10px;">Value A:</th>'
'<td style="padding: 4px;">@field_a</td>'
'</tr>'
'<tr style="text-align: left;">'
'<th style="padding: 4px; padding-right: 10px;">Value B:</th>'
'<td style="padding: 4px;">@field_b</td>'
'</tr>'
'</table>')
def test_tooltip_no_headers(self):
''' Tests if default tooltip is returned when no headers are given '''
self.assertEqual(BaseChart.build_tooltip({}), 'Tooltip!')
def test_tooltip_no_option(self):
''' Tests if default tooltip is returned when no option is given '''
self.assertEqual(BaseChart.build_tooltip(None), 'Tooltip!')
def test_legend_names_no_id_field(self):
''' Tests if no legend names are returned when no ID field is given '''
options_from_yaml = {'chart_options': {'legend_field': 'lgnd'}}
dataframe = pd.DataFrame([{'idx': 'A'}, {'idx': 'B'}])
self.assertEqual(
BaseChart.get_legend_names(dataframe, options_from_yaml), {})
def test_legend_names_empty_dataframe(self):
''' Tests if no legend names are returned when the dataframe is empty '''
options_from_yaml = {'chart_options': {'id': 'idx'}}
self.assertEqual(
BaseChart.get_legend_names(pd.DataFrame([]), options_from_yaml),
{})
def test_legend_names_no_dataframe(self):
''' Tests if no legend names are returned when there's no dataframe '''
options_from_yaml = {'chart_options': {'id': 'idx'}}
self.assertEqual(BaseChart.get_legend_names(None, options_from_yaml),
{})
def test_legend_names_no_chart_options(self):
''' Tests if no legend names are returned when there's no chart_options '''
dataframe = pd.DataFrame([{'idx': 'A'}, {'idx': 'B'}])
self.assertEqual(BaseChart.get_legend_names(dataframe, {}), {})