fixup! Add display data for MapElements transform

apache · May 10, 2016 · 8fe5624 · 8fe5624
1 parent 3a5bb16
commit 8fe5624
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/...va/src/test/java/org/apache/beam/runners/dataflow/transforms/DataflowMapElementsTest.java b/...va/src/test/java/org/apache/beam/runners/dataflow/transforms/DataflowMapElementsTest.java
@@ -48,7 +48,8 @@ public Integer apply(Integer input) {
     MapElements<?, ?> map = MapElements.via(mapFn);
     DisplayDataEvaluator evaluator = DataflowDisplayDataEvaluator.create();
 
-    Set<DisplayData> displayDatas = evaluator.displayDataForPrimitiveTransforms(map);
-    assertThat(displayDatas, hasItem(hasDisplayItem("mapFn", mapFn.getClass())));
+    Set<DisplayData> displayData = evaluator.displayDataForPrimitiveTransforms(map);
+    assertThat("MapElements should include the mapFn in its primitive display data",
+        displayData, hasItem(hasDisplayItem("mapFn", mapFn.getClass())));
   }
 }
diff --git a/sdks/java/core/src/test/java/org/apache/beam/sdk/transforms/MapElementsTest.java b/sdks/java/core/src/test/java/org/apache/beam/sdk/transforms/MapElementsTest.java
@@ -106,7 +106,7 @@ public String apply(String input) {
     assertThat(pipeline.getCoderRegistry().getDefaultCoder(output.getTypeDescriptor()),
         equalTo(pipeline.getCoderRegistry().getDefaultCoder(new TypeDescriptor<String>() {})));
 
-    // Make sure the  pipelien runs too
+    // Make sure the pipeline runs too
     pipeline.run();
   }