You are viewing a plain text version of this content. The canonical link for it is here.
Posted to commits@beam.apache.org by dh...@apache.org on 2017/05/02 05:44:07 UTC

[4/5] beam git commit: [BEAM-59] TFRecordIOTest: cleanup

[BEAM-59] TFRecordIOTest: cleanup

1. Shrink file size slightly to reduce test time.
2. Switch from relative to absolute local file paths, because in the future
   this distinction will matter.


Project: http://git-wip-us.apache.org/repos/asf/beam/repo
Commit: http://git-wip-us.apache.org/repos/asf/beam/commit/a77ed33e
Tree: http://git-wip-us.apache.org/repos/asf/beam/tree/a77ed33e
Diff: http://git-wip-us.apache.org/repos/asf/beam/diff/a77ed33e

Branch: refs/heads/master
Commit: a77ed33e91e83e13e4e32e03540a1c310ddbeb8c
Parents: e83e562
Author: Dan Halperin <dh...@google.com>
Authored: Mon May 1 18:46:24 2017 -0700
Committer: Dan Halperin <dh...@google.com>
Committed: Mon May 1 22:43:54 2017 -0700

----------------------------------------------------------------------
 .../src/test/java/org/apache/beam/sdk/io/TFRecordIOTest.java   | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)
----------------------------------------------------------------------


http://git-wip-us.apache.org/repos/asf/beam/blob/a77ed33e/sdks/java/core/src/test/java/org/apache/beam/sdk/io/TFRecordIOTest.java
----------------------------------------------------------------------
diff --git a/sdks/java/core/src/test/java/org/apache/beam/sdk/io/TFRecordIOTest.java b/sdks/java/core/src/test/java/org/apache/beam/sdk/io/TFRecordIOTest.java
index ae3a50d..9a9e840 100644
--- a/sdks/java/core/src/test/java/org/apache/beam/sdk/io/TFRecordIOTest.java
+++ b/sdks/java/core/src/test/java/org/apache/beam/sdk/io/TFRecordIOTest.java
@@ -93,7 +93,7 @@ public class TFRecordIOTest {
   private static final String[] FOO_BAR_RECORDS = {"foo", "bar"};
 
   private static final Iterable<String> EMPTY = Collections.emptyList();
-  private static final Iterable<String> LARGE = makeLines(5000);
+  private static final Iterable<String> LARGE = makeLines(1000);
 
   private static Path tempFolder;
 
@@ -159,7 +159,7 @@ public class TFRecordIOTest {
   @Test
   public void testWriteDisplayData() {
     TFRecordIO.Write write = TFRecordIO.write()
-        .to("foo")
+        .to("/foo")
         .withSuffix("bar")
         .withShardNameTemplate("-SS-of-NN-")
         .withNumShards(100)
@@ -167,7 +167,7 @@ public class TFRecordIOTest {
 
     DisplayData displayData = DisplayData.from(write);
 
-    assertThat(displayData, hasDisplayItem("filePrefix", "foo"));
+    assertThat(displayData, hasDisplayItem("filePrefix", "/foo"));
     assertThat(displayData, hasDisplayItem("fileSuffix", "bar"));
     assertThat(displayData, hasDisplayItem("shardNameTemplate", "-SS-of-NN-"));
     assertThat(displayData, hasDisplayItem("numShards", 100));