coverage related changes

2021-04-26 17:48:22 +02:00
parent 7626554ba9
commit 9e2a97e721
17 changed files with 1534 additions and 144 deletions
--- a/rowers/tests/test_courses.py
+++ b/rowers/tests/test_courses.py
@@ -4,6 +4,122 @@ from __future__ import print_function
 from __future__ import unicode_literals

 from .statements import *
+import rowers.courses as courses
+import rowers.dataprep as dataprep
+from rowers.courseutils import *
+from rowingdata import rowingdata as rdata
+from rowers.models import polygon_to_path
+
+class CourseUnitTest(TestCase):
+    def setUp(self):
+        self.c = Client()
+        self.u = User.objects.create_user('john',
+                                     'sander@ds.ds',
+                                     'koeinsloot')
+        self.r = Rower.objects.create(user=self.u,gdproptin=True,surveydone=True,
+                                      gdproptindate=timezone.now(),
+                                      rowerplan='coach',
+        )
+        self.nu = datetime.datetime.now()
+
+        cs = courses.kmltocourse('rowers/tests/testdata/thyro.kml')
+        course = cs[0]
+        cname = course['name']
+        cnotes = course['description']
+        self.polygons = course['polygons']
+        pstart = self.polygons[0]
+        self.ThyroBaantje = courses.createcourse(self.r,cname,self.polygons,notes=cnotes)
+        self.start = GeoPolygon.objects.filter(course=self.ThyroBaantje,order_in_course=0)[0]
+        self.ThyroBaantje.save()
+
+        result = get_random_file(filename='rowers/tests/testdata/thyro.csv')
+        self.wthyro = WorkoutFactory(user=self.r,
+                                 csvfilename=result['filename'],
+                                 starttime=result['starttime'],
+                                 startdatetime=result['startdatetime'],
+                                 duration=result['duration'],
+                                 distance=result['totaldist'],
+                                 workouttype = 'water',
+                                 )
+
+        self.wthyro.startdatetime = arrow.get(self.nu).datetime
+        self.wthyro.date = self.nu.date()
+        self.wthyro.save()
+
+    def test_time_in_path(self):
+        row = rdata(csvfile='rowers/tests/testdata/thyro.csv')
+
+        time = row.df['TimeStamp (sec)']
+        lat = row.df[' latitude']
+        lon = row.df[' longitude']
+        cum_dist = row.df['cum_dist']
+
+        data = pd.DataFrame(
+            {
+                'time':time,
+                'latitude':lat,
+                'longitude':lon,
+                'cum_dist':cum_dist,
+                }
+        )
+
+        startpath = polygon_to_path(self.start)
+
+        mintime,mindist = time_in_path(data,startpath)
+        self.assertEqual(mintime,78)
+        self.assertEqual(mindist,207.1)
+
+    def test_coursetime_first(self):
+        row = rdata(csvfile='rowers/tests/testdata/thyro.csv')
+
+        time = row.df['TimeStamp (sec)']
+        lat = row.df[' latitude']
+        lon = row.df[' longitude']
+        cum_dist = row.df['cum_dist']
+
+        data = pd.DataFrame(
+            {
+                'time':time,
+                'latitude':lat,
+                'longitude':lon,
+                'cum_dist':cum_dist,
+                }
+        )
+        paths = []
+        polygons = GeoPolygon.objects.filter(course=self.ThyroBaantje).order_by("order_in_course")
+        for p in polygons:
+            paths.append(polygon_to_path(p))
+
+        entrytime,entrydistance,coursecompleted = coursetime_first(data,paths)
+        self.assertEqual(entrytime,78)
+        self.assertEqual(entrydistance,207.1)
+        self.assertTrue(coursecompleted)
+
+    def test_coursetime_paths(self):
+        row = rdata(csvfile='rowers/tests/testdata/thyro.csv')
+
+        time = row.df['TimeStamp (sec)']
+        lat = row.df[' latitude']
+        lon = row.df[' longitude']
+        cum_dist = row.df['cum_dist']
+
+        data = pd.DataFrame(
+            {
+                'time':time,
+                'latitude':lat,
+                'longitude':lon,
+                'cum_dist':cum_dist,
+                }
+        )
+        paths = []
+        polygons = GeoPolygon.objects.filter(course=self.ThyroBaantje).order_by("order_in_course")
+        for p in polygons:
+            paths.append(polygon_to_path(p))
+
+        entrytime,entrydistance,coursecompleted = coursetime_paths(data,paths)
+        self.assertEqual(entrytime,435)
+        self.assertEqual(entrydistance,1348.8)
+        self.assertTrue(coursecompleted)

 class CoursesTest(TestCase):
    def setUp(self):