def test_parse_columns(self):
with self.assertRaises(ValueError):
FanChart(
pd.DataFrame({
'index': line_mock_data['index'],
'name': line_mock_data['name'],
'aherha': [1]
}), pd.DataFrame(obs_mock_data))
def test_negative_logy_raises_warning(self):
with warnings.catch_warnings(record=True) as w:
warnings.simplefilter("always")
FanChart(
pd.DataFrame({
'name': ['name'],
'mean': [-1],
'min': [-1],
'max': [-1],
'p10': [-1],
'p90': [-1]
}, index=['02-03-2006']),
logy=True
)
assert(issubclass(w[-1].category, UserWarning))
def test_add_wrong_observations(self):
with self.assertRaises(ValueError):
FanChart(pd.DataFrame(line_mock_data),
pd.DataFrame({'wrong': ['i']}))
def test_add_empty_observations(self):
FanChart(pd.DataFrame(line_mock_data), pd.DataFrame())
def test_parse_without_observations(self):
self.assertTrue(FanChart(pd.DataFrame(line_mock_data)))
def test_parse_logy(self):
self.assertTrue(FanChart(pd.DataFrame(line_mock_data), logy=True))