MAINT: test_nonlin - use assert_ instead of assert