Bläddra i källkod

test: correct imported name

tags/2.0.0-beta.1
-LAN- 1 månad sedan
förälder
incheckning
61c79b0013
Inget konto är kopplat till bidragsgivarens mejladress

+ 4
- 4
api/tests/unit_tests/core/workflow/graph_engine/test_graph_engine.py Visa fil

@@ -498,10 +498,10 @@ def test_layer_system_basic():

def test_layer_chaining():
"""Test chaining multiple layers."""
from core.workflow.graph_engine.layers import DebugLoggingLayer, Layer
from core.workflow.graph_engine.layers import DebugLoggingLayer, GraphEngineLayer

# Create a custom test layer
class TestLayer(Layer):
class TestLayer(GraphEngineLayer):
def __init__(self):
super().__init__()
self.events_received = []
@@ -560,10 +560,10 @@ def test_layer_chaining():

def test_layer_error_handling():
"""Test that layer errors don't crash the engine."""
from core.workflow.graph_engine.layers import Layer
from core.workflow.graph_engine.layers import GraphEngineLayer

# Create a layer that throws errors
class FaultyLayer(Layer):
class FaultyLayer(GraphEngineLayer):
def on_graph_start(self):
raise RuntimeError("Intentional error in on_graph_start")


Laddar…
Avbryt
Spara