wehub/microsoft--graphrag
1
0
Fork 0
Code Issues Pull Requests Actions Deployments Models Agent Notes Packages Projects Releases Wiki Activity

chore: import upstream snapshot with attribution
Python Build and Type Check / python-ci (ubuntu-latest, 3.11) (push) Has been cancelled
Details
Python Build and Type Check / python-ci (ubuntu-latest, 3.13) (push) Has been cancelled
Details
Python Build and Type Check / python-ci (windows-latest, 3.11) (push) Has been cancelled
Details
Python Build and Type Check / python-ci (windows-latest, 3.13) (push) Has been cancelled
Details
Python Integration Tests / python-ci (ubuntu-latest, 3.13) (push) Has been cancelled
Details
Python Integration Tests / python-ci (windows-latest, 3.13) (push) Has been cancelled
Details
Python Notebook Tests / python-ci (ubuntu-latest, 3.13) (push) Has been cancelled
Details
Python Notebook Tests / python-ci (windows-latest, 3.13) (push) Has been cancelled
Details
Python Smoke Tests / python-ci (ubuntu-latest, 3.13) (push) Has been cancelled
Details
Python Smoke Tests / python-ci (windows-latest, 3.13) (push) Has been cancelled
Details
Python Unit Tests / python-ci (ubuntu-latest, 3.13) (push) Has been cancelled
Details
Python Unit Tests / python-ci (windows-latest, 3.13) (push) Has been cancelled
Details
gh-pages / build (push) Has been cancelled
Details
Python Publish (pypi) / Upload release to PyPI (push) Has been cancelled
Details
Spellcheck / spellcheck (push) Has been cancelled
Details

Browse Source
This commit is contained in:
wehub-resource-sync
2026-07-13 12:37:31 +08:00
commit 6b7e6b44f1
897 changed files with 94808 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/unit/query/input/__init__.py
+2
View File
@@ -0,0 +1,2 @@
# Copyright (c) 2024 Microsoft Corporation.
# Licensed under the MIT License
tests/unit/query/input/retrieval/__init__.py
+2
View File
@@ -0,0 +1,2 @@
# Copyright (c) 2024 Microsoft Corporation.
# Licensed under the MIT License
tests/unit/query/input/retrieval/test_entities.py
+167
View File
@@ -0,0 +1,167 @@
# Copyright (c) 2024 Microsoft Corporation.
# Licensed under the MIT License
from graphrag.data_model.entity import Entity
from graphrag.query.input.retrieval.entities import (
get_entity_by_id,
get_entity_by_key,
)
def test_get_entity_by_id():
assert (
get_entity_by_id(
{
entity.id: entity
for entity in [
Entity(
id="2da37c7a-50a8-44d4-aa2c-fd401e19976c",
short_id="sid1",
title="title1",
),
]
},
"00000000-0000-0000-0000-000000000000",
)
is None
)
assert get_entity_by_id(
{
entity.id: entity
for entity in [
Entity(
id="2da37c7a-50a8-44d4-aa2c-fd401e19976c",
short_id="sid1",
title="title1",
),
Entity(
id="c4f93564-4507-4ee4-b102-98add401a965",
short_id="sid2",
title="title2",
),
Entity(
id="7c6f2bc9-47c9-4453-93a3-d2e174a02cd9",
short_id="sid3",
title="title3",
),
]
},
"7c6f2bc9-47c9-4453-93a3-d2e174a02cd9",
) == Entity(
id="7c6f2bc9-47c9-4453-93a3-d2e174a02cd9", short_id="sid3", title="title3"
)
assert get_entity_by_id(
{
entity.id: entity
for entity in [
Entity(
id="2da37c7a50a844d4aa2cfd401e19976c",
short_id="sid1",
title="title1",
),
Entity(
id="c4f9356445074ee4b10298add401a965",
short_id="sid2",
title="title2",
),
Entity(
id="7c6f2bc947c9445393a3d2e174a02cd9",
short_id="sid3",
title="title3",
),
]
},
"7c6f2bc9-47c9-4453-93a3-d2e174a02cd9",
) == Entity(id="7c6f2bc947c9445393a3d2e174a02cd9", short_id="sid3", title="title3")
assert get_entity_by_id(
{
entity.id: entity
for entity in [
Entity(id="id1", short_id="sid1", title="title1"),
Entity(id="id2", short_id="sid2", title="title2"),
Entity(id="id3", short_id="sid3", title="title3"),
]
},
"id3",
) == Entity(id="id3", short_id="sid3", title="title3")
def test_get_entity_by_key():
assert (
get_entity_by_key(
[
Entity(
id="2da37c7a-50a8-44d4-aa2c-fd401e19976c",
short_id="sid1",
title="title1",
),
],
"id",
"00000000-0000-0000-0000-000000000000",
)
is None
)
assert get_entity_by_key(
[
Entity(
id="2da37c7a-50a8-44d4-aa2c-fd401e19976c",
short_id="sid1",
title="title1",
),
Entity(
id="c4f93564-4507-4ee4-b102-98add401a965",
short_id="sid2",
title="title2",
),
Entity(
id="7c6f2bc9-47c9-4453-93a3-d2e174a02cd9",
short_id="sid3",
title="title3",
),
],
"id",
"7c6f2bc9-47c9-4453-93a3-d2e174a02cd9",
) == Entity(
id="7c6f2bc9-47c9-4453-93a3-d2e174a02cd9", short_id="sid3", title="title3"
)
assert get_entity_by_key(
[
Entity(
id="2da37c7a50a844d4aa2cfd401e19976c", short_id="sid1", title="title1"
),
Entity(
id="c4f9356445074ee4b10298add401a965", short_id="sid2", title="title2"
),
Entity(
id="7c6f2bc947c9445393a3d2e174a02cd9", short_id="sid3", title="title3"
),
],
"id",
"7c6f2bc9-47c9-4453-93a3-d2e174a02cd9",
) == Entity(id="7c6f2bc947c9445393a3d2e174a02cd9", short_id="sid3", title="title3")
assert get_entity_by_key(
[
Entity(id="id1", short_id="sid1", title="title1"),
Entity(id="id2", short_id="sid2", title="title2"),
Entity(id="id3", short_id="sid3", title="title3"),
],
"id",
"id3",
) == Entity(id="id3", short_id="sid3", title="title3")
assert get_entity_by_key(
[
Entity(id="id1", short_id="sid1", title="title1", rank=1),
Entity(id="id2", short_id="sid2", title="title2a", rank=2),
Entity(id="id3", short_id="sid3", title="title3", rank=3),
Entity(id="id2", short_id="sid2", title="title2b", rank=2),
],
"rank",
2,
) == Entity(id="id2", short_id="sid2", title="title2a", rank=2)