pure/

lib.rs

#![allow(deprecated)]

mod chrono_types;
mod custom_exceptions;
mod float_values;
mod ip_types;
mod manual_overloading;
mod manual_submit;
mod overloading;
mod overriding;
mod rust_type_marker;
mod skip_stub_type_test;
mod time_types;

use chrono_types::*;
use custom_exceptions::*;
use float_values::*;
use ip_types::*;
use manual_overloading::*;
use manual_submit::*;
use overloading::*;
use overriding::*;
use rust_type_marker::*;
use skip_stub_type_test::*;
use time_types::*;

#[cfg_attr(target_os = "macos", doc = include_str!("../../../README.md"))]
mod readme {}

use ahash::RandomState;
use pyo3::{prelude::*, types::*};
use pyo3_stub_gen::{
    define_stub_info_gatherer, derive::*, module_doc, module_variable,
    runtime::PyModuleTypeAliasExt, type_alias,
};
use rust_decimal::Decimal;
use std::{collections::HashMap, path::PathBuf};

/// Returns the sum of two numbers as a string.
#[gen_stub_pyfunction]
#[pyfunction]
fn sum(v: Vec<u32>) -> u32 {
    v.iter().sum()
}

#[gen_stub_pyfunction]
#[pyfunction]
fn read_dict(dict: HashMap<usize, HashMap<usize, usize>>) {
    for (k, v) in dict {
        for (k2, v2) in v {
            println!("{k} {k2} {v2}");
        }
    }
}

#[gen_stub_pyfunction]
#[pyfunction]
fn create_dict(n: usize) -> HashMap<usize, Vec<usize>> {
    let mut dict = HashMap::new();
    for i in 0..n {
        dict.insert(i, (0..i).collect());
    }
    dict
}

/// Add two decimal numbers with high precision
#[gen_stub_pyfunction]
#[pyfunction]
fn add_decimals(a: Decimal, b: Decimal) -> Decimal {
    a + b
}

#[gen_stub_pyclass]
#[pyclass(extends=PyDate)]
struct MyDate;

#[gen_stub_pyclass]
#[pyclass(subclass)]
#[derive(Debug)]
struct A {
    #[gen_stub(default = A::default().x)]
    #[pyo3(get, set)]
    x: usize,

    #[pyo3(get)]
    y: usize,
}

impl Default for A {
    fn default() -> Self {
        Self { x: 2, y: 10 }
    }
}

#[gen_stub_pymethods]
#[pymethods]
impl A {
    /// This is a constructor of :class:`A`.
    #[new]
    fn new(x: usize) -> Self {
        Self { x, y: 10 }
    }
    /// class attribute NUM1
    #[classattr]
    #[pyo3(name = "NUM")]
    const NUM1: usize = 2;

    /// deprecated class attribute NUM3 (will show warning)
    #[deprecated(since = "1.0.0", note = "This constant is deprecated")]
    #[classattr]
    const NUM3: usize = 3;
    /// class attribute NUM2
    #[expect(non_snake_case)]
    #[classattr]
    fn NUM2() -> usize {
        2
    }
    #[classmethod]
    fn classmethod_test1(cls: &Bound<'_, PyType>) {
        _ = cls;
    }

    #[deprecated(since = "1.0.0", note = "This classmethod is deprecated")]
    #[classmethod]
    fn deprecated_classmethod(cls: &Bound<'_, PyType>) {
        _ = cls;
    }

