[prism] registrations for Regression tests. (apache#27589)

Co-authored-by: lostluck <13907733+lostluck@users.noreply.github.com>
cushon · May 24, 2024 · 09a3074 · 09a3074
1 parent 28008af
commit 09a3074
Show file tree

Hide file tree

Showing 2 changed files with 19 additions and 0 deletions.
diff --git a/sdks/go/test/regression/lperror.go b/sdks/go/test/regression/lperror.go
@@ -22,8 +22,15 @@ import (
 	"sort"
 
 	"github.com/apache/beam/sdks/v2/go/pkg/beam"
+	"github.com/apache/beam/sdks/v2/go/pkg/beam/register"
 )
 
+func init() {
+	register.Function2x2(toFoo)
+	register.Iter1[*fruit]()
+	register.Function3x1(toID)
+}
+
 // REPRO found by https://github.com/zelliott
 
 type fruit struct {

diff --git a/sdks/go/test/regression/pardo.go b/sdks/go/test/regression/pardo.go
@@ -18,10 +18,22 @@ package regression
 
 import (
 	"github.com/apache/beam/sdks/v2/go/pkg/beam"
+	"github.com/apache/beam/sdks/v2/go/pkg/beam/register"
 	"github.com/apache/beam/sdks/v2/go/pkg/beam/testing/passert"
 	"github.com/apache/beam/sdks/v2/go/pkg/beam/testing/ptest"
 )
 
+func init() {
+	register.Function1x1(directFn)
+	register.Function2x0(emitFn)
+	register.Function3x0(emit2Fn)
+	register.Function2x1(mixedFn)
+	register.Function2x2(directCountFn)
+	register.Function3x1(emitCountFn)
+	register.Emitter1[int]()
+	register.Iter1[int]()
+}
+
 func directFn(elm int) int {
 	return elm + 1
 }