adding test for NK SpeedCoach Empower

rowers/tests/test_units.py

@@ -39,7 +39,7 @@ class ForceUnits(TestCase):
         login = self.c.login(username=self.u.username, password=self.password)
         self.assertTrue(login)
 
-        filename = 'rowers/tests/testdata/testdata.csv'
+        filename = 'rowers/tests/testdata/PainsledForce.csv'
         f = open(filename,'rb')
         file_data = {'file': f}
         form_data = {
@@ -68,4 +68,40 @@ class ForceUnits(TestCase):
 
         df = dataprep.getsmallrowdata_db(['averageforce'],ids=[1])
         average_N = int(df['averageforce'].mean())
-        self.assertEqual(average_N,398)
+        self.assertEqual(average_N,399)
+
+
+    def test_upload_speedcoach_N(self):
+        login = self.c.login(username=self.u.username, password=self.password)
+        self.assertTrue(login)
+
+        filename = 'rowers/tests/testdata/EmpowerSpeedCoachForce.csv'
+        f = open(filename,'rb')
+        file_data = {'file': f}
+        form_data = {
+        'title':'test',
+        'workouttype':'rower',
+        'boattype':'1x',
+        'notes':'aap noot mies',
+        'make_plot':False,
+        'upload_to_c2':False,
+        'plottype':'timeplot',
+        'file': f,
+        }
+
+        form = DocumentsForm(form_data,file_data)
+        response = self.c.post('/rowers/workout/upload/', form_data, follow=True)
+
+        self.assertRedirects(response, expected_url='/rowers/workout/'+encoded1+'/edit/',
+                             status_code=302,target_status_code=200)
+
+        self.assertEqual(response.status_code, 200)
+
+        f.close()
+
+        w = Workout.objects.get(id=1)
+        self.assertEqual(w.forceunit,'lbs')
+
+        df = dataprep.getsmallrowdata_db(['averageforce'],ids=[1])
+        average_N = int(df['averageforce'].mean())
+        self.assertEqual(average_N,267)