import assert instead of expect