def test_str0():
@signature(types.unicode0(), returns=types.str0())
def f(u):
return 'str'
assert getsig(f) == [
model.SomeUnicodeString(no_nul=True),
model.SomeString(no_nul=True)
]
class scoped_unicode2wcharp:
def __init__(self, value):
if value is not None:
self.buf = unicode2wcharp(value)
else:
self.buf = lltype.nullptr(CWCHARP.TO)
__init__._annenforceargs_ = [None,
annmodel.SomeUnicodeString(can_be_None=True)]
def __enter__(self):
return self.buf
def __exit__(self, *args):
if self.buf:
free_wcharp(self.buf)
def unicode0():
return model.SomeUnicodeString(no_nul=True)
def unicode():
return model.SomeUnicodeString()
def compute_result_annotation(self, s_ll_str):
if strtype is str:
return annmodel.SomeString(can_be_None=True)
else:
return annmodel.SomeUnicodeString(can_be_None=True)
def test_unicode():
@signature(types.unicode(), returns=types.int())
def f(u):
return len(u)
assert getsig(f) == [model.SomeUnicodeString(), model.SomeInteger()]
