Update test_database.py