diff --git a/rowers/tests/test_units.py b/rowers/tests/test_units.py index 82298a0f..dabd31c6 100644 --- a/rowers/tests/test_units.py +++ b/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)