Update menu_test.py

Tightened up the logic.