(t *testing.T)
| 204 | } |
| 205 | |
| 206 | func TestTensorSerializationErrors(t *testing.T) { |
| 207 | // String tensors cannot be serialized |
| 208 | t1, err := NewTensor("abcd") |
| 209 | if err != nil { |
| 210 | t.Fatal(err) |
| 211 | } |
| 212 | buf := new(bytes.Buffer) |
| 213 | if n, err := t1.WriteContentsTo(buf); n != 0 || err == nil || buf.Len() != 0 { |
| 214 | t.Errorf("Got (%v, %v, %v) want (0, <non-nil>, 0)", n, err, buf.Len()) |
| 215 | } |
| 216 | // Should fail to read a truncated value. |
| 217 | if t1, err = NewTensor(int8(8)); err != nil { |
| 218 | t.Fatal(err) |
| 219 | } |
| 220 | n, err := t1.WriteContentsTo(buf) |
| 221 | if err != nil { |
| 222 | t.Fatal(err) |
| 223 | } |
| 224 | r := bytes.NewReader(buf.Bytes()[:n-1]) |
| 225 | if _, err = ReadTensor(t1.DataType(), t1.Shape(), r); err == nil { |
| 226 | t.Error("ReadTensor should have failed if the tensor content was truncated") |
| 227 | } |
| 228 | } |
| 229 | |
| 230 | func TestReadTensorReadAll(t *testing.T) { |
| 231 | // Get the bytes of a tensor. |
nothing calls this directly
no test coverage detected