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