def test_deserializer():
array_data = [[1.0, 2.0, 3.0], [10.0, 20.0, 30.0]]
s = RecordSerializer()
buf = s.serialize(np.array(array_data))
d = RecordDeserializer()
for record, expected in zip(d.deserialize(buf, "who cares"), array_data):
assert record.features["values"].float64_tensor.values == expected
def test_serializer_accepts_one_dimensional_array():
s = RecordSerializer()
array_data = [1.0, 2.0, 3.0]
buf = s.serialize(np.array(array_data))
record_data = next(read_recordio(buf))
record = Record()
record.ParseFromString(record_data)
assert record.features["values"].float64_tensor.values == array_data
def test_serializer():
s = RecordSerializer()
array_data = [[1.0, 2.0, 3.0], [10.0, 20.0, 30.0]]
buf = s.serialize(np.array(array_data))
for record_data, expected in zip(read_recordio(buf), array_data):
record = Record()
record.ParseFromString(record_data)
assert record.features["values"].float64_tensor.values == expected