    #[classmethod]
    fn classmethod_test2(_: &Bound<'_, PyType>) {}

    // Qualified-path classmethod receiver: `cls` must still be detected
    // as a receiver when the user writes `pyo3::types::PyType` instead of
    // the unqualified `PyType`.
    #[classmethod]
    fn classmethod_test_qualified(cls: &Bound<'_, pyo3::types::PyType>) {
        _ = cls;
    }

    fn show_x(&self) {
        println!("x = {}", self.x);
    }

    // Test cases: every self-receiver shape `#[pymethods]` accepts
    // should be rendered as `self` in the generated stub.
    fn show_x_pyref(slf: PyRef<'_, Self>) -> usize {
        slf.x
    }

    fn show_x_pyrefmut(slf: PyRefMut<'_, Self>) -> usize {
        slf.x
    }

    fn show_x_bound(slf: Bound<'_, Self>) -> PyResult<usize> {
        Ok(slf.borrow().x)
    }

    fn show_x_bound_ref(slf: &Bound<'_, Self>) -> PyResult<usize> {
        Ok(slf.borrow().x)
    }

    fn show_x_py(slf: Py<Self>, py: Python<'_>) -> PyResult<usize> {
        Ok(slf.borrow(py).x)
    }

    fn ref_test<'a>(&self, x: Bound<'a, PyDict>) -> Bound<'a, PyDict> {
        x
    }

    async fn async_get_x(&self) -> usize {
        self.x
    }

    #[gen_stub(skip)]
    fn need_skip(&self) {}

    #[deprecated(since = "1.0.0", note = "This method is deprecated")]
    fn deprecated_method(&self) {
        println!("This method is deprecated");
    }

    #[deprecated(since = "1.0.0", note = "This method is deprecated")]
    #[getter]
    fn deprecated_getter(&self) -> usize {
        self.x
    }

    #[deprecated(since = "1.0.0", note = "This setter is deprecated")]
    #[setter]
    fn set_y(&mut self, value: usize) {
        self.y = value;
    }

    #[deprecated(since = "1.0.0", note = "This staticmethod is deprecated")]
    #[staticmethod]
    fn deprecated_staticmethod() -> usize {
        42
    }

    #[getter]
    /// Always returns `42`.
    fn forty_two(&self) -> i32 {
        42
    }

    // Test case: #[pyo3(name = "...")] should override the function name for getter
    #[getter]
    #[pyo3(name = "renamed_getter")]
    fn py_renamed_getter(&self) -> i32 {
        100
    }
}

/// Test that setter stubs use `type_input` (e.g. `Sequence`) while getter stubs use `type_output` (e.g. `list`).
///
/// For `Vec<i32>`:
/// - getter should produce `-> list[int]`
/// - setter should produce `value: Sequence[int]`
#[gen_stub_pyclass]
#[pyclass]
struct GetterSetterTypeTest {
    values: Vec<i32>,
}

#[gen_stub_pymethods]
#[pymethods]
impl GetterSetterTypeTest {
    #[new]
    fn new(values: Vec<i32>) -> Self {
        Self { values }
    }

    #[getter]
    fn values(&self) -> Vec<i32> {
        self.values.clone()
    }

    #[setter]
    fn set_values(&mut self, values: Vec<i32>) {
        self.values = values;
    }
}

#[gen_stub_pyfunction]
#[pyfunction]
#[pyo3(signature = (x = 2))]
fn create_a(x: usize) -> A {
    A { x, y: 10 }
}

// Negative regression tests: free `#[pyfunction]`s whose first argument
// is `Bound<'_, A>` / `&Bound<'_, A>` / `Py<A>` must NOT be mistaken for
// self receivers — only the literal `Self` spelling is a receiver.
#[gen_stub_pyfunction]
#[pyfunction]
fn echo_a_bound_ref<'py>(a: &Bound<'py, A>) -> Bound<'py, A> {
    a.clone()
}

#[gen_stub_pyfunction]
#[pyfunction]
fn echo_a_bound(a: Bound<'_, A>) -> Bound<'_, A> {
    a
}

#[gen_stub_pyfunction]
#[pyfunction]
fn echo_a_py(a: Py<A>) -> Py<A> {
    a
}

#[gen_stub_pyclass]
#[pyclass(extends=A)]
#[derive(Debug)]
struct B;

/// `C` only impl `FromPyObject`
#[derive(Debug)]
struct C {
    x: usize,
}
#[gen_stub_pyfunction]
#[pyfunction(signature = (c=None))]
fn print_c(c: Option<C>) {
    if let Some(c) = c {
        println!("{}", c.x);
    } else {
        println!("None");
    }
}
impl FromPyObject<'_, '_> for C {
    type Error = PyErr;

    fn extract(ob: Borrowed<'_, '_, PyAny>) -> std::result::Result<Self, Self::Error> {
        Ok(C { x: ob.extract()? })
    }
}
impl pyo3_stub_gen::PyStubType for C {
    fn type_output() -> pyo3_stub_gen::TypeInfo {
        usize::type_output()
    }
}
impl pyo3_stub_gen::runtime::PyRuntimeType for C {
    fn runtime_type_object(
        py: ::pyo3::Python<'_>,
    ) -> ::pyo3::PyResult<::pyo3::Bound<'_, ::pyo3::PyAny>> {
        usize::runtime_type_object(py)
    }
}

