108 lines
3.2 KiB
Python
108 lines
3.2 KiB
Python
from __future__ import absolute_import
|
|
from __future__ import division
|
|
from __future__ import print_function
|
|
from __future__ import unicode_literals
|
|
|
|
from .statements import *
|
|
|
|
import rowers.dataprep as dataprep
|
|
|
|
nu = datetime.datetime.now()
|
|
|
|
from django.core.files import File
|
|
from django.utils.six import BytesIO
|
|
|
|
from PIL import Image
|
|
from io import StringIO
|
|
|
|
@override_settings(TESTING=True)
|
|
class ForceUnits(TestCase):
|
|
def setUp(self):
|
|
redis_connection.publish('tasks','KILL')
|
|
self.c = Client()
|
|
|
|
self.u = UserFactory()
|
|
|
|
self.r = Rower.objects.create(user=self.u,
|
|
birthdate=faker.profile()['birthdate'],
|
|
gdproptin=True,
|
|
gdproptindate=timezone.now(),
|
|
rowerplan='coach')
|
|
|
|
|
|
self.factory = RequestFactory()
|
|
self.password = faker.word()
|
|
self.u.set_password(self.password)
|
|
self.u.save()
|
|
|
|
def test_upload_painsled_lbs(self):
|
|
login = self.c.login(username=self.u.username, password=self.password)
|
|
self.assertTrue(login)
|
|
|
|
filename = 'rowers/tests/testdata/PainsledForce.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,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)
|