pybind11/tests/test_smart_ptr_private_first_base.py

# -*- coding: utf-8 -*-
import pytest

from pybind11_tests import smart_ptr_private_first_base as m


def test_make_drvd_pass_base():
    d = m.make_shared_drvd()
    i = m.pass_shared_base(d)
    assert i == 200


def test_make_drvd_up_cast_pass_drvd():
    b = m.make_shared_drvd_up_cast()
    # the base return is down-cast immediately.
    assert b.__class__.__name__ == "drvd"
    i = m.pass_shared_drvd(b)
    assert i == 200