Update relationship_tests.py