/// Returns the length of the string.
#[gen_stub_pyfunction]
#[pyfunction]
fn str_len(x: &str) -> PyResult<usize> {
    Ok(x.len())
}

#[gen_stub_pyfunction]
#[pyfunction]
fn echo_path(path: PathBuf) -> PyResult<PathBuf> {
    Ok(path)
}

#[gen_stub_pyfunction]
#[pyfunction]
fn ahash_dict() -> HashMap<String, i32, RandomState> {
    let mut map: HashMap<String, i32, RandomState> = HashMap::with_hasher(RandomState::new());
    map.insert("apple".to_string(), 3);
    map.insert("banana".to_string(), 2);
    map.insert("orange".to_string(), 5);
    map
}

#[gen_stub_pyclass_enum]
#[pyclass(eq, eq_int, from_py_object)]
#[derive(Debug, Clone, PartialEq, Eq, Hash)]
pub enum Number {
    #[pyo3(name = "FLOAT")]
    Float,
    #[pyo3(name = "INTEGER")]
    Integer,
}

#[gen_stub_pyclass_enum]
#[pyclass(eq, eq_int, from_py_object)]
#[pyo3(rename_all = "UPPERCASE")]
#[derive(Debug, Clone, PartialEq, Eq, Hash)]
pub enum NumberRenameAll {
    /// Float variant
    Float,
    Integer,
}

#[gen_stub_pyclass_complex_enum]
#[pyclass(from_py_object)]
#[pyo3(rename_all = "UPPERCASE")]
#[derive(Debug, Clone)]
pub enum NumberComplex {
    /// Float variant
    Float(f64),
    /// Integer variant
    #[pyo3(constructor = (int=2))]
    Integer {
        /// The integer value
        int: i32,
    },
}

/// Example from PyO3 documentation for complex enum
/// https://pyo3.rs/v0.25.1/class.html#complex-enums
#[gen_stub_pyclass_complex_enum]
#[pyclass]
enum Shape1 {
    Circle { radius: f64 },
    Rectangle { width: f64, height: f64 },
    RegularPolygon(u32, f64),
    Nothing {},
}

/// Example from PyO3 documentation for complex enum
/// https://pyo3.rs/v0.25.1/class.html#complex-enums
#[gen_stub_pyclass_complex_enum]
#[pyclass]
enum Shape2 {
    #[pyo3(constructor = (radius=1.0))]
    Circle {
        radius: f64,
    },
    #[pyo3(constructor = (*, width, height))]
    Rectangle {
        width: f64,
        height: f64,
    },
    #[pyo3(constructor = (side_count, radius=1.0))]
    RegularPolygon {
        side_count: u32,
        radius: f64,
    },
    Nothing {},
}

#[gen_stub_pymethods]
#[pymethods]
impl Number {
    #[getter]
    /// Whether the number is a float.
    fn is_float(&self) -> bool {
        matches!(self, Self::Float)
    }

    #[getter]
    /// Whether the number is an integer.
    fn is_integer(&self) -> bool {
        matches!(self, Self::Integer)
    }
}

#[gen_stub_pyclass]
#[pyclass]
pub struct DecimalHolder {
    #[pyo3(get)]
    value: Decimal,
}

#[gen_stub_pymethods]
#[pymethods]
impl DecimalHolder {
    #[new]
    fn new(value: Decimal) -> Self {
        Self { value }
    }
}

module_variable!("pure", "MY_CONSTANT1", usize);
module_variable!("pure", "MY_CONSTANT2", usize, 123);

