From 43424503600e392dd4300a302ffdab65419b9b9b Mon Sep 17 00:00:00 2001
From: GudniRos <gkrosen@gmail.com>
Date: Thu, 17 Dec 2015 23:02:52 -0800
Subject: [PATCH] Updated test_MeshIO to remove the temp files after using
 them.

---
 tests/mesh/test_MeshIO.py | 7 +++++--
 1 file changed, 5 insertions(+), 2 deletions(-)

diff --git a/tests/mesh/test_MeshIO.py b/tests/mesh/test_MeshIO.py
index e3e49548..43c0e33e 100644
--- a/tests/mesh/test_MeshIO.py
+++ b/tests/mesh/test_MeshIO.py
@@ -1,5 +1,5 @@
 import numpy as np
-import unittest
+import unittest, os
 import SimPEG as simpeg
 from SimPEG.Mesh import TensorMesh, TreeMesh
 
@@ -29,17 +29,20 @@ class TestOcTreeIO(unittest.TestCase):
         assert np.sum(vec - vecUBC) == 0
         assert np.all(np.array(mesh.h) - np.array(meshUBC.h) == 0)
         print 'IO of UBC octree files is working'
+        os.remove('temp.msh')
+        os.remove('arange.txt')
 
     def test_VTUfiles(self):
         mesh = self.mesh
         vec = np.arange(mesh.nC)
         try:
-            simpeg.Utils.meshutils.writeVTUFile('test.vtu',mesh,{'arange':vec})
+            simpeg.Utils.meshutils.writeVTUFile('temp.vtu',mesh,{'arange':vec})
             run = True
         except:
             run = False
         assert run
         print 'Writing of VTU files is working'
+        os.remove('temp.vtu')