micropython/tests/extmod/json_dump_iobase.py
Jim Mussared 4216bc7d13 tests: Replace umodule with module everywhere.
This work was funded through GitHub Sponsors.

Signed-off-by: Jim Mussared <jim.mussared@gmail.com>
2023-06-08 17:54:24 +10:00

31 lines
595 B
Python

# test json.dump in combination with io.IOBase
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)