ENH: use assert_ instead of assert_true

Removes need for extra import.