You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
77 lines
1.8 KiB
77 lines
1.8 KiB
import pytest |
|
|
|
|
|
@pytest.fixture |
|
def example_file(example_root): |
|
return example_root / "Verbrauchsmittel-Toto-2020.xlsx" |
|
|
|
|
|
@pytest.fixture |
|
def example_workbook(example_file): |
|
import openpyxl |
|
|
|
yield openpyxl.open(example_file) |
|
|
|
|
|
@pytest.fixture |
|
def example_sheet(example_workbook): |
|
sheets = example_workbook.sheetnames |
|
first = sheets[0] |
|
yield example_workbook[first] |
|
|
|
|
|
def test_check_table_header_raises_error(): |
|
from superx_budget.budget import _check_table_header |
|
from superx_budget.exceptions import BudgetParserError |
|
|
|
row = ["not", "the", "expected", "row"] |
|
|
|
with pytest.raises(BudgetParserError): |
|
_check_table_header(row) |
|
|
|
|
|
def test_skip_empty_lines(): |
|
from superx_budget.budget import _skip_empty_lines |
|
|
|
rows = [[""], ["one"], [None], [""], ["two"], [""]] |
|
|
|
result = _skip_empty_lines(rows) |
|
|
|
assert list(result) == [(1, ["one"]), (4, ["two"])] |
|
|
|
|
|
def test_parse_data_table(): |
|
from superx_budget.budget import _parse_data_table |
|
|
|
rows = [ |
|
list("ABCDEFG"), |
|
[None for i in range(7)], |
|
list("tuvwxyzX"), # one item more |
|
] |
|
|
|
result = _parse_data_table(rows) |
|
first, last = list(result) |
|
|
|
assert first.row == 2 |
|
assert first.number == "A" |
|
assert first.rest == "G" |
|
assert last.row == 4 |
|
assert last.number == "t" |
|
assert last.rest == "z" |
|
|
|
|
|
def test_parse_budget_data(example_sheet): |
|
from superx_budget.budget import parse_budget_data |
|
|
|
result = parse_budget_data(example_sheet) |
|
first, last = result[0], result[-1] |
|
|
|
assert len(result) == 18 |
|
assert first.row == 3 |
|
assert first.number == 1 |
|
assert first.project_name == "Safegurard I (neu)" |
|
assert first.rest == "=E3-F3" |
|
assert last.row == 54 |
|
assert last.number == "=A51+1" |
|
assert last.project_name == "ZIM Microcoat II" |
|
assert last.rest == "=E54-F54"
|
|
|