#[gen_stub_pyfunction]
#[pyfunction]
async fn async_num() -> i32 {
    123
}

#[gen_stub_pyfunction]
#[pyfunction]
#[deprecated(since = "1.0.0", note = "This function is deprecated")]
fn deprecated_function() {
    println!("This function is deprecated");
}

// Test if non-any Py<PyAny> target can be a default value
#[gen_stub_pyfunction]
#[pyfunction]
#[pyo3(signature = (num = Number::Float))]
fn default_value(num: Number) -> Number {
    num
}

// These are the tests to test the treatment of `*args` and `**kwargs` in functions

/// Test struct for eq and ord comparison methods
#[gen_stub_pyclass]
#[pyclass(eq, ord, from_py_object)]
#[derive(Debug, Clone, PartialEq, PartialOrd)]
pub struct ComparableStruct {
    #[pyo3(get)]
    pub value: i32,
}

#[gen_stub_pymethods]
#[pymethods]
impl ComparableStruct {
    #[new]
    fn new(value: i32) -> Self {
        Self { value }
    }
}

/// Test struct for hash and str methods
#[gen_stub_pyclass]
#[pyclass(eq, hash, frozen, str, from_py_object)]
#[derive(Debug, Clone, Hash, PartialEq)]
pub struct HashableStruct {
    #[pyo3(get)]
    pub name: String,
}

impl std::fmt::Display for HashableStruct {
    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
        write!(f, "HashableStruct({})", self.name)
    }
}

#[gen_stub_pymethods]
#[pymethods]
impl HashableStruct {
    #[new]
    fn new(name: String) -> Self {
        Self { name }
    }
}

/// Takes a variable number of arguments and returns their string representation.
#[gen_stub_pyfunction]
#[pyfunction]
#[pyo3(signature = (*args))]
fn func_with_star_arg_typed(
    #[gen_stub(override_type(type_repr = "str"))] args: &Bound<PyTuple>,
) -> String {
    args.to_string()
}

/// Takes a variable number of arguments and returns their string representation.
#[gen_stub_pyfunction]
#[pyfunction]
#[pyo3(signature = (*args))]
fn func_with_star_arg(args: &Bound<PyTuple>) -> String {
    args.to_string()
}

/// Takes a variable number of keyword arguments and does nothing
#[gen_stub_pyfunction]
#[pyfunction]
#[pyo3(signature = (**kwargs))]
fn func_with_kwargs(kwargs: Option<&Bound<PyDict>>) -> bool {
    kwargs.is_some()
}

module_doc!("pure", "Document for {} ...", env!("CARGO_PKG_NAME"));

