Update query_tests.py