Add tests for set operations