@@ -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 )
218219def 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