diff options
Diffstat (limited to 'exec/java-exec/src/test/java/org/apache/drill/exec/TestOpSerialization.java')
-rw-r--r-- | exec/java-exec/src/test/java/org/apache/drill/exec/TestOpSerialization.java | 5 |
1 files changed, 3 insertions, 2 deletions
diff --git a/exec/java-exec/src/test/java/org/apache/drill/exec/TestOpSerialization.java b/exec/java-exec/src/test/java/org/apache/drill/exec/TestOpSerialization.java index 0f0bf0df0..7fb7bdbe4 100644 --- a/exec/java-exec/src/test/java/org/apache/drill/exec/TestOpSerialization.java +++ b/exec/java-exec/src/test/java/org/apache/drill/exec/TestOpSerialization.java @@ -38,6 +38,7 @@ import org.apache.drill.exec.physical.config.Screen; import org.apache.drill.exec.physical.config.UnionExchange; import org.apache.drill.exec.planner.PhysicalPlanReader; import org.apache.drill.exec.planner.PhysicalPlanReaderTestFactory; +import org.apache.drill.exec.planner.cost.PrelCostEstimates; import org.apache.drill.exec.proto.CoordinationProtos; import org.apache.drill.exec.store.direct.DirectSubScan; import org.apache.drill.exec.store.mock.MockSubScanPOP; @@ -58,7 +59,7 @@ public class TestOpSerialization { private static PhysicalOperator setupPhysicalOperator(PhysicalOperator operator) { operator.setOperatorId(1); - operator.setCost(1.0); + operator.setCost(new PrelCostEstimates(1.0, 1.0)); operator.setMaxAllocation(1000); return operator; } @@ -66,7 +67,7 @@ public class TestOpSerialization { private static void assertOperator(PhysicalOperator operator) { assertEquals(1, operator.getOperatorId()); - assertEquals(1.0, operator.getCost(), 0.00001); + assertEquals(1.0, operator.getCost().getOutputRowCount(), 0.00001); assertEquals(1000, operator.getMaxAllocation()); } |