added test for existing files

rowers/tests/test_units.py

@@ -6,6 +6,7 @@ from __future__ import unicode_literals
 from .statements import *
 
 import rowers.dataprep as dataprep
+from rowingdata import rowingdata
 
 nu = datetime.datetime.now()
 
@@ -35,6 +36,9 @@ class ForceUnits(TestCase):
         self.u.set_password(self.password)
         self.u.save()
 
+    def tearDown(self):
+        dataprep.delete_strokedata(1)
+
     def test_upload_painsled_lbs(self):
         login = self.c.login(username=self.u.username, password=self.password)
         self.assertTrue(login)
@@ -116,3 +120,52 @@ class ForceUnits(TestCase):
         df = dataprep.getsmallrowdata_db(['averageforce'],ids=[1])
         average_N = int(df['averageforce'].mean())
         self.assertEqual(average_N,267)
+
+@override_settings(TESTING=True)
+class TestForceUnit(TestCase):
+    def setUp(self):
+        self.u = UserFactory()
+
+        self.r = Rower.objects.create(user=self.u,
+                                      birthdate=faker.profile()['birthdate'],
+                                      gdproptin=True,
+                                      gdproptindate=timezone.now(),
+                                      rowerplan='coach')
+
+        self.c = Client()
+        self.factory = RequestFactory()
+        self.password = faker.word()
+        self.u.set_password(self.password)
+        self.u.save()
+
+        result = get_random_file(filename='rowers/tests/testdata/PainsledForce.csv')
+        self.csvfilename = result['filename']
+
+        self.wpainsled = WorkoutFactory(user=self.r,
+                                     csvfilename=result['filename'],
+                                     starttime=result['starttime'],
+                                     startdatetime=result['startdatetime'],
+                                     duration=result['duration'],
+                                     distance=result['totaldist'],
+                                     workouttype = 'rower',
+        )
+
+    def tearDown(self):
+        w = Workout.objects.filter(csvfilename=self.csvfilename)
+        id = w[0].id
+        dataprep.delete_strokedata(id)
+
+    def test_read_painsled_lbs(self):
+        login = self.c.login(username=self.u.username, password=self.password)
+        self.assertTrue(login)
+
+        w = Workout.objects.filter(csvfilename=self.csvfilename)
+        id = w[0].id
+        row = rowingdata(csvfile=w[0].csvfilename)
+        average_N = int(row.df[' AverageDriveForce (N)'].mean())
+        self.assertEqual(average_N,398)
+
+
+        df = dataprep.getsmallrowdata_db(['averageforce'],ids=[id])
+        average_N = int(df['averageforce'].mean())
+        self.assertEqual(average_N,399)