Skip to content
Original file line number Diff line number Diff line change
Expand Up @@ -166,8 +166,14 @@ class ReconfigurationIntegrationSpec
): Map[OperatorIdentity, List[Tuple]] =
TestUtils.shouldReconfigure(system, ctx, operators, links, targetOps, newOpExecInitInfo)

private def boundedCsvSource() = {
val src = TestOperators.mediumCsvScanOpDesc()
src.limit = Some(10000)
src
}
Comment thread
Ma77Ball marked this conversation as resolved.
Outdated

"Engine" should "be able to modify a python UDF worker in workflow" in {
val sourceOpDesc = TestOperators.smallCsvScanOpDesc()
val sourceOpDesc = boundedCsvSource()
val udfOpDesc = TestOperators.pythonOpDesc()
val code = """
|from pytexera import *
Expand Down Expand Up @@ -228,7 +234,7 @@ class ReconfigurationIntegrationSpec
}

"Engine" should "be able to modify two python UDFs in workflow" in {
val sourceOpDesc = TestOperators.smallCsvScanOpDesc()
val sourceOpDesc = boundedCsvSource()
val udfOpDesc1 = TestOperators.pythonOpDesc()
val udfOpDesc2 = TestOperators.pythonOpDesc()
val code = """
Expand Down
Loading