Add assert_raises