def test_deve_retornar_subplot_contendo_dados_corretos():
arquivo = tempfile.NamedTemporaryFile(delete=False)
arquivo.close()
dados = {'SP': 300, 'RJ': 200}
resultado = plota_grafico(dados, arquivo.name)
os.remove(arquivo.name)
x_tick_labels = resultado.get_xticklabels()
ufs = [x_tick_labels[0].get_text(), x_tick_labels[1].get_text()]
ufs.sort()
assert_equals(ufs, ['RJ', 'SP'])
y_values = [valor for valor in resultado.get_lines()[0].get_ydata()]
y_values.sort()
assert_equals(y_values, [200, 300])
#!/usr/bin/env python
# coding: utf-8
from resolucao import le_csv, agrega_valores, plota_grafico
registros = le_csv('exemplos/processos.csv')
valores_agregados = agrega_valores(registros)
plota_grafico(valores_agregados, 'processos-stf.png')
#!/usr/bin/env python
# coding: utf-8
from resolucao import le_csv, agrega_valores, plota_grafico
registros = le_csv('exemplos/processos.csv')
valores_agregados = agrega_valores(registros)
plota_grafico(valores_agregados, 'processos-stf.png')
