fix formatting for readability

ExpediaGroup · Oct 30, 2023 · f28bf74 · f28bf74
1 parent 0317133
commit f28bf74
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/sdk/python/tests/expediagroup/test_elasticsearch_online_store.py b/sdk/python/tests/expediagroup/test_elasticsearch_online_store.py
@@ -171,6 +171,7 @@ def test_elasticsearch_update_add_index(self, repo_config, caplog, index_params)
                 "type": index_params["index_type"].lower(),
                 **index_params["index_params"],
             }
+
         with ElasticsearchConnectionManager(repo_config.online_store) as es:
             created_index = es.indices.get(index=self.index_to_write)
             assert created_index.body[self.index_to_write]["mappings"] == mapping
@@ -208,6 +209,7 @@ def test_elasticsearch_update_add_existing_index(self, repo_config, caplog):
             entities_to_keep=[],
             partial=False,
         )
+
         with ElasticsearchConnectionManager(repo_config.online_store) as es:
             assert es.indices.exists(index=self.index_to_write).body is True
 
@@ -229,6 +231,7 @@ def test_elasticsearch_update_delete_index(self, repo_config, caplog):
             ),
         ]
         self._create_index_in_es(self.index_to_delete, repo_config)
+
         with ElasticsearchConnectionManager(repo_config.online_store) as es:
             assert es.indices.exists(index=self.index_to_delete).body is True
 
@@ -247,6 +250,7 @@ def test_elasticsearch_update_delete_index(self, repo_config, caplog):
             entities_to_keep=[],
             partial=False,
         )
+
         with ElasticsearchConnectionManager(repo_config.online_store) as es:
             assert es.indices.exists(index=self.index_to_delete).body is False
 
@@ -267,6 +271,7 @@ def test_elasticsearch_update_delete_unavailable_index(self, repo_config, caplog
                 dtype=String,
             ),
         ]
+
         with ElasticsearchConnectionManager(repo_config.online_store) as es:
             assert es.indices.exists(index=self.index_to_delete).body is False
 
@@ -285,6 +290,7 @@ def test_elasticsearch_update_delete_unavailable_index(self, repo_config, caplog
             entities_to_keep=[],
             partial=False,
         )
+
         with ElasticsearchConnectionManager(repo_config.online_store) as es:
             assert es.indices.exists(index=self.index_to_delete).body is False
 
@@ -333,8 +339,10 @@ def test_elasticsearch_online_read(self, repo_config, caplog):
             data=data,
             progress=None,
         )
+
         with ElasticsearchConnectionManager(repo_config.online_store) as es:
             es.indices.refresh(index=self.index_to_read)
+
         result = store.online_read(
             config=repo_config.online_store,
             table=feature_view,
@@ -373,8 +381,10 @@ def test_elasticsearch_online_read_with_requested_features(
             data=data,
             progress=None,
         )
+
         with ElasticsearchConnectionManager(repo_config.online_store) as es:
             es.indices.refresh(index=self.index_to_read)
+
         result = store.online_read(
             config=repo_config.online_store,
             table=feature_view,