def test_chart_states_returns_list():
"""Test the chart state function returns list."""
jd = JobData('test')
jd.count_states()
assert isinstance(jd.chart_states(), list)
def test_chart_states_list_lengths():
"""Test chart state function lists are same length."""
jd = JobData('test')
jd.count_states()
one, two = jd.chart_states()[0], jd.chart_states()[1]
assert len(one) == len(two)
def test_chart_states_raises_error():
"""Test chart state raises error if state dict not available."""
jd = JobData('test')
with pytest.raises(ValueError):
jd.chart_states()