Skip to content

Commit 0fcd66b

Browse files
committed
Remove use_gax, fix assert order and ctx manager.
1 parent 3755adb commit 0fcd66b

File tree

1 file changed

+15
-16
lines changed

1 file changed

+15
-16
lines changed

system_tests/speech.py

Lines changed: 15 additions & 16 deletions
Original file line numberDiff line numberDiff line change
@@ -63,7 +63,6 @@ class Config(object):
6363
def setUpModule():
6464
Config.CLIENT = speech.Client()
6565
Config.USE_GAX = Config.CLIENT._use_gax
66-
Config.CLIENT = speech.Client
6766
# Now create a bucket for GCS stored content.
6867
storage_client = storage.Client()
6968
bucket_name = 'new' + unique_resource_id()
@@ -92,8 +91,8 @@ def tearDown(self):
9291
value.delete()
9392

9493
def _make_sync_request(self, content=None, source_uri=None,
95-
max_alternatives=None, use_gax=True):
96-
client = Config.CLIENT(use_gax=use_gax)
94+
max_alternatives=None):
95+
client = Config.CLIENT
9796
sample = client.sample(content=content,
9897
source_uri=source_uri,
9998
encoding=speech.Encoding.LINEAR16,
@@ -126,14 +125,16 @@ def _check_best_results(self, results):
126125

127126
def test_sync_recognize_local_file(self):
128127
with open(AUDIO_FILE, 'rb') as file_obj:
129-
results = self._make_sync_request(content=file_obj.read(),
130-
max_alternatives=2)
131-
second_alternative = results[1]
132-
self.assertEqual(len(results), 2)
133-
self._check_best_results(results)
134-
self.assertIsInstance(second_alternative, Transcript)
135-
self.assertEqual(second_alternative.transcript, self.ASSERT_TEXT)
136-
self.assertEqual(second_alternative.confidence, 0.0)
128+
content = file_obj.read()
129+
results = self._make_sync_request(content=content,
130+
max_alternatives=2)
131+
132+
self.assertEqual(len(results), 2)
133+
second_alternative = results[1]
134+
self._check_best_results(results)
135+
self.assertIsInstance(second_alternative, Transcript)
136+
self.assertEqual(second_alternative.transcript, self.ASSERT_TEXT)
137+
self.assertEqual(second_alternative.confidence, 0.0)
137138

138139
def test_sync_recognize_gcs_file(self):
139140
bucket_name = Config.TEST_BUCKET.name
@@ -151,14 +152,13 @@ def test_sync_recognize_gcs_file(self):
151152
def test_sync_recognize_local_file_rest(self):
152153
with open(AUDIO_FILE, 'rb') as file_obj:
153154
results = self._make_sync_request(content=file_obj.read(),
154-
max_alternatives=2,
155-
use_gax=False)
155+
max_alternatives=2)
156156
second_alternative = results[1]
157157
self.assertEqual(len(results), 2)
158158
self._check_best_results(results)
159159
self.assertIsInstance(second_alternative, Transcript)
160160
self.assertEqual(second_alternative.transcript, self.ASSERT_TEXT)
161-
self.assertEqual(second_alternative.confidence, None)
161+
self.assertEqual(second_alternative.confidence, 0.0)
162162

163163
def test_sync_recognize_gcs_file_rest(self):
164164
bucket_name = Config.TEST_BUCKET.name
@@ -170,8 +170,7 @@ def test_sync_recognize_gcs_file_rest(self):
170170

171171
source_uri = 'gs://%s/%s' % (bucket_name, blob_name)
172172
result = self._make_sync_request(source_uri=source_uri,
173-
max_alternatives=1,
174-
use_gax=False)
173+
max_alternatives=1)
175174
self._check_best_results(result)
176175

177176
def test_async_recognize_local_file(self):

0 commit comments

Comments
 (0)