-
Notifications
You must be signed in to change notification settings - Fork 0
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
Showing
10 changed files
with
94 additions
and
10 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,36 @@ | ||
from io import StringIO | ||
|
||
from zetta_utils.parsing import json | ||
|
||
|
||
def test_str_roundtrip(): | ||
assert json.loads(json.dumps([1, "2"])) == [1, "2"] | ||
|
||
|
||
def test_str_tuple_roundtrip(): | ||
assert json.loads(json.dumps((1, "2"))) == (1, "2") | ||
|
||
|
||
def test_str_nested_tuple_roundtrip(): | ||
assert json.loads(json.dumps((1, "2", (3, "4")))) == (1, "2", (3, "4")) | ||
|
||
|
||
def test_fp_roundtrip(): | ||
fp = StringIO() | ||
json.dump([1, "2"], fp) | ||
fp.seek(0) | ||
assert json.load(fp) == [1, "2"] | ||
|
||
|
||
def test_fp_tuple_roundtrip(): | ||
fp = StringIO() | ||
json.dump((1, "2"), fp) | ||
fp.seek(0) | ||
assert json.load(fp) == (1, "2") | ||
|
||
|
||
def test_fp_nested_tuple_roundtrip(): | ||
fp = StringIO() | ||
json.dump((1, "2", (3, "4")), fp) | ||
fp.seek(0) | ||
assert json.load(fp) == (1, "2", (3, "4")) |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -1,2 +1,3 @@ | ||
from . import cue | ||
from . import ngl_state | ||
from . import json |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,50 @@ | ||
from __future__ import annotations | ||
|
||
import json | ||
from collections.abc import Iterator | ||
from typing import TYPE_CHECKING, Any | ||
|
||
if TYPE_CHECKING: | ||
from _typeshed import SupportsRead, SupportsWrite | ||
|
||
|
||
def _mark_python_types(obj: Any) -> Any: | ||
if isinstance(obj, tuple): | ||
return {"__tuple__": [_mark_python_types(e) for e in obj]} | ||
if isinstance(obj, list): | ||
return [_mark_python_types(e) for e in obj] | ||
if isinstance(obj, dict): | ||
return {key: _mark_python_types(value) for key, value in obj.items()} | ||
else: | ||
return obj | ||
|
||
|
||
class ZettaSpecJSONEncoder(json.JSONEncoder): | ||
def encode(self, o: Any) -> str: | ||
return super().encode(_mark_python_types(o)) | ||
|
||
def iterencode(self, o: Any, _one_shot: bool = False) -> Iterator[str]: | ||
return super().iterencode(_mark_python_types(o), _one_shot=_one_shot) | ||
|
||
|
||
def tuple_hook(obj): | ||
if "__tuple__" in obj: | ||
return tuple(obj["__tuple__"]) | ||
else: | ||
return obj | ||
|
||
|
||
def dumps(obj, **kwargs) -> str: | ||
return json.dumps(obj, cls=ZettaSpecJSONEncoder, **kwargs) | ||
|
||
|
||
def dump(obj: Any, fp: SupportsWrite[str], **kwargs) -> None: | ||
json.dump(obj, fp, cls=ZettaSpecJSONEncoder, **kwargs) | ||
|
||
|
||
def loads(s: str, **kwargs) -> Any: | ||
return json.loads(s, object_hook=tuple_hook, **kwargs) | ||
|
||
|
||
def load(fp: SupportsRead[str | bytes], **kwargs) -> Any: | ||
return json.load(fp, object_hook=tuple_hook, **kwargs) |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters