summaryrefslogtreecommitdiff
path: root/tests/extmod/ujson_dump_iobase.py
blob: 51d507cd1a1673ebc4c647cb94dd5b1b5ff864ef (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
# test ujson.dump in combination with uio.IOBase

try:
    import uio as io
    import ujson as json
except ImportError:
    try:
        import io, json
    except ImportError:
        print('SKIP')
        raise SystemExit

if not hasattr(io, 'IOBase'):
    print('SKIP')
    raise SystemExit


# a user stream that only has the write method
class S(io.IOBase):
    def __init__(self):
        self.buf = ''
    def write(self, buf):
        if type(buf) == bytearray:
            # uPy passes a bytearray, CPython passes a str
            buf = str(buf, 'ascii')
        self.buf += buf
        return len(buf)


# dump to the user stream
s = S()
json.dump([123, {}], s) 
print(s.buf)