(t *testing.T)
| 19 | import "testing" |
| 20 | |
| 21 | func TestSavedModel(t *testing.T) { |
| 22 | bundle, err := LoadSavedModel("../cc/saved_model/testdata/half_plus_two/00000123", []string{"serve"}, nil) |
| 23 | if err != nil { |
| 24 | t.Fatalf("LoadSavedModel(): %v", err) |
| 25 | } |
| 26 | if op := bundle.Graph.Operation("y"); op == nil { |
| 27 | t.Fatalf("\"y\" not found in graph") |
| 28 | } |
| 29 | // TODO(jhseu): half_plus_two has a tf.Example proto dependency to run. Add a |
| 30 | // more thorough test when the generated protobufs are available. |
| 31 | } |
nothing calls this directly
no test coverage detected