Feat: use fonction to build graphs

This commit is contained in:
2025-01-06 07:12:08 +01:00
parent 0ae6439217
commit 90472ac868
3 changed files with 27 additions and 14 deletions

View File

@@ -128,7 +128,7 @@ def test_listing_content(dataplatform: DataPlateform):
def test_content_from_graphset(dataplatform: DataPlateform):
assert dataplatform.graphset.node_sets == {
assert dataplatform.graphset().node_sets == {
frozenset(
{
Node(name="test-bronze-username"),
@@ -151,7 +151,7 @@ def test_content_from_graphset(dataplatform: DataPlateform):
}
),
}
assert dataplatform.graphset.edges == [
assert dataplatform.graphset().edges == [
EdgeOnSet(
arrow="foo_flux",
sources=[Node(name="test-raw-username"), Node(name="test-raw-recovery")],
@@ -168,7 +168,7 @@ def test_content_from_graphset(dataplatform: DataPlateform):
def test_content_from_graph(dataplatform: DataPlateform):
assert dataplatform.graph.nodes == {
assert dataplatform.graph().nodes == {
Node(name="test-raw-recovery", infos={}),
Node(name="test-raw-salary", infos={}),
Node(name="test-raw-username", infos={}),
@@ -176,7 +176,7 @@ def test_content_from_graph(dataplatform: DataPlateform):
Node(name="test-bronze-foo", infos={}),
Node(name="test-raw-username", infos={}),
}
assert dataplatform.graph.edges == [
assert dataplatform.graph().edges == [
Edge(
arrow="foo_flux",
source=Node(name="test-raw-username"),