condacy error fix

4dn-dcic · Nov 14, 2016 · a7e187a · a7e187a
1 parent 52d4b88
commit a7e187a
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/tests/test_get_field_info.py b/tests/test_get_field_info.py
@@ -114,9 +114,13 @@ def test_create_xls(connection, mocker, returned_vendor_schema):
     import os
     try:
         os.remove(xls_file)
-    except:
+    except OSError:
         pass
     with mocker.patch('wranglertools.fdnDCIC.requests.get', return_value=returned_vendor_schema):
         field_dict = gfi.get_uploadable_fields(connection, ['Vendor'])
         gfi.create_xls(field_dict, xls_file)
         assert os.path.isfile(xls_file)
+    try:
+        os.remove(xls_file)
+    except OSError:
+        pass