diff --git a/tests.py b/tests.py index 4b853e4fd0d61247fe68a75c481ffc935a542ba8..bf92c9ffb3aa9b417d2aefce64111128f83f329b 100644 --- a/tests.py +++ b/tests.py @@ -4,6 +4,7 @@ import os import shutil import pyfstat import lalpulsar +import logging class Test(unittest.TestCase): @@ -12,7 +13,11 @@ class Test(unittest.TestCase): @classmethod def setUpClass(self): if os.path.isdir(self.outdir): - shutil.rmtree(self.outdir) + try: + shutil.rmtree(self.outdir) + except OSError: + logging.warning( + "{} not removed prior to tests".format(self.outdir)) h0 = 1 sqrtSX = 1 F0 = 30 @@ -38,7 +43,11 @@ class Test(unittest.TestCase): @classmethod def tearDownClass(self): if os.path.isdir(self.outdir): - shutil.rmtree(self.outdir) + try: + shutil.rmtree(self.outdir) + except OSError: + logging.warning( + "{} not removed prior to tests".format(self.outdir)) class Writer(Test):