Update tests.py

Added retrieve restroom eq