Additional samples in GA - model-selection

nirupa-kumar · nirupa-kumar · commit 64fbff59c347 · 2018-09-17T14:46:38.000-07:00
diff --git a/speech/beta/src/test/java/com/example/speech/RecognizeIT.java b/speech/beta/src/test/java/com/example/speech/RecognizeIT.java
@@ -60,22 +60,6 @@ public void tearDown() {
     System.setOut(null);
   }
 
-  @Test
-  public void testModelSelection() throws Exception {
-    Recognize.transcribeModelSelection(videoFileName);
-    String got = bout.toString();
-    assertThat(got).contains("OK Google");
-    assertThat(got).contains("the weather outside is sunny");
-  }
-
-  @Test
-  public void testGcsModelSelection() throws Exception {
-    Recognize.transcribeModelSelectionGcs(gcsVideoPath);
-    String got = bout.toString();
-    assertThat(got).contains("OK Google");
-    assertThat(got).contains("the weather outside is sunny");
-  }
-
   @Test
   public void testMetadata() throws Exception {
     Recognize.transcribeFileWithMetadata(recognitionAudioFile);
diff --git a/speech/cloud-client/src/test/java/com/example/speech/RecognizeIT.java b/speech/cloud-client/src/test/java/com/example/speech/RecognizeIT.java
@@ -38,9 +38,12 @@ public class RecognizeIT {
   // The path to the audio file to transcribe
   private String audioFileName = "./resources/audio.raw";
   private String gcsAudioPath = "gs://" + BUCKET + "/speech/brooklyn.flac";
-
   private String recognitionAudioFile = "./resources/commercial_mono.wav";
 
+  // The path to the video file to transcribe
+  private String videoFileName = "./resources/Google_Gnome.wav";
+  private String gcsVideoPath = "gs://" + BUCKET + "/speech/Google_Gnome.wav";
+
   @Before
   public void setUp() {
     bout = new ByteArrayOutputStream();
@@ -132,4 +135,19 @@ public void testEnhancedModel() throws Exception {
     assertThat(got).contains("Chrome");
   }
 
+  @Test
+  public void testModelSelection() throws Exception {
+    Recognize.transcribeModelSelection(videoFileName);
+    String got = bout.toString();
+    assertThat(got).contains("OK Google");
+    assertThat(got).contains("the weather outside is sunny");
+  }
+
+  @Test
+  public void testGcsModelSelection() throws Exception {
+    Recognize.transcribeModelSelectionGcs(gcsVideoPath);
+    String got = bout.toString();
+    assertThat(got).contains("OK Google");
+    assertThat(got).contains("the weather outside is sunny");
+  }
 }
