Restore the tests I accidentally removed.

Should've checked the upstream repo for updates before copy-pasting my own changes.