/// Initializes the Python module
#[pymodule]
fn pure(m: &Bound<PyModule>) -> PyResult<()> {
    m.add("MY_CONSTANT1", 19937)?;
    m.add("MY_CONSTANT2", 123)?;
    m.add_class::<A>()?;
    m.add_class::<B>()?;
    m.add_class::<MyDate>()?;
    m.add_class::<Number>()?;
    m.add_class::<NumberRenameAll>()?;
    m.add_class::<NumberComplex>()?;
    m.add_class::<Shape1>()?;
    m.add_class::<Shape2>()?;
    m.add_class::<ManualSubmit>()?;
    m.add_class::<PartialManualSubmit>()?;
    m.add_class::<OverrideType>()?;
    m.add_class::<ComparableStruct>()?;
    m.add_class::<HashableStruct>()?;
    m.add_class::<DecimalHolder>()?;
    m.add_class::<DataContainer>()?;
    m.add_class::<Placeholder>()?;
    m.add_class::<Calculator>()?;
    m.add_class::<InstanceValue>()?;
    m.add_class::<Problem>()?;
    m.add_class::<CustomStubType>()?;
    m.add_class::<NormalClass>()?;
    m.add_class::<CustomEnum>()?;
    m.add_class::<CustomComplexEnum>()?;
    m.add_class::<GetterSetterTypeTest>()?;
    m.add_function(wrap_pyfunction!(sum, m)?)?;
    m.add_function(wrap_pyfunction!(create_dict, m)?)?;
    m.add_function(wrap_pyfunction!(read_dict, m)?)?;
    m.add_function(wrap_pyfunction!(create_a, m)?)?;
    m.add_function(wrap_pyfunction!(echo_a_bound_ref, m)?)?;
    m.add_function(wrap_pyfunction!(echo_a_bound, m)?)?;
    m.add_function(wrap_pyfunction!(echo_a_py, m)?)?;
    m.add_function(wrap_pyfunction!(print_c, m)?)?;
    m.add_function(wrap_pyfunction!(str_len, m)?)?;
    m.add_function(wrap_pyfunction!(echo_path, m)?)?;
    m.add_function(wrap_pyfunction!(ahash_dict, m)?)?;
    m.add_function(wrap_pyfunction!(async_num, m)?)?;
    m.add_function(wrap_pyfunction!(deprecated_function, m)?)?;
    m.add_function(wrap_pyfunction!(default_value, m)?)?;
    m.add_function(wrap_pyfunction!(fn_override_type, m)?)?;
    m.add_function(wrap_pyfunction!(fn_with_python_param, m)?)?;
    m.add_function(wrap_pyfunction!(fn_with_python_stub, m)?)?;
    m.add_function(wrap_pyfunction!(overload_example_1, m)?)?;
    m.add_function(wrap_pyfunction!(overload_example_2, m)?)?;
    m.add_function(wrap_pyfunction!(as_tuple, m)?)?;
    m.add_function(wrap_pyfunction!(manual_overload_example_1, m)?)?;
    m.add_function(wrap_pyfunction!(manual_overload_example_2, m)?)?;
    m.add_function(wrap_pyfunction!(manual_overload_as_tuple, m)?)?;
    m.add_function(wrap_pyfunction!(add_decimals, m)?)?;
    m.add_function(wrap_pyfunction!(process_container, m)?)?;
    m.add_function(wrap_pyfunction!(sum_list, m)?)?;
    m.add_function(wrap_pyfunction!(create_containers, m)?)?;
    // Test-cases for `*args` and `**kwargs`
    m.add_function(wrap_pyfunction!(func_with_star_arg, m)?)?;
    m.add_function(wrap_pyfunction!(func_with_star_arg_typed, m)?)?;
    m.add_function(wrap_pyfunction!(func_with_kwargs, m)?)?;

    // Test cases for type: ignore functionality
    m.add_function(wrap_pyfunction!(test_type_ignore_specific, m)?)?;
    m.add_function(wrap_pyfunction!(test_type_ignore_all, m)?)?;
    m.add_function(wrap_pyfunction!(test_type_ignore_pyright, m)?)?;
    m.add_function(wrap_pyfunction!(test_type_ignore_custom, m)?)?;
    m.add_function(wrap_pyfunction!(test_type_ignore_no_comment_all, m)?)?;
    m.add_function(wrap_pyfunction!(test_type_ignore_no_comment_specific, m)?)?;

    // Test case for custom exceptions
    m.add("MyError", m.py().get_type::<MyError>())?;
    m.add_class::<NotIntError>()?;

    // Test case for runtime type alias (type_alias! macro)
    m.add_type_alias::<RuntimeNumberOrString>()?;

    // Test class for type: ignore functionality
    m.add_class::<TypeIgnoreTest>()?;

    // Test cases for time crate types
    m.add_function(wrap_pyfunction!(get_date, m)?)?;
    m.add_function(wrap_pyfunction!(get_time, m)?)?;
    m.add_function(wrap_pyfunction!(get_duration, m)?)?;
    m.add_function(wrap_pyfunction!(get_primitive_datetime, m)?)?;
    m.add_function(wrap_pyfunction!(get_offset_datetime, m)?)?;
    m.add_function(wrap_pyfunction!(get_utc_offset, m)?)?;
    m.add_function(wrap_pyfunction!(get_utc_datetime, m)?)?;
    m.add_function(wrap_pyfunction!(add_duration_to_date, m)?)?;
    m.add_function(wrap_pyfunction!(time_difference, m)?)?;

    // Test cases for chrono crate types
    m.add_function(wrap_pyfunction!(get_naive_date, m)?)?;
    m.add_function(wrap_pyfunction!(get_naive_time, m)?)?;
    m.add_function(wrap_pyfunction!(get_naive_datetime, m)?)?;
    m.add_function(wrap_pyfunction!(get_datetime_utc, m)?)?;
    m.add_function(wrap_pyfunction!(get_datetime_fixed_offset, m)?)?;
    m.add_function(wrap_pyfunction!(get_chrono_duration, m)?)?;
    m.add_function(wrap_pyfunction!(get_fixed_offset, m)?)?;
    m.add_function(wrap_pyfunction!(get_utc, m)?)?;
    m.add_function(wrap_pyfunction!(add_chrono_duration_to_date, m)?)?;
    m.add_function(wrap_pyfunction!(naive_time_difference, m)?)?;

    // Test cases for std::net IP address types
    m.add_function(wrap_pyfunction!(ipv4_localhost, m)?)?;
    m.add_function(wrap_pyfunction!(ipv6_localhost, m)?)?;
    m.add_function(wrap_pyfunction!(parse_ip, m)?)?;
    m.add_function(wrap_pyfunction!(is_loopback, m)?)?;

    // Test cases for f64 special values (INFINITY, NEG_INFINITY, NAN)
    m.add_class::<FloatValues>()?;
    m.add_function(wrap_pyfunction!(with_infinity_default, m)?)?;
    m.add_function(wrap_pyfunction!(with_neg_infinity_default, m)?)?;
    m.add_function(wrap_pyfunction!(with_nan_default, m)?)?;
    m.add_function(wrap_pyfunction!(with_float_default, m)?)?;
    Ok(())
}

