summaryrefslogtreecommitdiff
path: root/tests/extmod/json_dump_separators.py
blob: 4f8e56dceb53618034107b390e907b10d406706d (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
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
try:
    from io import StringIO
    import json
except ImportError:
    print("SKIP")
    raise SystemExit

for sep in [
    None,
    (", ", ": "),
    (",", ": "),
    (",", ":"),
    [", ", ": "],
    [",", ": "],
    [",", ":"],
]:
    s = StringIO()
    json.dump(False, s, separators=sep)
    print(s.getvalue())

    s = StringIO()
    json.dump({"a": (2, [3, None])}, s, separators=sep)
    print(s.getvalue())

    # dump to a small-int not allowed
    try:
        json.dump(123, 1, separators=sep)
    except (AttributeError, OSError):  # CPython and uPy have different errors
        print("Exception")

    # dump to an object not allowed
    try:
        json.dump(123, {}, separators=sep)
    except (AttributeError, OSError):  # CPython and uPy have different errors
        print("Exception")


try:
    s = StringIO()
    json.dump(False, s, separators={"a": 1})
except (TypeError, ValueError):  # CPython and uPy have different errors
    print("Exception")

# invalid separator types
for sep in [1, object()]:
    try:
        s = StringIO()
        json.dump(False, s, separators=sep)
    except TypeError:
        print("Exception")

# too many/ not enough separators
for sep in [(), (",", ":", "?"), (",",), []]:
    try:
        s = StringIO()
        json.dump(False, s, separators=sep)
    except ValueError:
        print("Exception")