46 lines
1.3 KiB
Python
46 lines
1.3 KiB
Python
|
from pathlib import Path
|
||
|
|
|
||
|
|
from app.config import AppConfig
|
||
|
|
from app.pcf_parser import PcfParser, parse_blocks
|
||
|
|
|
||
|
|
|
||
|
|
def test_sample_pcf_weld_count():
|
||
|
|
config = AppConfig.load()
|
||
|
|
parser = PcfParser(config)
|
||
|
|
files = list(Path("PCF文件样本(1)").glob("*.pcf"))
|
||
|
|
records, candidates, messages = parser.parse_files(files)
|
||
|
|
|
||
|
|
assert len(files) == 20
|
||
|
|
assert len(records) == 169
|
||
|
|
assert not messages
|
||
|
|
assert records[0].line_no
|
||
|
|
assert records[0].pipeline_reference
|
||
|
|
assert records[0].weld_area in {"安装", "预制"}
|
||
|
|
|
||
|
|
|
||
|
|
def test_parse_nonstandard_field_candidate():
|
||
|
|
text = """
|
||
|
|
PIPELINE-REFERENCE L-001
|
||
|
|
PIPING-SPEC C10C
|
||
|
|
WELD
|
||
|
|
COMPONENT-IDENTIFIER 1
|
||
|
|
LINE-NO 01ABC
|
||
|
|
WELD-DIA 150
|
||
|
|
WELD-ATTRIBUTE3 Shop weld
|
||
|
|
""".strip()
|
||
|
|
blocks = parse_blocks(text)
|
||
|
|
assert blocks[0].block_type == "PIPELINE-REFERENCE"
|
||
|
|
assert blocks[1].block_type == "WELD"
|
||
|
|
|
||
|
|
tmp = Path("tests/_tmp_nonstandard.pcf")
|
||
|
|
tmp.write_text(text, encoding="utf-8")
|
||
|
|
try:
|
||
|
|
config = AppConfig.load()
|
||
|
|
parser = PcfParser(config)
|
||
|
|
_records, candidates, _messages = parser.parse_files([tmp])
|
||
|
|
fields = {c.source_field for c in candidates}
|
||
|
|
assert "LINE-NO" in fields
|
||
|
|
assert "WELD-DIA" in fields
|
||
|
|
finally:
|
||
|
|
tmp.unlink(missing_ok=True)
|