test: correct imported name

This commit is contained in:
-LAN- 2025-09-04 02:15:46 +08:00
parent 8332472944
commit 61c79b0013
No known key found for this signature in database
GPG Key ID: 6BA0D108DED011FF
1 changed files with 4 additions and 4 deletions

View File

@ -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")