def test_json_serialize_dict_with_numpy():
data = {'tensor1': np.asarray([1, 2, 3]), 'tensor2': np.asarray([4, 5, 6])}
serialized = tf_json_serializer(data)
# deserialize again for assertion, since dict order is not guaranteed
deserialized = json.loads(serialized)
assert deserialized == {'tensor1': [1, 2, 3], 'tensor2': [4, 5, 6]}
def test_json_serialize_numpy():
data = np.asarray([[1, 2, 3], [4, 5, 6]])
assert tf_json_serializer(data) == '[[1, 2, 3], [4, 5, 6]]'
def test_json_serialize_dict():
data = {'tensor1': [1, 2, 3], 'tensor2': [4, 5, 6]}
serialized = tf_json_serializer(data)
# deserialize again for assertion, since dict order is not guaranteed
deserialized = json.loads(serialized)
assert deserialized == data