164 lines
6.3 KiB
Python
164 lines
6.3 KiB
Python
#from __future__ import print_function
|
|
import pytest
|
|
|
|
pytestmark = pytest.mark.django_db
|
|
|
|
from bs4 import BeautifulSoup
|
|
import re
|
|
from nose_parameterized import parameterized
|
|
from django.test import TestCase, Client,override_settings
|
|
from django.core.management import call_command
|
|
from django.utils.six import StringIO
|
|
from django.test.client import RequestFactory
|
|
from rowers.views import checkworkoutuser,c2_open
|
|
from rowers.models import Workout, User, Rower, WorkoutForm,RowerForm,GraphImage
|
|
from rowers.forms import DocumentsForm,CNsummaryForm,RegistrationFormUniqueEmail
|
|
import rowers.plots as plots
|
|
import rowers.interactiveplots as iplots
|
|
import datetime
|
|
from rowingdata import rowingdata as rdata
|
|
from rowingdata import rower as rrower
|
|
from django.utils import timezone
|
|
from rowers.rows import handle_uploaded_file
|
|
from django.core.files.uploadedfile import SimpleUploadedFile
|
|
from time import strftime,strptime,mktime,time,daylight
|
|
import os
|
|
from rowers.tasks import handle_makeplot
|
|
from rowers.utils import serialize_list,deserialize_list
|
|
from rowers.utils import NoTokenError
|
|
from shutil import copyfile
|
|
from nose.tools import assert_true
|
|
from mock import Mock, patch
|
|
from minimocktest import MockTestCase
|
|
import pandas as pd
|
|
import rowers.c2stuff as c2stuff
|
|
|
|
import json
|
|
import numpy as np
|
|
|
|
from rowers import urls
|
|
from rowers.views import error500_view,error404_view,error400_view,error403_view
|
|
|
|
from rowers.dataprep import delete_strokedata
|
|
from rowers.tests.mocks import *
|
|
|
|
from redis import StrictRedis
|
|
redis_connection = StrictRedis()
|
|
|
|
|
|
class InteractiveChartTest(TestCase):
|
|
def setUp(self):
|
|
redis_connection.publish('tasks','KILL')
|
|
u = User.objects.create_user('john',
|
|
'sander@ds.ds',
|
|
'koeinsloot')
|
|
r = Rower.objects.create(user=u,gdproptin=True,
|
|
gdproptindate=timezone.now()
|
|
)
|
|
self.nu = datetime.datetime.now()
|
|
|
|
self.filename = 'rowers/tests/testdata/testdata.csv'
|
|
self.wotw = Workout.objects.create(name='testworkout',
|
|
workouttype='water',
|
|
user=r,date=self.nu.strftime('%Y-%m-%d'),
|
|
starttime=self.nu.strftime('%H:%M:%S'),
|
|
duration="0:55:00",distance=8000,
|
|
csvfilename=self.filename)
|
|
|
|
self.wote = Workout.objects.create(name='testworkout',
|
|
workouttype='Indoor Rower',
|
|
user=r,date=self.nu.strftime('%Y-%m-%d'),
|
|
starttime=self.nu.strftime('%H:%M:%S'),
|
|
duration="0:55:00",distance=8000,
|
|
csvfilename=self.filename)
|
|
|
|
|
|
@patch('rowers.dataprep.create_engine')
|
|
@patch('rowers.dataprep.read_df_sql')
|
|
def test_painsled(self, mocked_sqlalchemy, mocked_read_df_sql):
|
|
u = User.objects.get(username='john')
|
|
r = Rower.objects.get(user=u)
|
|
|
|
rr = rrower(hrmax=r.max,hrut2=r.ut2,
|
|
hrut1=r.ut1,hrat=r.at,
|
|
hrtr=r.tr,hran=r.an,ftp=r.ftp)
|
|
row = rdata(self.filename,rower=rr)
|
|
|
|
fig1 = plots.mkplot(row,'test')
|
|
|
|
@patch('rowers.dataprep.create_engine')
|
|
@patch('rowers.dataprep.read_df_sql')
|
|
def test_interactive_chart1(self, mocked_sqlalchemy,mocked_read_df_sql):
|
|
res = iplots.interactive_chart(self.wote.id)
|
|
|
|
@patch('rowers.dataprep.create_engine')
|
|
@patch('rowers.dataprep.read_df_sql')
|
|
def test_interactive_chart2(self, mocked_sqlalchemy,mocked_read_df_sql):
|
|
res = iplots.interactive_chart(self.wote.id,promember=1)
|
|
|
|
@patch('rowers.dataprep.create_engine')
|
|
@patch('rowers.dataprep.read_df_sql')
|
|
def test_interactive_chart3(self, mocked_sqlalchemy,mocked_read_df_sql):
|
|
res = iplots.interactive_bar_chart(self.wote.id)
|
|
|
|
@patch('rowers.dataprep.create_engine')
|
|
@patch('rowers.dataprep.read_df_sql')
|
|
def test_interactive_chart4(self, mocked_sqlalchemy,mocked_read_df_sql):
|
|
res = iplots.interactive_bar_chart(self.wote.id,promember=1)
|
|
|
|
@patch('rowers.dataprep.create_engine')
|
|
@patch('rowers.dataprep.read_df_sql')
|
|
def test_interactive_chart5(self, mocked_sqlalchemy,mocked_read_df_sql):
|
|
res = iplots.interactive_flex_chart2(self.wote.id,promember=0,
|
|
xparam='time',
|
|
yparam1='pace',yparam2='hr')
|
|
|
|
@patch('rowers.dataprep.create_engine')
|
|
@patch('rowers.dataprep.read_df_sql')
|
|
def test_interactive_chart6(self, mocked_sqlalchemy,mocked_read_df_sql):
|
|
res = iplots.interactive_flex_chart2(self.wote.id,
|
|
promember=0,xparam='distance',
|
|
yparam1='pace',yparam2='hr')
|
|
|
|
@patch('rowers.dataprep.create_engine')
|
|
@patch('rowers.dataprep.read_df_sql')
|
|
def test_interactive_chart7(self, mocked_sqlalchemy,mocked_read_df_sql):
|
|
res = iplots.interactive_flex_chart2(self.wote.id,promember=0,
|
|
xparam='time',
|
|
yparam1='pace',yparam2='spm')
|
|
|
|
@patch('rowers.dataprep.create_engine')
|
|
@patch('rowers.dataprep.read_df_sql')
|
|
def test_interactive_chart8(self, mocked_sqlalchemy,mocked_read_df_sql):
|
|
res = iplots.interactive_flex_chart2(self.wote.id,
|
|
promember=0,xparam='distance',
|
|
yparam1='pace',yparam2='spm')
|
|
|
|
@patch('rowers.dataprep.create_engine')
|
|
@patch('rowers.dataprep.read_df_sql')
|
|
def test_interactive_chart9(self, mocked_sqlalchemy,mocked_read_df_sql):
|
|
res = iplots.interactive_flex_chart2(self.wote.id,
|
|
promember=1,xparam='time',
|
|
yparam1='pace',yparam2='hr')
|
|
@patch('rowers.dataprep.create_engine')
|
|
@patch('rowers.dataprep.read_df_sql')
|
|
def test_interactive_chart10(self, mocked_sqlalchemy,mocked_read_df_sql):
|
|
res = iplots.interactive_flex_chart2(self.wote.id,
|
|
promember=1,xparam='distance',
|
|
yparam1='pace',yparam2='hr')
|
|
@patch('rowers.dataprep.create_engine')
|
|
@patch('rowers.dataprep.read_df_sql')
|
|
def test_interactive_chart11(self, mocked_sqlalchemy,mocked_read_df_sql):
|
|
res = iplots.interactive_flex_chart2(self.wote.id,
|
|
promember=1,xparam='time',
|
|
yparam1='pace',yparam2='spm')
|
|
|
|
@patch('rowers.dataprep.create_engine')
|
|
@patch('rowers.dataprep.read_df_sql')
|
|
def test_interactive_chart12(self, mocked_sqlalchemy,mocked_read_df_sql):
|
|
res = iplots.interactive_flex_chart2(self.wote.id,promember=1,
|
|
xparam='distance',
|
|
yparam1='pace',yparam2='spm')
|
|
|
|
|