62 lines
2.2 KiB
Python
62 lines
2.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 *
|
|
|
|
|
|
nu = datetime.datetime.now()
|
|
|
|
# interactive plots
|
|
from rowers import interactiveplots
|
|
|
|
class InteractivePlotTests(TestCase):
|
|
def setUp(self):
|
|
self.u = UserFactory()
|
|
|
|
self.r = Rower.objects.create(user=self.u,
|
|
birthdate=faker.profile()['birthdate'],
|
|
gdproptin=True,surveydone=True,
|
|
gdproptindate=timezone.now(),
|
|
rowerplan='coach')
|
|
|
|
self.c = Client()
|
|
self.user_workouts = WorkoutFactory.create_batch(5, user=self.r)
|
|
self.factory = RequestFactory()
|
|
self.password = faker.word()
|
|
self.u.set_password(self.password)
|
|
self.u.save()
|
|
|
|
def test_interactive_hr_piechart(self):
|
|
df = pd.read_csv('rowers/tests/testdata/getrowdata_mock.csv')
|
|
|
|
script, div = interactiveplots.interactive_hr_piechart(df, self.r,'')
|
|
self.assertFalse(len(script)==0)
|
|
self.assertFalse(len(div)==0)
|
|
|
|
def test_interactive_workouttype_piechart(self):
|
|
workouts = Workout.objects.filter(user=self.r)
|
|
script, div = interactiveplots.interactive_workouttype_piechart(workouts)
|
|
self.assertFalse(len(script)==0)
|
|
self.assertFalse(len(div)==0)
|
|
|
|
def test_interactive_boxchart(self):
|
|
df = pd.read_csv('rowers/tests/testdata/boxplotdata.csv')
|
|
print(df.info())
|
|
|
|
script, div = interactiveplots.interactive_boxchart(df, 'spm')
|
|
|
|
self.assertFalse(len(script)==0)
|
|
self.assertFalse(len(div)==0)
|
|
|
|
def test_interactive_activitychart(self):
|
|
workouts = Workout.objects.filter(user=self.r).order_by('date')
|
|
workouts2 = Workout.objects.filter(user=self.r).order_by('-date')
|
|
|
|
startdate = workouts[0].date-datetime.timedelta(days=5)
|
|
enddate = workouts2[0].date+datetime.timedelta(days=5)
|
|
script, div = interactiveplots.interactive_activitychart(workouts,startdate,enddate)
|
|
self.assertFalse(len(script)==0)
|
|
self.assertFalse(len(div)==0)
|