/// Test function with type: ignore for specific rules
#[gen_stub_pyfunction]
#[gen_stub(type_ignore = ["arg-type", "return-value"])]
#[pyfunction]
fn test_type_ignore_specific() -> i32 {
    42
}

/// Test function with type: ignore (without equals for catch-all)
#[gen_stub_pyfunction]
#[gen_stub(type_ignore)]
#[pyfunction]
fn test_type_ignore_all() -> i32 {
    42
}

/// Test function with Pyright diagnostic rules
#[gen_stub_pyfunction]
#[gen_stub(type_ignore = ["reportGeneralTypeIssues", "reportReturnType"])]
#[pyfunction]
fn test_type_ignore_pyright() -> i32 {
    42
}

/// Test function with custom (unknown) rule
#[gen_stub_pyfunction]
#[gen_stub(type_ignore = ["custom-rule", "attr-defined"])]
#[pyfunction]
fn test_type_ignore_custom() -> i32 {
    42
}

// NOTE: Doc-comment MUST NOT be added to the next function,
// as it tests if `type_ignore` without no doccomment is handled correctly;
// i.e. it emits comment after `...`, not before.

#[gen_stub_pyfunction]
#[gen_stub(type_ignore)]
#[pyfunction]
fn test_type_ignore_no_comment_all() -> i32 {
    42
}

#[gen_stub_pyfunction]
#[gen_stub(type_ignore=["arg-type", "reportIncompatibleMethodOverride"])]
#[pyfunction]
fn test_type_ignore_no_comment_specific() -> i32 {
    42
}

/// Test class for method type: ignore functionality
#[gen_stub_pyclass]
#[pyclass]
pub struct TypeIgnoreTest {}

#[gen_stub_pymethods]
#[pymethods]
impl TypeIgnoreTest {
    #[new]
    fn new() -> Self {
        Self {}
    }

    /// Test method with type: ignore for specific rules
    #[gen_stub(type_ignore = ["union-attr", "return-value"])]
    fn test_method_ignore(&self, value: i32) -> i32 {
        value * 2
    }

    /// Test method with type: ignore (without equals for catch-all)
    #[gen_stub(type_ignore)]
    fn test_method_all_ignore(&self) -> i32 {
        42
    }
}

// Test type aliases WITHOUT docstrings (backward compatibility)
pyo3_stub_gen::type_alias!("pure", SimpleAlias = Option<usize>);
pyo3_stub_gen::type_alias!("pure", StrIntMap = HashMap<String, i32>);

