From 687093e0e4c4db73c4855a33da781b3ce2e68589 Mon Sep 17 00:00:00 2001 From: Sander Roosendaal Date: Sat, 16 Jan 2021 14:00:20 +0100 Subject: [PATCH] cleaning up temp --- rowers/tests/test_async_tasks.py | 18 +++++++++++++++++- 1 file changed, 17 insertions(+), 1 deletion(-) diff --git a/rowers/tests/test_async_tasks.py b/rowers/tests/test_async_tasks.py index 4c05c1d8..ca3dad44 100644 --- a/rowers/tests/test_async_tasks.py +++ b/rowers/tests/test_async_tasks.py @@ -50,6 +50,22 @@ class AsyncTaskTests(TestCase): workouttype = 'water', ) + def tearDown(self): + for workout in self.user_workouts: + try: + os.remove(workout.csvfilename) + except (IOError,OSError,FileNotFoundError): + pass + + for filename in os.listdir('rowers/tests/testdata/temp'): + path = os.path.join('rowers/tests/testdata/temp/',filename) + if not os.path.isdir(path): + try: + os.remove(path) + except: + pass + + def test_safetimedelta(self): x = 5 y = tasks.safetimedelta(x) @@ -411,7 +427,7 @@ class AsyncTaskTests(TestCase): @patch('rowers.dataprepnodjango.getsmallrowdata_db') def test_handle_setcp(self,mocked_getsmallrowdata_db): strokesdf = pd.read_csv('rowers/tests/testdata/uhfull.csv') - filename = 'rowers/tests/temp/pq.gz' + filename = 'rowers/tests/testdata/temp/pq.gz' workoutids = 1 res = tasks.handle_setcp(strokesdf,filename,1) self.assertEqual(res,1)