Dictionary lookups exercise (PyBite #109)

[Drone4four]

Based on your feedback, @ibreeden, I’ve rewritten my script from scratch. In my latest iteration, I’ve done a few things. I've:

• wrapped my algorithm inside a function and did away with the dictionary loop
  
• replaced lowercase-ing of the dictionary keys and am now using capitalize()
  

Here is my script now:

INVALID_DAY = 'Not a valid day'
REST = 'Rest'
CHILL_OUT = 'Chill out!'
TRAIN = 'Go train {}'
WORKOUT_SCHEDULE = {
   'Friday': 'Shoulders',
   'Monday': 'Chest+biceps',
   'Saturday': 'Rest',
   'Sunday': 'Rest',
   'Thursday': 'Legs',
   'Tuesday': 'Back+triceps',
   'Wednesday': 'Core'
   }

def get_workout_motd(day):
   if WORKOUT_SCHEDULE[day.capitalize()] == ('Monday', 'Tuesday', 'Wednesday', 'Thursday', 'Friday'):
       return TRAIN
   elif WORKOUT_SCHEDULE[day.capitalize()] == ('Saturday','Sunday'):
       return (REST, CHILL_OUT)
   else:
       return INVALID_DAY

Here is my test script:

import pytest
from workouts1 import (get_workout_motd,
                     CHILL_OUT, INVALID_DAY)


# About parametrize: https://pybit.es/pytest-coding-100-tests.html
@pytest.mark.parametrize("day, expected", [
   ('Monday', 'Go train Chest+biceps'),
   ('monday', 'Go train Chest+biceps'),
   ('Tuesday', 'Go train Back+triceps'),
   ('TuEsdAy', 'Go train Back+triceps'),
   ('Wednesday', 'Go train Core'),
   ('wednesdaY', 'Go train Core'),
   ('Thursday', 'Go train Legs'),
   ('Friday', 'Go train Shoulders'),
   ('Saturday', CHILL_OUT),
   ('Sunday', CHILL_OUT),
   ('sundAy', CHILL_OUT),
   ('nonsense', INVALID_DAY),
   ('monday2', INVALID_DAY),
])
def test_get_workout_valid_case_insensitive_dict_lookups(day, expected):
   assert get_workout_motd(day) == expected

When I run that test script, here is my traceback: https://pastebin.com/kjt6GjjT

I put my traceback inside a pastebin because it is verbose.

As you can see in my traceback, everything fails with a KeyError. I am way off. All I understand is that my WORKOUT_SCHEDULE dictionary splicing is malformed.

What would you people suggest I try next? Any hints?

This is not homework for a course, but I am pretending that it is. So my ask is that you people provide a series of hints and general guidance rather than a full answer.

Guides I’ve leveraged this time:

• "string capitalize() in Python"
  
• "Python String capitalize()"