summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--tests/extmod/ujson_dump_iobase.py32
1 files changed, 32 insertions, 0 deletions
diff --git a/tests/extmod/ujson_dump_iobase.py b/tests/extmod/ujson_dump_iobase.py
new file mode 100644
index 000000000..d30d1b561
--- /dev/null
+++ b/tests/extmod/ujson_dump_iobase.py
@@ -0,0 +1,32 @@
+# 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
+
+
+# dump to the user stream
+s = S()
+json.dump([123, {}], s)
+print(s.buf)