From 3a5209c4115ef7cff56ad541662a91236536e77f Mon Sep 17 00:00:00 2001 From: Philipp Horstenkamp Date: Thu, 11 Jan 2024 00:35:23 +0100 Subject: [PATCH] Reworked the tests. --- .../utils/networkx/networkx_data.py | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/src/aki_prj23_transparenzregister/utils/networkx/networkx_data.py b/src/aki_prj23_transparenzregister/utils/networkx/networkx_data.py index e1f9690..4bd089f 100644 --- a/src/aki_prj23_transparenzregister/utils/networkx/networkx_data.py +++ b/src/aki_prj23_transparenzregister/utils/networkx/networkx_data.py @@ -27,8 +27,7 @@ COLOR_PERSON = "#ff5200" @cached( # type: ignore - cache=TTLCache(maxsize=100, ttl=int(os.getenv("PYTHON_CACHE_TTL", "3600"))), - key=lambda session, company_id: hash((company_id, str(session.bind))), + cache=TTLCache(maxsize=100, ttl=int(os.getenv("PYTHON_CACHE_TTL", "3600"))) ) def find_all_company_relations() -> pd.DataFrame: """_summary_. @@ -110,7 +109,7 @@ def get_all_company_relations() -> pd.DataFrame: @cached( # type: ignore cache=TTLCache(maxsize=100, ttl=int(os.getenv("PYTHON_CACHE_TTL", "3600"))), - key=lambda session, company_id: hash((company_id, str(session.bind))), + key=lambda drop_min_person_links: hash(drop_min_person_links), ) def get_all_person_relations(drop_min_person_links: bool) -> pd.DataFrame: """These method makes a Database Request for all Persons and their relations, modifies the ID Column and returns the Result as an DataFrame.