package pserver_test import ( "reflect" "sync" "testing" "time" "github.com/PaddlePaddle/Paddle/go/pserver" ) func TestFull(t *testing.T) { s := pserver.NewService() var p pserver.Parameter p.Name = "param_a" p.Content = []byte{1, 0, 0, 0, 2, 0, 0, 0, 3, 0, 0, 0} p.ElementType = pserver.Int32 err := s.InitParam(pserver.ParameterWithConfig{Param: p, Config: nil}, nil) if err != nil { t.FailNow() } var p1 pserver.Parameter p1.Name = "param_b" p1.Content = []byte{0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0} p1.ElementType = pserver.Float32 err = s.InitParam(pserver.ParameterWithConfig{Param: p1, Config: nil}, nil) if err != nil { t.FailNow() } err = s.FinishInitParams(0, nil) if err != nil { t.FailNow() } var param pserver.Parameter err = s.GetParam("param_b", ¶m) if err != nil { t.FailNow() } if !reflect.DeepEqual(param, p1) { t.FailNow() } g1, g2 := pserver.Gradient(p1), pserver.Gradient(p) err = s.SendGrad(g1, nil) if err != nil { t.FailNow() } err = s.SendGrad(g2, nil) if err != nil { t.FailNow() } var param1 pserver.Parameter err = s.GetParam("param_a", ¶m1) if err != nil { t.FailNow() } // don't compare content, since it's already changed by // gradient update. param1.Content = nil p.Content = nil if !reflect.DeepEqual(param1, p) { t.FailNow() } } func TestMultipleInit(t *testing.T) { s := pserver.NewService() err := s.FinishInitParams(0, nil) if err != nil { t.FailNow() } err = s.FinishInitParams(0, nil) if err.Error() != pserver.AlreadyInitialized { t.FailNow() } } func TestUninitialized(t *testing.T) { s := pserver.NewService() err := s.SendGrad(pserver.Gradient{}, nil) if err.Error() != pserver.Uninitialized { t.FailNow() } } func TestBlockUntilInitialized(t *testing.T) { s := pserver.NewService() ch := make(chan struct{}, 2) errCh := make(chan error, 2) var wg sync.WaitGroup wg.Add(1) go func() { var param pserver.Parameter err := s.GetParam("param_a", ¶m) if err != nil { errCh <- err } wg.Done() ch <- struct{}{} }() wg.Add(1) go func() { err := s.Save("", nil) if err != nil { errCh <- err } wg.Done() ch <- struct{}{} }() time.Sleep(50 * time.Millisecond) select { case <-ch: // some function returned before initialization is completed. t.FailNow() case <-errCh: t.FailNow() default: } var p pserver.Parameter p.Name = "param_a" p.Content = []byte{1, 0, 0, 0, 2, 0, 0, 0, 3, 0, 0, 0} p.ElementType = pserver.Int32 err := s.InitParam(pserver.ParameterWithConfig{Param: p, Config: nil}, nil) if err != nil { t.FailNow() } err = s.FinishInitParams(0, nil) if err != nil { t.FailNow() } wg.Wait() }