punishment in the grave is actually a disputed matter, as per the qur'an chapter 36:
"They will not (have to) wait for aught but a single Blast: it will seize them while they are yet disputing among themselves!
No (chance) will they then have, by will, to dispose (of their affairs), nor to return to their own people!
The trumpet shall be sounded, when behold! from the sepulchers (men) will rush forth to their Lord!
They will say: "Ah! Woe unto us! Who hath raised us up from our beds of repose?"... (A voice will say: ) "This is what (Allah) Most Gracious had promised. And true was the word of the messengers!"
Then, on that Day, not a soul will be wronged in the least, and ye shall but be repaid the meeds of your past Deeds."
notice that they are described here as surprised and shocked at having been resurrected, and at the prophets' warnings manifest in front of their eyes, if there were indeed anything happening to dead people other than being dead they would probably know what was happening at the day of judgment.
there some ahadeeth supposedly, but if anything is in opposition with what the qur'an states it is not the prophet's saying.