TST Use numpy.testing instead of raw asserts