TST: Simplify this test somewhat.