diff --git a/go/pserver/service_test.go b/go/pserver/service_test.go index 175c3c3ad876e12d42fa6554e0542ac28e5e35af..b746d13e1ca71e697c464f84d915af029d37120c 100644 --- a/go/pserver/service_test.go +++ b/go/pserver/service_test.go @@ -79,7 +79,7 @@ func TestMultipleInit(t *testing.T) { } err = s.FinishInitParams(0, nil) - if err != pserver.ErrAlreadyInitialized { + if err.Error() != pserver.AlreadyInitialized { t.FailNow() } } @@ -87,7 +87,7 @@ func TestMultipleInit(t *testing.T) { func TestUninitialized(t *testing.T) { s := pserver.NewService() err := s.SendGrad(pserver.Gradient{}, nil) - if err != pserver.ErrUninitialized { + if err.Error() != pserver.Uninitialized { t.FailNow() } }