Update menu_test.py

Tightened up the logic.
Merge pull request #29 from cclauss/patch-11

Update menu_test.py