def test_add_workout(self):
ws1 = [0, 2, 5]
rt1 = FirstRaceType(name='Marathon', distance=FirstDistance.from_string('42.195 km'))
rd1 = date(year=2017, month=7, day=29)
r1 = FirstRace(name='SFM', race_type=rt1, race_date=rd1)
rn1 = FirstRunner(name='DBD')
p1 = FirstPlan(name='My first marathon training plan', weekly_schedule=ws1, race=r1, runner=rn1)
t_warmup = FirstTime.from_string('0:15:00')
p_warmup = FirstPace.from_string('0:10:00 min per mile')
s_warmup = FirstStepBody(name='Warm up', pace=p_warmup, time=t_warmup)
s_repeat = FirstStepRepeat(name='repeat X 8', repeat=8)
d_interval = FirstDistance.from_string('400 m')
p_fast = FirstPace.from_string('0:08:00 min per mile')
s_fast = FirstStepBody(name='Fast', pace=p_fast, distance=d_interval)
s_repeat.add_step(s_fast)
s_slow = FirstStepBody(name='Rest', pace=p_warmup, distance=d_interval)
s_repeat.add_step(s_slow)
t_cooldown = FirstTime.from_string('0:10:00')
s_cooldown = FirstStepBody(name='Cool down', pace=p_warmup, time=t_cooldown)
wo = FirstWorkout(name='Week 1 Key-run 1', workout_date=date(year=2017, month=6, day=24))
wo.add_step(step=s_warmup)
wo.add_step(step=s_repeat)
wo.add_step(step=s_cooldown)
try: # first workout
p1.add_workout(workout=wo)
cmp_string = ('Training Plan:\nName - "My first marathon training plan"\n' +
'Workout days: Mon, Wed, Sat\nRace:\n' +
' Name - "SFM" of type Marathon - 42.195 km\nRunner:\n Name - "DBD"\nWorkouts:\n' +
' "Week 1 Key-run 1"\n Sat 2017-06-24\n scheduled\n' +
'Total 1 workouts\n')
self.assertEqual(cmp_string, str(p1))
file_name = 'cmp_plan2.tcx'
with open('{}/{}'.format(Config.TEST_RESOURCE_DIR, file_name), 'r') as from_file:
cmp_string = from_file.read()
self.assertEqual(cmp_string, p1.tcx())
file_name = 'cmp_plan2.json'
with open('{}/{}'.format(Config.TEST_RESOURCE_DIR, file_name), 'r') as from_file:
cmp_json = json.load(from_file)
self.assertEqual(cmp_json, p1.to_json())
file_name = 'cmp_plan2_km.json'
with open('{}/{}'.format(Config.TEST_RESOURCE_DIR, file_name), 'r') as from_file:
cmp_json = json.load(from_file)
self.assertEqual(cmp_json, p1.to_json(output_unit='km'))
file_name = 'cmp_plan2.html'
with open('{}/{}'.format(Config.TEST_RESOURCE_DIR, file_name), 'r') as from_file:
cmp_html = from_file.read()
self.assertEqual(cmp_html, p1.to_html())
file_name = 'cmp_plan2_km.html'
with open('{}/{}'.format(Config.TEST_RESOURCE_DIR, file_name), 'r') as from_file:
cmp_html = from_file.read()
self.assertEqual(cmp_html, p1.to_html(output_unit='km'))
except TypeError as ex:
self.fail(str(ex))
def test_to_string(self):
ws1 = [0, 2, 5]
ws2 = [1, 3, 6]
try: # name only
p1 = FirstPlan(name='My first marathon training plan', weekly_schedule=ws1)
self.assertEqual('Training Plan:\nName - "My first marathon training plan"\nWorkout days: Mon, Wed, Sat\n',
str(p1))
file_name = 'cmp_plan1.tcx'
with open('{}/{}'.format(Config.TEST_RESOURCE_DIR, file_name), 'r') as from_file:
cmp_string = from_file.read()
self.assertEqual(cmp_string, p1.tcx())
cmp_json = {"name": "My first marathon training plan",
"weekly_schedule": ["mon", "wed", "sat"],
"workouts": []}
self.assertEqual(cmp_json, p1.to_json())
cmp_html = ('<!DOCTYPE html>\n' +
'<html>\n' +
' <head>\n' +
' </head>\n' +
' <body>\n' +
' <h1>Training Plan: My first marathon training plan</h1>\n' +
' <div>\n' +
' <h2>\n' +
' Schedule:\n' +
' </h2>\n' +
' </div>\n' +
' </body>\n' +
'</html>')
self.assertEqual(cmp_html, p1.to_html())
except TypeError as tex:
self.fail(str(tex))
except ValueError as vex:
self.fail(str(vex))
rt1 = FirstRaceType(name='Marathon', distance=FirstDistance.from_string('42.195 km'))
rd1 = date(year=2017, month=7, day=29)
r1 = FirstRace(name='SFM', race_type=rt1, race_date=rd1)
rn1 = FirstRunner(name='DBD')
try: # all
p2 = FirstPlan(name='My first marathon training plan', weekly_schedule=ws2, race=r1, runner=rn1)
cmp_string = ('Training Plan:\nName - "My first marathon training plan"\nWorkout days: Tue, Thu, Sun\n' +
'Race:\n Name - "SFM" of type Marathon - 42.195 km\nRunner:\n Name - "DBD"\n')
self.assertEqual(cmp_string, str(p2))
except TypeError as tex:
self.fail(str(tex))
except ValueError as vex:
self.fail(str(vex))
def main():
args = process_args()
if args.output not in ['text', 'tcx', 'both']:
raise ValueError('output should be one of "text", "tcx", or "both"')
data_file_path = expanduser('~') + '/PycharmProjects/first/database/FIRSTregularPlans.xml'
data = FirstData(xml_path=data_file_path)
runner = FirstRunner(name=args.runner_name)
target_time = FirstTime.from_string(args.target_time)
if args.ref_race_type is not None:
target_time = data.equivalent_time(time_from=target_time,
race_index_from=data.race_type_index_by_name(args.ref_race_type),
race_index_to=data.race_type_index_by_name(args.race_type))
if args.race_name is not None:
race_name = args.race_name
else:
race_name = args.race_type
race_date = datetime.datetime.strptime(args.race_date, '%m/%d/%Y').date()
race = FirstRace(race_type=data.get_race_type_by_name(args.race_type),
name=race_name, race_date=race_date, target_time=target_time)
ws = get_keyrun_days(args.keyrun_days)
plan = FirstPlan(name=args.race_name, weekly_schedule=ws, race=race, runner=runner)
plan.generate_workouts(data=data)
base_file_name = str(race_date) + race_name
if args.output == 'text' or args.output == 'both':
file_name = expanduser('~/Downloads/') + base_file_name + '.txt'
target = open(file_name, 'w')
target.write(plan.details(level=3))
target.close()
if args.output == 'tcx' or args.output == 'both':
file_name = expanduser('~/Downloads/') + base_file_name + '.tcx'
target = open(file_name, 'w')
target.write(plan.tcx())
target.close()
def test_add_workout(self):
ws1 = [0, 2, 5]
rt1 = FirstRaceType(name='Marathon', distance=42.195, unit='km')
rd1 = date(year=2017, month=7, day=29)
r1 = FirstRace(name='SFM', race_type=rt1, race_date=rd1)
rn1 = FirstRunner(name='DBD')
p1 = FirstPlan(name='My first marathon training plan',
weekly_schedule=ws1,
race=r1,
runner=rn1)
t_warmup = FirstTime.from_string('0:15:00')
p_warmup = FirstPace.from_string('0:10:00 min per mile')
s_warmup = FirstStepBody(name='Warm up', pace=p_warmup, time=t_warmup)
s_repeat = FirstStepRepeat(name='repeat X 8', repeat=8)
d_interval = FirstDistance.from_string('400 m')
p_fast = FirstPace.from_string('0:08:00 min per mile')
s_fast = FirstStepBody(name='Fast', pace=p_fast, distance=d_interval)
s_repeat.add_step(s_fast)
s_slow = FirstStepBody(name='Rest', pace=p_warmup, distance=d_interval)
s_repeat.add_step(s_slow)
t_cooldown = FirstTime.from_string('0:10:00')
s_cooldown = FirstStepBody(name='Cool down',
pace=p_warmup,
time=t_cooldown)
wo = FirstWorkout(name='Week 1 Key-run 1',
workout_date=date(year=2017, month=6, day=24))
wo.add_step(step=s_warmup)
wo.add_step(step=s_repeat)
wo.add_step(step=s_cooldown)
try: # first workout
p1.add_workout(workout=wo)
cmp_string = (
'Training Plan:\nName - "My first marathon training plan"\n' +
'Workout days: Mon, Wed, Sat\nRace:\n' +
' Name - "SFM" of type Marathon - 42.195 km\nRunner:\n Name - "DBD"\nWorkouts:\n'
+ ' "Week 1 Key-run 1"\n Sat 2017-06-24\n scheduled\n' +
'Total 1 workouts\n')
self.assertEqual(cmp_string, str(p1))
file_name = expanduser(
'~/PycharmProjects/first/database/cmp_plan2.tcx')
# to_file = open(file_name, 'w')
# to_file.write(p1.tcx())
# to_file.close()
from_file = open(file_name)
cmp_string = from_file.read()
from_file.close()
self.assertEqual(cmp_string, p1.tcx())
except TypeError as ex:
self.fail(str(ex))
try: # bad workout
p1.add_workout(workout='workout')
self.fail('Should not get here with bad workout')
except TypeError as ex:
self.assertEqual(
'FirstPlan.add_workout - workout must be an instance of FirstWorkout',
str(ex))
def test_to_string(self):
ws1 = [0, 2, 5]
ws2 = [1, 3, 6]
ws3 = [2, 4, 6]
try: # name only
p1 = FirstPlan(name='My first marathon training plan',
weekly_schedule=ws1)
self.assertEqual(
'Training Plan:\nName - "My first marathon training plan"\nWorkout days: Mon, Wed, Sat\n',
str(p1))
file_name = expanduser(
'~/PycharmProjects/first/database/cmp_plan1.tcx')
# to_file = open(file_name, 'w')
# to_file.write(p1.tcx())
# to_file.close()
from_file = open(file_name)
cmp_string = from_file.read()
from_file.close()
self.assertEqual(cmp_string, p1.tcx())
except TypeError as tex:
self.fail(str(tex))
except ValueError as vex:
self.fail(str(vex))
rt1 = FirstRaceType(name='Marathon', distance=42.195, unit='km')
rd1 = date(year=2017, month=7, day=29)
r1 = FirstRace(name='SFM', race_type=rt1, race_date=rd1)
rn1 = FirstRunner(name='DBD')
try: # all
p2 = FirstPlan(name='My first marathon training plan',
weekly_schedule=ws2,
race=r1,
runner=rn1)
cmp_string = (
'Training Plan:\nName - "My first marathon training plan"\nWorkout days: Tue, Thu, Sun\n'
+
'Race:\n Name - "SFM" of type Marathon - 42.195 km\nRunner:\n Name - "DBD"\n'
)
self.assertEqual(cmp_string, str(p2))
except TypeError as tex:
self.fail(str(tex))
except ValueError as vex:
self.fail(str(vex))
try: # bad name type
p3 = FirstPlan(name=123, weekly_schedule=ws3)
self.fail('Should not get here with bad name type')
except TypeError as ex:
self.assertEqual('FirstPlan.__init__ - name must be a string',
str(ex))
try: # bad race type
p3 = FirstPlan(name='Test', weekly_schedule=ws3, race='marathon')
self.fail('Should not get here with bad race type')
except TypeError as ex:
self.assertEqual(
'FirstPlan.__init__ - race must be an instance of FirstRace',
str(ex))
try: # bad runner type
p3 = FirstPlan(name='Test', weekly_schedule=ws3, runner=345)
self.fail('Should not get here with bad runner type')
except TypeError as ex:
self.assertEqual(
'FirstPlan.__init__ - runner must be an instance of FirstRunner',
str(ex))
def test_generate_workouts(self):
data_file_path = expanduser(
'~') + '/PycharmProjects/first/database/FIRSTregularPlans.xml'
data = FirstData(xml_path=data_file_path)
ws1 = [0, 2, 5]
target_time = data.equivalent_time(
time_from=FirstTime(minutes=30),
race_index_from=data.race_type_index_by_name('5K'),
race_index_to=data.race_type_index_by_name('Marathon'))
sf_marathon = FirstRace(
race_type=data.get_race_type_by_name('Marathon'),
name='San Francisco Marathon',
race_date=date(year=2017, month=7, day=23),
target_time=target_time)
me = FirstRunner(name='Daniel BenDavid',
age=56,
gender='m',
email='*****@*****.**')
p1 = FirstPlan(name='My first marathon training plan',
weekly_schedule=ws1,
race=sf_marathon,
runner=me)
try: # positive
p1.generate_workouts(data=data)
# print p1.details(1)
self.assertEqual(48, len(p1.workouts))
wo = p1.workouts[0]
self.assertEqual('Week 1 Keyrun 1', wo.name)
self.assertEqual(3, len(wo.steps))
step = wo.steps[0]
self.assertEqual('warmup', step.name)
self.assertEqual(0, step.step_id)
self.assertEqual('time', step.get_duration_type())
self.assertEqual('0:15:00', str(step.time))
self.assertEqual('0:11:31 min per mile', str(step.pace))
step = wo.steps[1]
self.assertEqual('repeat X 3', step.name)
self.assertEqual(1, step.step_id)
self.assertEqual(3, step.repeat) # repeat
self.assertEqual(2, len(step.steps))
substep = step.steps[0]
self.assertEqual('1600m', substep.name)
self.assertEqual(2, substep.step_id)
self.assertEqual('distance', substep.get_duration_type())
self.assertEqual('1600.0 m', str(substep.distance))
self.assertEqual('0:09:26 min per mile', str(substep.pace))
substep = step.steps[1]
self.assertEqual('200 m@RI', substep.name)
self.assertEqual(3, substep.step_id)
self.assertEqual('distance', substep.get_duration_type())
self.assertEqual('200.0 m', str(substep.distance))
self.assertEqual('0:11:31 min per mile', str(substep.pace))
step = wo.steps[2]
self.assertEqual('cooldown', step.name)
self.assertEqual(4, step.step_id)
self.assertEqual('time', step.get_duration_type())
self.assertEqual('0:10:00', str(step.time))
self.assertEqual('0:11:31 min per mile', str(step.pace))
file_name = expanduser(
'~/PycharmProjects/first/database/cmp_plan_marathon.tcx')
# to_file = open(file_name, 'w')
# to_file.write(p1.tcx())
# to_file.close()
from_file = open(file_name)
cmp_string = from_file.read()
from_file.close()
self.assertEqual(cmp_string, p1.tcx())
except ValueError as vex:
self.fail(str(vex))
except TypeError as tex:
self.fail(str(tex))
ws1 = [0, 3, 6]
target_time = data.equivalent_time(
time_from=FirstTime(minutes=22, seconds=36),
race_index_from=data.race_type_index_by_name('5K'),
race_index_to=data.race_type_index_by_name('HalfMarathon'))
sf_half_marathon = FirstRace(
race_type=data.get_race_type_by_name('HalfMarathon'),
name='San Francisco Marathon',
race_date=date(year=2017, month=7, day=23),
target_time=target_time)
me = FirstRunner(name='Daniel BenDavid',
age=56,
gender='m',
email='*****@*****.**')
p2 = FirstPlan(name='San Francisco half-marathon training plan',
weekly_schedule=ws1,
race=sf_half_marathon,
runner=me)
try: # positive
p2.generate_workouts(data=data)
# print p2.details(1)
# print p2.details(2)
file_name = expanduser(
'~/PycharmProjects/first/database/cmp_plan_half_marathon.tcx')
# to_file = open(file_name, 'w')
# $ to_file.write(p2.tcx())
# to_file.close()
from_file = open(file_name)
cmp_string = from_file.read()
from_file.close()
self.assertEqual(cmp_string, p2.tcx())
except ValueError as vex:
self.fail(str(vex))
except TypeError as tex:
self.fail(str(tex))
def test_generate_workouts(self):
data = FirstData(json_path=Config.DATABASE_JSON)
ws1 = [0, 2, 5]
target_time = data.equivalent_time(time_from=FirstTime(minutes=30),
race_index_from=data.race_type_index_by_name('5K'),
race_index_to=data.race_type_index_by_name('Marathon'))
sf_marathon = FirstRace(race_type=data.get_race_type_by_name('Marathon'),
name='San Francisco Marathon',
race_date=date(year=2017, month=7, day=23),
target_time=target_time)
me = FirstRunner(name='Daniel BenDavid', age=56, gender='m', email='*****@*****.**')
p1 = FirstPlan(name='My first marathon training plan', weekly_schedule=ws1, race=sf_marathon, runner=me)
try: # positive
p1.generate_workouts(data=data)
self.assertEqual(48, len(p1.workouts))
wo = p1.workouts[0]
self.assertEqual('Week 1 Keyrun 1', wo.name)
self.assertEqual(3, len(wo.steps))
step = wo.steps[0]
self.assertEqual('warmup', step.name)
self.assertEqual(0, step.step_id)
self.assertEqual('time', step.get_duration_type())
self.assertEqual('0:15:00', str(step.time))
self.assertEqual('0:11:31 min per mile', str(step.pace))
step = wo.steps[1]
self.assertEqual('repeat X 3', step.name)
self.assertEqual(1, step.step_id)
self.assertEqual(3, step.repeat) # repeat
self.assertEqual(2, len(step.steps))
substep = step.steps[0]
self.assertEqual('1600m', substep.name)
self.assertEqual(2, substep.step_id)
self.assertEqual('distance', substep.get_duration_type())
self.assertEqual('1600.0 m', str(substep.distance))
self.assertEqual('0:09:26 min per mile', str(substep.pace))
substep = step.steps[1]
self.assertEqual('200 m@RI', substep.name)
self.assertEqual(3, substep.step_id)
self.assertEqual('distance', substep.get_duration_type())
self.assertEqual('200.0 m', str(substep.distance))
self.assertEqual('0:11:31 min per mile', str(substep.pace))
step = wo.steps[2]
self.assertEqual('cooldown', step.name)
self.assertEqual(4, step.step_id)
self.assertEqual('time', step.get_duration_type())
self.assertEqual('0:10:00', str(step.time))
self.assertEqual('0:11:31 min per mile', str(step.pace))
file_name = 'cmp_plan_marathon.tcx'
with open('{}/{}'.format(Config.TEST_RESOURCE_DIR, file_name), 'r') as from_file:
cmp_string = from_file.read()
self.assertEqual(cmp_string, p1.tcx())
file_name = 'cmp_plan_marathon.json'
with open('{}/{}'.format(Config.TEST_RESOURCE_DIR, file_name), 'r') as from_file:
cmp_json = json.load(from_file)
self.assertEqual(cmp_json, p1.to_json())
file_name = 'cmp_plan_marathon.html'
with open('{}/{}'.format(Config.TEST_RESOURCE_DIR, file_name), 'r') as from_file:
cmp_html = from_file.read()
self.assertEqual(cmp_html, p1.to_html())
except ValueError as vex:
self.fail(str(vex))
except TypeError as tex:
self.fail(str(tex))
ws1 = [0, 3, 6]
target_time = data.equivalent_time(time_from=FirstTime(minutes=22, seconds=36),
race_index_from=data.race_type_index_by_name('5K'),
race_index_to=data.race_type_index_by_name('HalfMarathon'))
sf_half_marathon = FirstRace(race_type=data.get_race_type_by_name('HalfMarathon'),
name='San Francisco Marathon',
race_date=date(year=2017, month=7, day=23),
target_time=target_time)
me = FirstRunner(name='Daniel BenDavid', age=56, gender='m', email='*****@*****.**')
p2 = FirstPlan(name='San Francisco half-marathon training plan', weekly_schedule=ws1,
race=sf_half_marathon, runner=me)
try: # positive
p2.generate_workouts(data=data)
file_name = 'cmp_plan_half_marathon.tcx'
from_file = open('{}/{}'.format(Config.TEST_RESOURCE_DIR, file_name))
cmp_string = from_file.read()
from_file.close()
self.assertEqual(cmp_string, p2.tcx())
except ValueError as vex:
self.fail(str(vex))
except TypeError as tex:
self.fail(str(tex))
def main():
args = process_args()
if 'text' not in args.output and 'tcx' not in args.output and \
'json' not in args.output and 'html' not in args.output:
raise ValueError(
'Unknown output formats (). Available text, tcx, html, and json'.
format(args.output))
data = FirstData(json_path=Config.DATABASE_JSON)
runner = FirstRunner(name=args.runner_name)
target_time = FirstTime.from_string(string=args.target_time)
if args.ref_race_type is not None:
target_time = data.equivalent_time(
time_from=target_time,
race_index_from=data.race_type_index_by_name(
name=args.ref_race_type),
race_index_to=data.race_type_index_by_name(name=args.race_type))
if args.race_name is not None:
race_name = args.race_name
else:
race_name = args.race_type
race_date = datetime.datetime.strptime(args.race_date, '%m/%d/%Y').date()
race = FirstRace(race_type=data.get_race_type_by_name(name=args.race_type),
name=race_name,
race_date=race_date,
target_time=target_time)
ws = get_keyrun_days(user_string=args.keyrun_days)
plan = FirstPlan(name=args.race_name,
weekly_schedule=ws,
race=race,
runner=runner)
plan.generate_workouts(data=data)
base_file_name = str(race_date) + race_name
if 'text' in args.output:
file_name = '{}/{}.{}'.format(Config.DOWNLOADS_DIR, base_file_name,
'txt')
target = open(file_name, 'w')
target.write(plan.details(level=3))
target.close()
if 'tcx' in args.output:
file_name = '{}/{}.{}'.format(Config.DOWNLOADS_DIR, base_file_name,
'tcx')
target = open(file_name, 'w')
target.write(plan.tcx())
target.close()
if 'json' in args.output:
file_name = '{}/{}.{}'.format(Config.DOWNLOADS_DIR, base_file_name,
'json')
target = open(file_name, 'w')
target.write(json.dumps(plan.to_json(args.length_unit)))
target.close()
if 'html' in args.output:
file_name = '{}/{}.{}'.format(Config.DOWNLOADS_DIR, base_file_name,
'html')
target = open(file_name, 'w')
target.write(plan.to_html(args.length_unit))
target.close()