async def METHOD_NAME(self, *, man, storage, table):
    await man.on_rebalance(table, {TP1}, {TP1}, {TP1})
    man.storage.on_rebalance.assert_called_once_with(
        table,
        {TP1},
        {TP1},
        {TP1},
    )