Added some more test_probe tests.

Sorry @tbekolay, I couldn't quite sneak this into the last PR.