Philipp/aki_prj23_transparenzregister
Philipp/aki_prj23_transparenzregister
1
0
Fork 0
mirror of https://github.com/fhswf/aki_prj23_transparenzregister.git synced 2025-05-12 16:30:06 +02:00
Code Issues Packages Projects Releases Wiki Activity

refactor: Adapt to extended unit tests

Browse Source
This commit is contained in:
TrisNol 2023-10-15 13:21:41 +02:00
parent fd47487367
commit f1474feaf8
2 changed files with 12 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
src/aki_prj23_transparenzregister/utils/data_extraction/unternehmensregister/transform.py
View File

@ -505,16 +505,16 @@ def map_co_relation(data: dict) -> dict:
trimmed_part = part.strip()
result = re.findall(r"^c\/o(.*)$", trimmed_part)
if len(result) == 1:
co_company = result[0]
co_company = result[0].strip()
co_company_index = index
if co_company_index is not None:
del parts[co_company_index]
street = "".join(parts).strip()
data["location"].street = street
if co_company is not None:
if co_company is not None and co_company != "":
relation = CompanyToCompanyRelationship(
RelationshipRoleEnum.LOKATION_BEI, # type: ignore
RelationshipRoleEnum.CARE_OF, # type: ignore
Location(
data["location"].city,
street,

18
tests/utils/data_extraction/unternehmensregister/transform_test.py
View File

@ -723,7 +723,7 @@ def test_map_last_update() -> None:
"location": Location("", "Abc ffda", None, None),
"relationships": [
CompanyToCompanyRelationship(
RelationshipRoleEnum.LOKATION_BEI, # type: ignore
RelationshipRoleEnum.CARE_OF, # type: ignore
Location("", "Abc ffda", None, None),
CompanyRelationshipEnum.COMPANY,
"Youco24 Business Center",
@ -734,7 +734,7 @@ def test_map_last_update() -> None:
(
{
"location": Location(
"", "c/o Youco24 Business Center, Abc Str.", "42", "58644"
"Iserlohn", "c/o Youco24 Business Center, Abc Str.", "42", "58644"
),
"relationships": [],
},
@ -742,8 +742,8 @@ def test_map_last_update() -> None:
"location": Location("Iserlohn", "Abc Str.", "42", "58644"),
"relationships": [
CompanyToCompanyRelationship(
RelationshipRoleEnum.LOKATION_BEI, # type: ignore
Location("Iserlohn", "Abc ffda", "42", "58644"),
RelationshipRoleEnum.CARE_OF, # type: ignore
Location("Iserlohn", "Abc Str.", "42", "58644"),
CompanyRelationshipEnum.COMPANY,
"Youco24 Business Center",
)
@ -753,7 +753,7 @@ def test_map_last_update() -> None:
(
{
"location": Location(
"Iserlohn", "Abc Str. 42, c/o Youco24 Business Center", None, "58644"
"Iserlohn", "Abc Str., c/o Youco24 Business Center", "42", "58644"
),
"relationships": [],
},
@ -761,8 +761,8 @@ def test_map_last_update() -> None:
"location": Location("Iserlohn", "Abc Str.", "42", "58644"),
"relationships": [
CompanyToCompanyRelationship(
RelationshipRoleEnum.LOKATION_BEI, # type: ignore
Location("Iserlohn", "Abc ffda", "42", "58644"),
RelationshipRoleEnum.CARE_OF, # type: ignore
Location("Iserlohn", "Abc Str.", "42", "58644"),
CompanyRelationshipEnum.COMPANY,
"Youco24 Business Center",
)
@ -771,9 +771,7 @@ def test_map_last_update() -> None:
),
(
{
"location": Location(
"Iserlohn", "Abc Str. 42, c/o", None, "58644"
),
"location": Location("Iserlohn", "Abc Str., c/o", "42", "58644"),
"relationships": [],
},
{