lafrite/plesna
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

refact: move Transformation to flux model

Browse Source
This commit is contained in:
Bertrand Benjamin
2025-01-05 15:43:29 +01:00
parent 9a5c581f31
commit e4af62b136
4 changed files with 20 additions and 24 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
tests/dataplatform/test_dataplateform.py
View File

@@ -4,9 +4,8 @@ from pathlib import Path
import pytest
from plesna.dataplatform import DataPlateform
from plesna.graph.graph import Node
from plesna.models.flux import Flux
from plesna.models.transformation import Transformation
from plesna.models.graphs import Node
from plesna.models.flux import Flux, Transformation
from plesna.storage.repository.fs_repository import FSRepository
FIXTURE_DIR = Path(__file__).parent.parent / Path("raw_datas")
@@ -55,7 +54,7 @@ def copy_flux(repository: FSRepository) -> Flux:
raw_brz_copy_username = Flux(
sources=raw_username,
targets=bronze_username,
transformation=Transformation(function=copy, extra_kwrds=extra_kwrds),
transformation=Transformation(name="copy", function=copy, extra_kwrds=extra_kwrds),
)
return raw_brz_copy_username
@@ -76,7 +75,7 @@ def foo_flux(repository: FSRepository) -> Flux:
flux = Flux(
sources=src,
targets=targets,
transformation=Transformation(function=foo, extra_kwrds=extra_kwrds),
transformation=Transformation(name="foo", function=foo, extra_kwrds=extra_kwrds),
)
return flux