Update db_test.py