def test_o2b2o():
import pickle
import sys
import unittest
from ase.db.core import object_to_bytes, bytes_to_object
from ase.cell import Cell
import numpy as np
if sys.version_info < (3, 6):
# pickles are not sorted
raise unittest.SkipTest
for o1 in [
1.0, {
'a': np.zeros((2, 2), np.float32),
'b': np.zeros((0, 2), int)
}, ['a', 42, True, None, np.nan, np.inf, 1j],
Cell(np.eye(3)), {
'a': {
'b': {
'c': np.ones(3)
}
}
}
]:
p1 = pickle.dumps(o1)
b1 = object_to_bytes(o1)
o2 = bytes_to_object(b1)
p2 = pickle.dumps(o2)
print(o2)
print(b1)
print()
assert p1 == p2, (o1, p1, p2, vars(o1), vars(p1), vars(p2))
def data(self):
"""Data dict."""
if isinstance(self._data, str):
self._data = decode(self._data) # lazy decoding
elif isinstance(self._data, bytes):
from ase.db.core import bytes_to_object
self._data = bytes_to_object(self._data) # lazy decoding
return FancyDict(self._data)
def test_o2b2o():
for o1 in [
1.0, {
'a': np.zeros((2, 2), np.float32),
'b': np.zeros((0, 2), int)
}, ['a', 42, True, None, np.nan, np.inf, 1j],
Cell(np.eye(3)), {
'a': {
'b': {
'c': np.ones(3)
}
}
}
]:
p1 = pickle.dumps(o1)
b1 = object_to_bytes(o1)
o2 = bytes_to_object(b1)
p2 = pickle.dumps(o2)
print(o2)
print(b1)
print()
assert p1 == p2, (o1, p1, p2, vars(o1), vars(p1), vars(p2))
def decode(self, txt, lazy=False):
if lazy:
return txt
if isinstance(txt, str):
return ase.io.jsonio.decode(txt)
return bytes_to_object(txt)
import pickle
from ase.db.core import object_to_bytes, bytes_to_object
from ase.cell import Cell
import numpy as np
for o1 in [1.0,
{'a': np.zeros((2, 2), np.float32),
'b': np.zeros((0, 2), int)},
['a', 42, True, None, np.nan, np.inf, 1j],
Cell(np.eye(3)),
{'a': {'b': {'c': np.ones(3)}}}]:
p1 = pickle.dumps(o1)
b1 = object_to_bytes(o1)
o2 = bytes_to_object(b1)
p2 = pickle.dumps(o2)
print(o2)
print(b1)
print()
assert p1 == p2