Update test_db.py