Simplify field key assertions using set comparison

[denini08]

This PR updates some assertions in test_aggify.py to:

• Compare directly with a set instead of converting keys to a list for faster lookups O(1).
• Make the tests simpler and cleaner.
• Keep the same test logic and behavior.