Python read_table_with_template Examples

Example #1

File: test_extract_table.py Project: mfagundes/extract_table_from_pdf

def test_format_demanda_por_mercado_isolados():
    name = 'demanda-malha-interligada'
    settings_ = settings.TABELAS.get(name)
    kwargs = settings_['args']
    df = read_table_with_template(FILE, TEMPLATES_FILE, settings_['titulo'])
    balanco = format_demanda_por_mercado(df, **kwargs)
    assert balanco.shape == (8,17)

Example #2

File: test_extract_table.py Project: mfagundes/extract_table_from_pdf

def test_format_balanco_malha_isolada():
    name = 'balanco-sistemas-isolados'
    settings_ = settings.TABELAS.get(name)
    kwargs = settings_['args']
    df = read_table_with_template(FILE, TEMPLATES_FILE, settings_['titulo'])
    balanco = format_balanco_gas_geral(df, **kwargs)
    assert balanco.shape == (6,17)

Example #3

File: test_extract_table.py Project: mfagundes/extract_table_from_pdf

def test_format_balanco_gas_natural():
    name = 'balanco-brasil'
    settings_ = settings.TABELAS.get(name)
    kwargs = settings_['args']
    df = read_table_with_template(FILE, TEMPLATES_FILE, settings_['titulo'])
    balanco = format_balanco_gas_geral(df, **kwargs)
    assert balanco.shape == (13,18)

Example #4

File: test_extract_table.py Project: mfagundes/extract_table_from_pdf

def test_format_balanco_malha_interligada():
    name = 'balanco-malha-interligada'
    settings_ = settings.TABELAS.get(name)
    df = read_table_with_template(FILE, TEMPLATES_FILE, settings_['titulo'])
    kwargs = settings_['args']
    balanco = format_balanco_gas_geral(df, **kwargs)
    assert balanco.shape == (12,17)

Example #5

File: test_extract_table.py Project: mfagundes/extract_table_from_pdf

def test_read_table_with_templates(file, template_file, template_name):
    table = read_table_with_template(file, template_file, template_name)
    assert isinstance(table, pd.core.frame.DataFrame)