Update test code from adios-query