update

tests/antithesis/avalanchego/gencomposeconfig/main.go

@@ -7,14 +7,16 @@ import (
 	"log"
 
 	"github.com/ava-labs/avalanchego/tests/antithesis"
-	"github.com/ava-labs/avalanchego/tests/fixture/tmpnet"
 )
 
-const baseImageName = "antithesis-avalanchego"
+const (
+	baseImageName = "antithesis-avalanchego"
+	NumNodes      = 6
+)
 
 // Creates docker-compose.yml and its associated volumes in the target path.
 func main() {
-	network := tmpnet.LocalNetworkOrPanic()
+	network := antithesis.CreateNetwork(NumNodes)
 	if err := antithesis.GenerateComposeConfig(network, baseImageName, "" /* runtimePluginDir */); err != nil {
 		log.Fatalf("failed to generate compose config: %v", err)
 	}

tests/antithesis/network.go

@@ -0,0 +1,15 @@
+// Copyright (C) 2019-2024, Ava Labs, Inc. All rights reserved.
+// See the file LICENSE for licensing terms.
+
+package antithesis
+
+import (
+	"github.com/ava-labs/avalanchego/tests/fixture/tmpnet"
+)
+
+// Creates a network with the given number of nodes.
+func CreateNetwork(nodesCount int) *tmpnet.Network {
+	network := tmpnet.NewDefaultNetwork("antithesis-network")
+	network.Nodes = tmpnet.NewNodesOrPanic(nodesCount)
+	return network
+}

tests/antithesis/xsvm/gencomposeconfig/main.go

@@ -12,11 +12,14 @@ import (
 	"github.com/ava-labs/avalanchego/tests/fixture/tmpnet"
 )
 
-const baseImageName = "antithesis-xsvm"
+const (
+	baseImageName = "antithesis-xsvm"
+	NumNodes      = 6
+)
 
 // Creates docker-compose.yml and its associated volumes in the target path.
 func main() {
-	network := tmpnet.LocalNetworkOrPanic()
+	network := antithesis.CreateNetwork(NumNodes)
 	network.Subnets = []*tmpnet.Subnet{
 		subnet.NewXSVMOrPanic("xsvm", genesis.VMRQKey, network.Nodes...),
 	}