def load_setup(jobid):
"""Loads but does not type setup.json from jobid.
You probably want to use extras.job_params instead.
"""
d = json_load('setup.json', jobid)
version = d.version
if version == 1:
d.jobs = d.pop('jobids')
version = 2
if version == 2:
if 'exectime' not in d and 'profile' in d:
d.exectime = d.pop('profile')
d.versions = DotDict()
python_path = d.pop('python', None)
if python_path:
d.versions.python_path = python_path
version = 3
if version == 3:
# no changes here, it's only used to know how to find datasets
version = 4
if version != 4:
raise AcceleratorError(
"Don't know how to load setup.json version %d (in %s)" % (
d.version,
jobid,
))
return d
def json_load(self,
filename='result.json',
sliceno=None,
unicode_as_utf8bytes=PY2):
from accelerator.extras import json_load
return json_load(self.filename(filename, sliceno),
unicode_as_utf8bytes=unicode_as_utf8bytes)
def test(name, input, want_obj, want_bytes, **kw):
json_save(input, name, **kw)
with open(name, "rb") as fh:
got_bytes_raw = fh.read()
assert got_bytes_raw[
-1:] == b"\n", name + " didn't even end with a newline"
got_bytes_raw = got_bytes_raw[:-1]
as_str = json_encode(input, as_str=True, **kw)
as_bytes = json_encode(input, as_str=False, **kw)
assert isinstance(as_str, str) and isinstance(
as_bytes, bytes), "json_encode returns the wrong types: %s %s" % (
type(as_str),
type(as_bytes),
)
assert as_bytes == got_bytes_raw, "json_save doesn't save the same thing json_encode returns for " + name
if PY3:
as_str = as_str.encode("utf-8")
assert as_bytes == as_str, "json_encode doesn't return the same data for as_str=True and False"
got_obj = json_load(name)
assert want_obj == got_obj, "%s roundtrips wrong (wanted %r, got %r)" % (
name, want_obj, got_obj)
with open(name, "rb") as fh:
got_bytes_fuzzy = b"".join(line.strip() for line in fh)
assert want_bytes == got_bytes_fuzzy, "%s wrong on disk (but decoded right)" % (
name, )
def load_setup(jobid):
"""Loads but does not type setup.json from jobid.
You probably want to use extras.job_params instead.
"""
d = json_load('setup.json', jobid)
version = d.version
if version == 1:
d.jobs = d.pop('jobids')
version = 2
if version == 2:
if 'exectime' not in d and 'profile' in d:
d.exectime = d.pop('profile')
d.versions = DotDict()
python_path = d.pop('python', None)
if python_path:
d.versions.python_path = python_path
version = 3
if version == 3:
if '_typing' in d:
d['_typing'] = {d.method: d['_typing']}
d.params = {
d.method:
DotDict({k: d[k]
for k in ('options', 'datasets', 'jobs')})
}
else:
raise Exception(
"Don't know how to load setup.json version %d (in %s)" % (
d.version,
jobid,
))
return d
def load_setup(jobid):
"""Loads but does not type setup.json from jobid.
You probably want to use extras.job_params instead.
"""
d = json_load('setup.json', jobid)
if d['version'] == 1:
if '_typing' in d:
d['_typing'] = {d.method: d['_typing']}
d.params = {
d.method:
DotDict({k: d[k]
for k in ('options', 'datasets', 'jobids')})
}
else:
raise Exception("Don't know how to load setup.json version %d" %
(d['version'], ))
return d
def load_setup(jobid):
"""Loads but does not type setup.json from jobid.
You probably want to use extras.job_params instead.
"""
d = json_load('setup.json', jobid)
version = d.version
if version == 1:
d.jobs = d.pop('jobids')
version = 2
if version == 2:
if 'exectime' not in d and 'profile' in d:
d.exectime = d.pop('profile')
d.versions = DotDict()
python_path = d.pop('python', None)
if python_path:
d.versions.python_path = python_path
version = 3
if version != 3:
raise Exception(
"Don't know how to load setup.json version %d (in %s)" % (
d.version,
jobid,
))
return d
def synthesis():
test(
"simple.json",
dict(a=1, b="2"),
dict(a=1, b="2"),
b'{"a": 1,"b": "2"}',
)
test(
"True False None.json",
[True, False, None],
[True, False, None],
b'[true,false,null]',
)
test(
"list like types.json",
dict(
list=[
1,
2,
3,
],
set={"foo"},
tuple=(
"a",
"b",
"c",
),
),
dict(
list=[
1,
2,
3,
],
set=["foo"],
tuple=[
"a",
"b",
"c",
],
),
b'{"list": [1,2,3],"set": ["foo"],"tuple": ["a","b","c"]}',
)
unicode_want = u"bl\xe4"
if PY2:
unicode_want = unicode_want.encode("utf-8")
test(
"unicode.json",
u"bl\xe4",
unicode_want,
b'"bl\\u00e4"',
)
# Verify that utf-8 encoding also works for reading.
with open("utf-8.json", "wb") as fh:
fh.write(b'"bl\xc3\xa4"')
assert json_load("utf-8.json") == unicode_want
# This is supposed to work on PY2, but not PY3.
try:
test(
"string encoding.json",
[
u"\xe4",
b"\xc3\xa4",
[
b"\xe4",
{
b"\xe4": b"\xc3\xa4",
},
],
],
[
b"\xc3\xa4",
b"\xc3\xa4",
[
b"\xc3\xa4",
{
b"\xc3\xa4": b"\xc3\xa4",
},
],
],
b'["\\u00e4","\\u00e4",["\\u00e4",{"\\u00e4": "\\u00e4"}]]',
)
string_encoding_ok = True
except TypeError:
string_encoding_ok = False
if PY2:
assert string_encoding_ok, "Bytes are supposed to work in json_encode on PY2"
else:
assert not string_encoding_ok, "Bytes are not supposed to work in json_encode on PY3"
# 720 permutations might be a bit much, but at least it's unlikely to
# miss ordering problems.
sorted_s = None
for ix, pairs in enumerate(permutations(zip("abcdef", range(6)))):
d = OrderedDict()
s = "{"
for k, v in pairs:
d[k] = v
s += '"%s": %s,' % (k, v)
s = (s[:-1] + "}").encode("ascii")
if not sorted_s:
sorted_s = s
test("ordered%d.json" % (ix, ), d, d, s, sort_keys=False)
test("sorted%d.json" % (ix, ), d, d, sorted_s, sort_keys=True)
def json_load(self, sliceno=None, unicode_as_utf8bytes=PY2):
from accelerator.extras import json_load
return json_load(self.resolve(sliceno),
unicode_as_utf8bytes=unicode_as_utf8bytes)
