Adding cleanup_saved_downloads function to help the testcleanup directive

Author: bsipocz

astroquery/utils/cleanup_downloads.py

@@ -0,0 +1,28 @@
+# Licensed under a 3-clause BSD style license - see LICENSE.rst
+"""
+Utility to cleanup files created during doctesting.
+"""
+import glob
+import os
+import shutil
+
+
+def cleanup_saved_downloads(names):
+    """ Function to clean up save files.
+
+    Parameters
+    ----------
+    names : str or list of str
+        Files or directories to clean up. Wildcards are excepted.
+    """
+
+    if isinstance(names, str):
+        names = [names]
+
+    for path in names:
+        files = glob.glob(path)
+        for saved_download in files:
+            try:
+                shutil.rmtree(saved_download)
+            except NotADirectoryError:
+                os.remove(saved_download)