Replace more conditionals with assert_equal.
Oops.