gmakstutis/cognee
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Fix poetry issues

Browse source
This commit is contained in:
vasilije 2025-01-08 17:28:58 +01:00
parent 70ced35557
commit 27e7808a2c
1 changed files with 6 additions and 14 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

20
cognee/tests/unit/processing/utils/utils_test.py
View file

@ -118,21 +118,13 @@ async def test_create_cognee_style_network_with_logo():
# Convert the graph to a tuple format for serialization
graph_tuple = graph_to_tuple(graph)
original_open = open
result = await create_cognee_style_network_with_logo(
graph_tuple,
title="Test Network",
layout_func=nx.spring_layout,
layout_scale=3.0,
logo_alpha=0.5,)
def mock_open_read_side_effect(*args, **kwargs):
if "cognee-logo.png" in args[0]:
return BytesIO(b"mock_png_data")
return original_open(*args, **kwargs)
with patch("builtins.open", side_effect=mock_open_read_side_effect):
result = await create_cognee_style_network_with_logo(
graph_tuple,
title="Test Network",
layout_func=nx.spring_layout,
layout_scale=3.0,
logo_alpha=0.5,
)
assert result is not None
assert isinstance(result, str)