lint

GoogleCloudPlatform · Apr 2, 2020 · 6e8d352 · 6e8d352
1 parent c6254e4
commit 6e8d352
Show file tree

Hide file tree

Showing 2 changed files with 34 additions and 36 deletions.
diff --git a/healthcare/api-client/hl7v2/hl7v2_messages_test.py b/healthcare/api-client/hl7v2/hl7v2_messages_test.py
@@ -15,7 +15,6 @@
 import os
 import pytest
 import random
-import string
 import sys
 import uuid
 
@@ -89,17 +88,17 @@ def test_CRUD_hl7v2_message(test_dataset, test_hl7v2_store, capsys):
     hl7v2_message_id = ""
     @eventually_consistent.call
     def _():
-      hl7v2_messages_list = hl7v2_messages.list_hl7v2_messages(
-          service_account_json,
-          project_id,
-          cloud_region,
-          dataset_id,
-          hl7v2_store_id)
-
-      assert len(hl7v2_messages_list) > 0
-      hl7v2_message_name = hl7v2_messages_list[0].get('name')
-      nonlocal hl7v2_message_id
-      hl7v2_message_id = hl7v2_message_name.split('/', 9)[9]
+        hl7v2_messages_list = hl7v2_messages.list_hl7v2_messages(
+            service_account_json,
+            project_id,
+            cloud_region,
+            dataset_id,
+            hl7v2_store_id)
+
+        assert len(hl7v2_messages_list) > 0
+        hl7v2_message_name = hl7v2_messages_list[0].get('name')
+        nonlocal hl7v2_message_id
+        hl7v2_message_id = hl7v2_message_name.split('/', 9)[9]
 
     hl7v2_messages.get_hl7v2_message(
         service_account_json,
@@ -137,17 +136,17 @@ def test_ingest_hl7v2_message(test_dataset, test_hl7v2_store, capsys):
     hl7v2_message_id = ""
     @eventually_consistent.call
     def _():
-      hl7v2_messages_list = hl7v2_messages.list_hl7v2_messages(
-          service_account_json,
-          project_id,
-          cloud_region,
-          dataset_id,
-          hl7v2_store_id)
-
-      assert len(hl7v2_messages_list) > 0
-      hl7v2_message_name = hl7v2_messages_list[0].get('name')
-      nonlocal hl7v2_message_id
-      hl7v2_message_id = hl7v2_message_name.split('/', 9)[9]
+        hl7v2_messages_list = hl7v2_messages.list_hl7v2_messages(
+            service_account_json,
+            project_id,
+            cloud_region,
+            dataset_id,
+            hl7v2_store_id)
+
+        assert len(hl7v2_messages_list) > 0
+        hl7v2_message_name = hl7v2_messages_list[0].get('name')
+        nonlocal hl7v2_message_id
+        hl7v2_message_id = hl7v2_message_name.split('/', 9)[9]
 
     hl7v2_messages.get_hl7v2_message(
         service_account_json,
@@ -185,17 +184,17 @@ def test_patch_hl7v2_message(test_dataset, test_hl7v2_store, capsys):
     hl7v2_message_id = ""
     @eventually_consistent.call
     def _():
-      hl7v2_messages_list = hl7v2_messages.list_hl7v2_messages(
-          service_account_json,
-          project_id,
-          cloud_region,
-          dataset_id,
-          hl7v2_store_id)
-
-      assert len(hl7v2_messages_list) > 0
-      hl7v2_message_name = hl7v2_messages_list[0].get('name')
-      nonlocal hl7v2_message_id
-      hl7v2_message_id = hl7v2_message_name.split('/', 9)[9]
+        hl7v2_messages_list = hl7v2_messages.list_hl7v2_messages(
+            service_account_json,
+            project_id,
+            cloud_region,
+            dataset_id,
+            hl7v2_store_id)
+
+        assert len(hl7v2_messages_list) > 0
+        hl7v2_message_name = hl7v2_messages_list[0].get('name')
+        nonlocal hl7v2_message_id
+        hl7v2_message_id = hl7v2_message_name.split('/', 9)[9]
 
     hl7v2_messages.patch_hl7v2_message(
         service_account_json,

diff --git a/healthcare/api-client/hl7v2/hl7v2_stores_test.py b/healthcare/api-client/hl7v2/hl7v2_stores_test.py
@@ -15,9 +15,7 @@
 import os
 import pytest
 import random
-import string
 import sys
-import time
 import uuid
 
 # Add datasets for bootstrapping datasets for testing
@@ -34,6 +32,7 @@
 dataset_id = 'test_dataset_{}'.format(uuid.uuid4())
 hl7v2_store_id = 'test_hl7v2_store-{}'.format(uuid.uuid4())
 
+
 @pytest.fixture(scope='module')
 def test_dataset():
     dataset = datasets.create_dataset(