test: add test on graph for dataplatform

This commit is contained in:
2025-01-05 18:27:26 +01:00
parent 5ebde14be9
commit 2f170d91b6
4 changed files with 55 additions and 11 deletions

View File

@@ -4,7 +4,7 @@ from pathlib import Path
import pytest
from plesna.dataplatform import DataPlateform
from plesna.models.graphs import Node
from plesna.models.graphs import Edge, EdgeOnSet, Node
from plesna.models.flux import Flux, Transformation
from plesna.storage.repository.fs_repository import FSRepository
@@ -127,13 +127,7 @@ def test_listing_content(dataplatform: DataPlateform):
]
def test_content_from_graph(dataplatform: DataPlateform):
assert dataplatform.graph.nodes == {
Node(name="test-raw-recovery", infos={}),
Node(name="test-raw-salary", infos={}),
Node(name="test-raw-username", infos={}),
}
def test_content_from_graphset(dataplatform: DataPlateform):
assert dataplatform.graphset.node_sets == {
frozenset(
{
@@ -157,6 +151,51 @@ def test_content_from_graph(dataplatform: DataPlateform):
}
),
}
assert dataplatform.graphset.edges == [
EdgeOnSet(
arrow="foo_flux",
sources=[Node(name="test-raw-username"), Node(name="test-raw-recovery")],
targets=[Node(name="test-bronze-foo")],
edge_kwrds={},
),
EdgeOnSet(
arrow="copy_flux",
sources=[Node(name="test-raw-username")],
targets=[Node(name="test-bronze-username")],
edge_kwrds={},
),
]
def test_content_from_graph(dataplatform: DataPlateform):
assert dataplatform.graph.nodes == {
Node(name="test-raw-recovery", infos={}),
Node(name="test-raw-salary", infos={}),
Node(name="test-raw-username", infos={}),
Node(name="test-bronze-username", infos={}),
Node(name="test-bronze-foo", infos={}),
Node(name="test-raw-username", infos={}),
}
assert dataplatform.graph.edges == [
Edge(
arrow="foo_flux",
source=Node(name="test-raw-username"),
target=Node(name="test-bronze-foo"),
edge_kwrds={},
),
Edge(
arrow="foo_flux",
source=Node(name="test-raw-recovery"),
target=Node(name="test-bronze-foo"),
edge_kwrds={},
),
Edge(
arrow="copy_flux",
source=Node(name="test-raw-username"),
target=Node(name="test-bronze-username"),
edge_kwrds={},
),
]
def test_execute_flux(dataplatform: DataPlateform):