Update database_test.py