[BEAM-14338] Update watermark unit tests to use time.Time.Equals()

apache · Apr 20, 2022 · e90d218 · e90d218
1 parent e4d2050
commit e90d218
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/sdks/go/pkg/beam/core/runtime/exec/sdf_test.go b/sdks/go/pkg/beam/core/runtime/exec/sdf_test.go
@@ -941,14 +941,14 @@ func TestAsSplittableUnit(t *testing.T) {
 					if got, ok := ow["output1"]; !ok {
 						t.Errorf("ProcessSizedElementsAndRestrictions(%v) has no watermark for ouptput1, want: %v",
 							test.in, test.want)
-					} else if got.AsTime() != test.want {
+					} else if !got.AsTime().Equal(test.want) {
 						t.Errorf("ProcessSizedElementsAndRestrictions(%v) has incorrect watermark for output1: got: %v, want: %v",
 							test.in, got.AsTime(), test.want)
 					}
 					if got, ok := ow["output2"]; !ok {
 						t.Errorf("ProcessSizedElementsAndRestrictions(%v) has no watermark for ouptput2, want: %v",
 							test.in, test.want)
-					} else if got.AsTime() != test.want {
+					} else if !got.AsTime().Equal(test.want) {
 						t.Errorf("ProcessSizedElementsAndRestrictions(%v) has incorrect watermark for output2: got: %v, want: %v",
 							test.in, got.AsTime(), test.want)
 					}