# mode: run

# tag: cpp, werror

# cython: c_string_encoding=ascii, c_string_type=str

cimport cython

from libcpp.string cimport string

b_asdf = b'asdf'

u_asdf = u'asdf'

s_asdf = 'asdf'

s_s = 's'

def test_conversion(py_obj):

    """

    >>> test_conversion(b_asdf) == s_asdf or test_conversion(b_asdf)

    True

    >>> test_conversion(u_asdf) == s_asdf or test_conversion(u_asdf)

    True

    >>> test_conversion(123)  # doctest: +ELLIPSIS

    Traceback (most recent call last):

    TypeError: expected ..., int found

    """

    cdef string s = py_obj

    assert <size_t>len(py_obj) == s.length(), '%d != %d' % (len(py_obj), s.length())

    return s

def test_empty(py_obj):

    """

    >>> test_empty('')

    True

    >>> test_empty('abc')

    False

    >>> test_empty(u_asdf[:0])

    True

    >>> test_empty(u_asdf)

    False

    """

    cdef string a = py_obj

    return a.empty()

def test_push_back(a):

    """

    >>> test_push_back(b_asdf) == s_asdf + s_s

    True

    >>> test_push_back(u_asdf) == s_asdf + s_s

    True

    """

    cdef string s = a

    s.push_back(<char>ord('s'))

    return s

def test_clear(a):

    """

    >>> test_clear(u_asdf) == s_s[:0]

    True

    >>> test_clear(b_asdf) == s_s[:0]

    True

    """

    cdef string s = a

    s.clear()

    return s

def test_assign(char *a):

    """

    >>> test_assign(b_asdf) == 'ggg'

    True

    """

    cdef string s = string(a)

    s.assign(<char *>"ggg")

    return s.c_str()

def test_bytes_cast(a):

    """

    >>> b = test_bytes_cast(b'abc')

    >>> isinstance(b, bytes)

    True

    >>> print(b.decode('ascii'))

    abc

    >>> b = test_bytes_cast(b'abc\\xe4\\xfc')

    >>> isinstance(b, bytes)

    True

    >>> len(b)

    5

    >>> print(b[:3].decode('ascii'))

    abc

    >>> print(ord(b[3:4]))

    228

    >>> print(ord(b[4:5]))

    252

    """

    cdef string s = a

    assert s.length() == <size_t>len(a), "%d != %d" % (s.length(), len(a))

    return <bytes>s

def test_bytearray_cast(a):

    """

    >>> b = test_bytearray_cast(b'abc')

    >>> isinstance(b, bytearray)

    True

    >>> print(b.decode('ascii'))

    abc

    >>> b = test_bytearray_cast(b'abc\\xe4\\xfc')

    >>> isinstance(b, bytearray)

    True

    >>> len(b)

    5

    >>> print(b[:3].decode('ascii'))

    abc

    >>> print(ord(b[3:4]))

    228

    >>> print(ord(b[4:5]))

    252

    """

    cdef string s = a

    assert s.length() == <size_t>len(a), "%d != %d" % (s.length(), len(a))

    return <bytearray>s

def test_unicode_cast(a):

    """

    >>> u = test_unicode_cast(b'abc')

    >>> type(u) is type(u_asdf) or type(u)

    True

    >>> print(u)

    abc

    """

    cdef string s = a

    assert s.length() == <size_t>len(a), "%d != %d" % (s.length(), len(a))

    return <unicode>s

def test_str_cast(a):

    """

    >>> s = test_str_cast(b'abc')

    >>> type(s) is type(s_asdf) or type(s)

    True

    >>> print(s)

    abc

    """

    cdef string s = a

    assert s.length() == <size_t>len(a), "%d != %d" % (s.length(), len(a))

    return <str>s