including age records in tests

rowers/tests/test_cpchart.py

@@ -69,6 +69,8 @@ class CPChartTest(TestCase):
                                       gdproptindate=timezone.now(),
                                       rowerplan='coach')
 
+        r = self.u.rower
+        age = calculate_age(r.birthdate)
         self.c = Client()
         self.user_workouts = WorkoutFactory.create_batch(5, user=self.r)
         self.factory = RequestFactory()
@@ -91,6 +93,21 @@ class CPChartTest(TestCase):
             )
 
             record.save()
+
+        perfsdf = pd.read_csv('rowers/tests/testdata/calcageperformance.csv')
+
+        r = self.u.rower
+        
+        for i in range(len(perfsdf)):
+            perf = CalcAgePerformance(
+                age = age,
+                duration = perfsdf.ix[i,'duration'],
+                power = perfsdf.ix[i,'power'],
+                sex = r.sex,
+                weightcategory = r.weightcategory
+                )
+            perf.save()
+            
             
     def test_analytics_page(self):
         login = self.c.login(username=self.u.username,password=self.password)
@@ -179,7 +196,7 @@ class CPChartTest(TestCase):
         self.assertEqual(result,1)
         
         
-    @patch('rowers.dataprep.fetchcperg')
+    @patch('rowers.dataprep.fetchcperg', side_effect = mocked_fetchcperg)
     @patch('rowers.dataprep.create_engine')
     def test_rankingpieces(self, mocked_fetchcperg, mocked_sqlalchemy):
         url = '/rowers/ote-bests2/'