query_tests: test uda with multiple columns

Exposes bug hinted at https://github.com/uwescience/raco/pull/325/files#r17342664