Remove duplicate test
Add tests for query_entities