Merge pull request #110 from nextmv-io/merschformann/region-allocation

Adds Region Allocation apps

Author: merschformann

.DS_Store

@@ -0,0 +1,50 @@
+package mip
+
+import (
+	"os"
+	"testing"
+
+	"github.com/nextmv-io/sdk/golden"
+)
+
+func TestMain(m *testing.M) {
+	code := m.Run()
+	os.Exit(code)
+}
+
+func TestGolden(t *testing.T) {
+	golden.FileTests(
+		t,
+		"inputs",
+		golden.Config{
+			Args: []string{
+				"-duration",
+				"30",
+			},
+			TransientFields: []golden.TransientField{
+				{
+					Key:         "$.statistics.result.duration",
+					Replacement: golden.StableFloat,
+				},
+				{
+					Key:         "$.statistics.run.duration",
+					Replacement: golden.StableFloat,
+				},
+				{
+					Key:         "$.options.output",
+					Replacement: "output.json",
+				},
+				{
+					Key:         "$.options.input",
+					Replacement: "input.json",
+				},
+			},
+			ExecutionConfig: &golden.ExecutionConfig{
+				Command:    "python3",
+				Args:       []string{"../../../python-ortools-region-allocation/main.py"},
+				InputFlag:  "-input",
+				OutputFlag: "-output",
+			},
+		},
+	)
+}

.nextmv/golden/python-ortools-region-allocation/inputs/input.json

@@ -1,7 +1,10 @@
 package mip
 
 import (
+	"flag"
+	"fmt"
 	"os"
+	"regexp"
 	"slices"
 	"strings"
 	"testing"
@@ -10,6 +13,11 @@ import (
 	"gopkg.in/yaml.v2"
 )
 
+// filter is a flag to filter the tests to run by name. Allows regex
+// matching. For example, to run all tests that start with "python-" use
+// -filter='python-.*.'.
+var filter = flag.String("filter", "", "filter tests by name")
+
 func TestMain(m *testing.M) {
 	code := m.Run()
 	os.Exit(code)
@@ -67,6 +75,16 @@ func TestGolden(t *testing.T) {
 		{Regex: `xpress\.init\(.*\)`, Replacement: `xpress.init("path/to/xpress")`},
 	}
 
+	// Use the filter flag to filter the tests to run
+	var regex *regexp.Regexp
+	if *filter != "" {
+		regex, err = regexp.Compile(*filter)
+		if err != nil {
+			fmt.Println("Error compiling filter regex:", err)
+			return
+		}
+	}
+
 	// Run all readme tests
 	dirs, err := os.ReadDir(".")
 	if err != nil {
@@ -101,10 +119,14 @@ func TestGolden(t *testing.T) {
 			if scriptConfig.Skip {
 				continue
 			}
-			t.Run(app+"/"+script, func(t *testing.T) {
+			t.Run("readme", func(t *testing.T) {
+				testName := app + "/" + script
+				if regex != nil && !regex.MatchString(testName) {
+					t.Skipf("skipping test %s", testName)
+				}
 				golden.BashTestFile(
 					t,
-					app+"/"+script,
+					testName,
 					golden.BashConfig{
 						DisplayStdout: !scriptConfig.Silent,
 						WorkingDir:    "../../" + app,

.nextmv/golden/python-ortools-region-allocation/inputs/input.json.golden

@@ -0,0 +1 @@
+pip3 install -r requirements.txt

.nextmv/golden/python-ortools-region-allocation/main_test.go

@@ -0,0 +1,2 @@
+python3 main.py -input input.json -output output.json \
+  -duration 30 -provider scip