adding some tests

rowers/dataprep.py

@@ -135,7 +135,6 @@ from scipy.signal import savgol_filter
 
 import datetime
 
-
 def get_video_data(w,groups=['basic'],mode='water'):
     modes = [mode,'both','basic']
     columns = ['time','velo','spm']
@@ -1292,7 +1291,7 @@ def update_wps(r,types,mode='water',asynchron=True):
 
     if mode == 'water':
         r.median_wps = wps_median
-    else:
+    else:# pragma: no cover
         r.median_wps_erg = wps_median
 
     r.save()

rowers/tests/mocks.py

@@ -350,6 +350,11 @@ def mocked_getsmallrowdata_db_water(*args, **kwargs):
 
     return df
 
+def mocked_getsmallrowdata_db_wps(*args, **kwargs):
+    df = pd.read_csv('rowers/tests/testdata/driveenergies.csv')
+
+    return df
+
 def mocked_getpowerdata_db(*args, **kwargs):
     df = pd.read_csv('rowers/tests/testdata/fake_powerdata.csv')
 

rowers/tests/test_async_tasks.py

@@ -510,7 +510,7 @@ class AsyncTaskTests(TestCase):
         self.assertEqual(res,1)
 
     @patch('rowers.dataprepnodjango.getsmallrowdata_db')
-    def test_handle_setcp(self,mocked_getsmallrowdata_db_db_setcp):
+    def test_handle_setcp(self,mocked_getsmallrowdata_db_setcp):
         strokesdf = pd.read_csv('rowers/tests/testdata/uhfull.csv')
         filename = 'rowers/tests/testdata/temp/pq.gz'
         workoutids = 1
@@ -520,3 +520,11 @@ class AsyncTaskTests(TestCase):
             os.remove(filename)
         except FileNotFoundError:
             pass
+
+    @patch('rowers.dataprepnodjango.getsmallrowdata_db')
+    def test_handle_update_wps(self,mocked_getsmallrowdata_db_wps):
+      ids = [1,2,3]
+      result = tasks.handle_update_wps(self.r.id,['water'],ids,mode='water')
+      self.assertEqual(result,110)
+      result = tasks.handle_update_wps(self.r.id,['water'],ids,mode='erg')
+      self.assertEqual(result,110)