Skip to content

Commit bbaaf54

Browse files
1 parent 5f79b53 commit bbaaf54

File tree

1 file changed

+18
-6
lines changed

1 file changed

+18
-6
lines changed

‎tests/unit/test_to_gbq.py‎

Lines changed: 18 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -214,21 +214,27 @@ def test_create_user_agent_vscode():
214214

215215
assert create_user_agent() == f"pandas-{pd.__version__} vscode"
216216

217+
217218
@mock.patch.dict(os.environ, {"VSCODE_PID": "1234"}, clear=True)
218219
def test_create_user_agent_vscode_plugin():
219220
from pandas_gbq.gbq import create_user_agent
220221

221222
# simulate plugin installation by creating plugin config on disk
222223
user_home = os.path.expanduser("~")
223-
plugin_dir = os.path.join(user_home, ".vscode", "extensions", "googlecloudtools.cloudcode-0.12")
224+
plugin_dir = os.path.join(
225+
user_home, ".vscode", "extensions", "googlecloudtools.cloudcode-0.12"
226+
)
224227
plugin_config = os.path.join(plugin_dir, "package.json")
225-
assert not os.path.exists(plugin_config) # initially does not exist
228+
assert not os.path.exists(plugin_config) # initially does not exist
226229
os.makedirs(plugin_dir, exist_ok=True)
227230
with open(plugin_config, "w") as f:
228231
f.write("{}")
229232

230233
# test
231-
assert create_user_agent() == f"pandas-{pd.__version__} vscode googlecloudtools.cloudcode"
234+
assert (
235+
create_user_agent()
236+
== f"pandas-{pd.__version__} vscode googlecloudtools.cloudcode"
237+
)
232238

233239
# clean up disk
234240
os.remove(plugin_config)
@@ -251,7 +257,13 @@ def custom_import_module_side_effect(name, package=None):
251257
return mock.MagicMock()
252258
else:
253259
import importlib
260+
254261
return importlib.import_module(name, package)
255-
256-
with mock.patch('importlib.import_module', side_effect=custom_import_module_side_effect):
257-
assert create_user_agent() == f"pandas-{pd.__version__} jupyter bigquery_jupyter_plugin"
262+
263+
with mock.patch(
264+
"importlib.import_module", side_effect=custom_import_module_side_effect
265+
):
266+
assert (
267+
create_user_agent()
268+
== f"pandas-{pd.__version__} jupyter bigquery_jupyter_plugin"
269+
)

0 commit comments

Comments
 (0)