some more testing coverage nuances
This commit is contained in:
@@ -2802,7 +2802,7 @@ def handle_update_wps(rid,types,ids,mode,debug=False,**kwargs):
|
||||
df = getsmallrowdata_db(['time','driveenergy'],ids=ids)
|
||||
try:
|
||||
mask = df['driveenergy'] > 100
|
||||
except (KeyError, TypeError):
|
||||
except (KeyError, TypeError): # pragma: no cover
|
||||
return 0
|
||||
wps_median = int(df.loc[mask,'driveenergy'].median())
|
||||
|
||||
|
||||
@@ -524,7 +524,8 @@ class AsyncTaskTests(TestCase):
|
||||
@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)
|
||||
self.assertTrue(result>=110)
|
||||
result = tasks.handle_update_wps(self.r.id,['water'],ids,mode='erg')
|
||||
self.assertEqual(result,110)
|
||||
self.assertTrue(result>=100)
|
||||
|
||||
Reference in New Issue
Block a user