Add Speech sync recognize system test.

daspecster · daspecster · commit aa3b0f5ff357 · 2016-10-27T11:03:07.000-04:00
diff --git a/system_tests/attempt_system_tests.py b/system_tests/attempt_system_tests.py
@@ -59,6 +59,7 @@
 MODULES = (  # ordered from most to least stable
     'datastore',
     'storage',
+    'speech',
     'bigquery',
     'pubsub',
     'language',
diff --git a/system_tests/data/hello.wav b/system_tests/data/hello.wav
diff --git a/system_tests/run_system_test.py b/system_tests/run_system_test.py
@@ -24,13 +24,15 @@
 import monitoring
 import pubsub
 import storage
+import speech
 import system_test_utils
 import translate
 
 
 TEST_MODULES = {
     'datastore': datastore,
     'storage': storage,
+    'speech': speech,
     'pubsub': pubsub,
     'bigquery': bigquery,
     'bigtable': bigtable,
diff --git a/system_tests/speech.py b/system_tests/speech.py
@@ -0,0 +1,54 @@
+# Copyright 2016 Google Inc.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+import unittest
+
+
+class TestSpeechClient(unittest.TestCase):
+
+    def test_sync_recognize_local_file(self):
+        import io
+        from google.cloud import speech
+        client = speech.Client()
+        file_name = 'system_tests/data/hello.wav'
+
+        with io.open(file_name, 'rb') as file_obj:
+            sample = client.sample(content=file_obj.read(),
+                                   encoding=speech.Encoding.LINEAR16,
+                                   sample_rate=16000)
+            res = client.sync_recognize(sample,
+                                        language_code='en-US',
+                                        max_alternatives=2,
+                                        profanity_filter=True,
+                                        speech_context=['Google', 'cloud'])
+            self.assertEqual(res[0].transcript,
+                             'hello thank you for using Google Cloud platform')
+            self.assertEqual(len(res), 2)
+
+    def test_sync_recognize_gcs_file(self):
+        from google.cloud import speech
+        client = speech.Client()
+        source_uri = 'gs://ferrous-arena-my-test-bucket/hello.wav'
+
+        sample = client.sample(source_uri=source_uri,
+                               encoding=speech.Encoding.LINEAR16,
+                               sample_rate=16000)
+        res = client.sync_recognize(sample,
+                                    language_code='en-US',
+                                    max_alternatives=1,
+                                    profanity_filter=True,
+                                    speech_context=['Google', 'cloud'])
+        self.assertEqual(res[0].transcript,
+                         'hello thank you for using Google Cloud platform')
+        self.assertEqual(len(res), 1)
