Formatting

delftdata · Dec 16, 2024 · 8cb6353 · 8cb6353
1 parent 99a002d
commit 8cb6353
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 10 deletions.
diff --git a/tests/integration/flink-runtime/test_merge_operator.py b/tests/integration/flink-runtime/test_merge_operator.py
@@ -1,9 +1,8 @@
-"""A test script for dataflows with merge operators
-"""
-from pyflink.datastream.data_stream import CloseableIterator
+"""A test script for dataflows with merge operators"""
 
+from pyflink.datastream.data_stream import CloseableIterator
 from common import Item, User, item_op, user_op
-from cascade.dataflow.dataflow import DataFlow, Edge, Event, EventResult, InitClass, InvokeMethod, MergeNode, Node, OpNode
+from cascade.dataflow.dataflow import Event, EventResult, InitClass, InvokeMethod, OpNode
 from cascade.runtime.flink_runtime import FlinkOperator, FlinkRuntime
 import pytest
 

diff --git a/tests/integration/flink-runtime/test_two_entities.py b/tests/integration/flink-runtime/test_two_entities.py
@@ -1,12 +1,8 @@
-"""A test script for dataflows with multiple Operators
-"""
-
-import time
+"""A test script for dataflows with multiple operators"""
 
 from pyflink.datastream.data_stream import CloseableIterator
-
 from common import Item, User, item_op, user_op
-from cascade.dataflow.dataflow import DataFlow, Edge, Event, EventResult, InitClass, InvokeMethod, Node, OpNode
+from cascade.dataflow.dataflow import Event, EventResult, InitClass, InvokeMethod, OpNode
 from cascade.runtime.flink_runtime import FlinkOperator, FlinkRuntime
 import pytest