First of all, thank you for your amazing contribution!
I would like to suggest considering one of my work that studies the expressive power of graph pooling: https://arxiv.org/abs/2102.11533 (ICLR 2021), showing that the proposed pooling operation is as powerful as the 1-WL test, which seems relevant to your work.