#[pyfunction]
fn report_unraisable(py: Python) {
    use pyo3::exceptions::PyRuntimeError;
    let err = PyRuntimeError::new_err("foo");
    err.write_unraisable(py, None);

    let err = PyRuntimeError::new_err("bar");
    err.write_unraisable(py, Some(py.NotImplemented().as_ref(py)));
}

#[test]
fn test_write_unraisable() {
    Python::with_gil(|py| {
        let report_unraisable = wrap_pyfunction!(report_unraisable, py).unwrap();
        let locals = PyDict::new(py);
        locals
            .set_item("report_unraisable", report_unraisable)
            .unwrap();

        py_run!(
            py,
            report_unraisable,
            r#"
                import sys

                captured = []
                def report(data):
                    captured.append(list(data))

                original_hook = sys.unraisablehook
                try:
                    sys.unraisablehook = report
                    report_unraisable()

                    assert len(captured) == 2

                    assert captured[0][0] is RuntimeError
                    assert str(captured[0][1]) == 'foo'
                    assert captured[0][4] is None

                    assert captured[1][0] is RuntimeError
                    assert str(captured[1][1]) == 'bar'
                    assert captured[1][4] is NotImplemented
                finally:
                    sys.unraisablehook = original_hook
            "#
        );
    });
}