Add `setup()` function to test script

Add `setup()` function to script to test Query class.