using pop for array element deletion

passing exception during folder removal
removal of additional variable

Author: anjsudh

pandas/io/parquet.py

@@ -225,8 +225,7 @@ def write(self, df, path, compression='snappy', index=None,
         # Use tobytes() instead.
 
         if 'partition_on' in kwargs:
-            partition_cols = kwargs['partition_on']
-            del kwargs['partition_on']
+            partition_cols = kwargs.pop('partition_on')
 
         if partition_cols is not None:
             kwargs['file_scheme'] = 'hive'

pandas/tests/util/test_testing.py

@@ -878,9 +878,7 @@ def test_datapath_missing(datapath, request):
 
 
 def test_create_temp_directory():
-    temppath = ''
     with tm.ensure_clean_dir() as path:
         assert os.path.exists(path)
         assert os.path.isdir(path)
-        temppath = path
-    assert not os.path.exists(temppath)
+    assert not os.path.exists(path)

pandas/util/testing.py

@@ -51,6 +51,7 @@
 
 from pandas._libs import testing as _testing
 from pandas.io.common import urlopen
+from shutil import rmtree
 
 
 N = 30
@@ -786,10 +787,9 @@ def ensure_clean_dir():
         yield directory_name
     finally:
         try:
-            import shutil
-            shutil.rmtree(directory_name)
+            rmtree(directory_name)
         except Exception as e:
-            print("Exception on removing folder: {error}".format(error=e))
+            pass
 
 
 # -----------------------------------------------------------------------------