modify test_db.py
modify test_db.py