Update query_test.py

simple edit to test / sample code