Add more tests for coincidence_queries

Clean up imports