// Type alias referring to locally defined class
pyo3_stub_gen::type_alias!(
    "pure",
    MaybeDecimal = Option<Bound<'static, DecimalHolder>>
);

// Direct union type syntax (no impl_stub_type! needed)
pyo3_stub_gen::type_alias!("pure", NumberOrStringAlias = i32 | String);

// Union of locally defined types using direct syntax
pyo3_stub_gen::type_alias!("pure", StructUnion = Bound<'static, ComparableStruct> | Bound<'static, HashableStruct>);

// Additional test cases for the new syntax
pyo3_stub_gen::type_alias!("pure", TripleUnion = i32 | String | bool);
pyo3_stub_gen::type_alias!("pure", GenericUnion = Option<i32> | Vec<String>);
pyo3_stub_gen::type_alias!("pure", SingleTypeAlias = Option<usize>); // Backward compatibility test

// Test type aliases WITH docstrings
pyo3_stub_gen::type_alias!(
    "pure",
    DocumentedAlias = Option<usize>,
    "This is a simple type alias with documentation"
);

pyo3_stub_gen::type_alias!(
    "pure",
    DocumentedUnion = i32 | String,
    "A union type with documentation"
);

pyo3_stub_gen::type_alias!(
    "pure",
    DocumentedMap = HashMap<String, Vec<i32>>,
    "A map type alias with detailed documentation.\n\nThis can have multiple lines of documentation."
);

// Test runtime type alias using type_alias! macro
// This type alias is available both in stubs AND at runtime
// Uses Rust types which are mapped to Python types via PyRuntimeType::runtime_type_object
type_alias!(
    "pure",
    RuntimeNumberOrString = i32 | String,
    "Either an integer or a string, available at runtime."
);

// Test type aliases using Python syntax (without docstrings)
pyo3_stub_gen::derive::gen_type_alias_from_python!(
    "pure",
    r#"
    from typing import TypeAlias
    import collections.abc

    CallbackType: TypeAlias = collections.abc.Callable[[str], None]
    OptionalCallback: TypeAlias = collections.abc.Callable[[str], None] | None
    SequenceOfInts: TypeAlias = collections.abc.Sequence[int]
    "#
);

// Test type aliases using Python syntax (with docstrings)
pyo3_stub_gen::derive::gen_type_alias_from_python!(
    "pure",
    r#"
    from typing import TypeAlias
    import collections.abc

    DocumentedCallback: TypeAlias = collections.abc.Callable[[str], None]
    """A callback function that takes a string and returns nothing"""

    UndocumentedCallback: TypeAlias = collections.abc.Callable[[int], bool]

    MultiLineDocCallback: TypeAlias = collections.abc.Callable[[str, int], bool]
    """
    A callback with multi-line documentation.

    This callback takes a string and an integer, and returns a boolean.
    """
    "#
);

// Test RustType markers in type aliases (TypeAlias syntax)
pyo3_stub_gen::derive::gen_type_alias_from_python!(
    "pure",
    r#"
    from typing import TypeAlias

    SimpleContainer: TypeAlias = pyo3_stub_gen.RustType["DataContainer"]
    ContainerList: TypeAlias = list[pyo3_stub_gen.RustType["DataContainer"]]
    ContainerMap: TypeAlias = dict[str, pyo3_stub_gen.RustType["DataContainer"]]
    OptionalContainer: TypeAlias = pyo3_stub_gen.RustType["DataContainer"] | None
    "#
);

// Test RustType markers in type aliases (Python 3.12+ type statement syntax)
pyo3_stub_gen::derive::gen_type_alias_from_python!(
    "pure",
    r#"
    type ContainerTuple = tuple[pyo3_stub_gen.RustType["DataContainer"], pyo3_stub_gen.RustType["DataContainer"]]
    type NestedContainer = list[list[pyo3_stub_gen.RustType["DataContainer"]]]
    "#
);

define_stub_info_gatherer!(stub_info);

/// Test of unit test for testing link problem
#[cfg(test)]
mod test {
    #[test]
    fn test() {
        assert_eq!(2 + 2, 4);
    }
}