diff --git a/Makefile b/Makefile index 04691cf7..c4fddb1d 100644 --- a/Makefile +++ b/Makefile @@ -77,7 +77,11 @@ format: ## Format Go source files .PHONY: test test: check-ginkgo download-tokenizer download-zmq ## Run tests @printf "\033[33;1m==== Running tests ====\033[0m\n" +ifdef GINKGO_FOCUS + CGO_ENABLED=1 ginkgo -ldflags="$(GO_LDFLAGS)" -v -r --focus="$(GINKGO_FOCUS)" +else CGO_ENABLED=1 ginkgo -ldflags="$(GO_LDFLAGS)" -v -r +endif .PHONY: post-deploy-test post-deploy-test: ## Run post deployment tests diff --git a/pkg/llm-d-inference-sim/simulator_test.go b/pkg/llm-d-inference-sim/simulator_test.go index cf9fd468..88d87759 100644 --- a/pkg/llm-d-inference-sim/simulator_test.go +++ b/pkg/llm-d-inference-sim/simulator_test.go @@ -25,6 +25,7 @@ import ( "net/http" "os" "strings" + "time" "github.com/llm-d/llm-d-inference-sim/pkg/common" . "github.com/onsi/ginkgo/v2" @@ -754,6 +755,8 @@ var _ = Describe("Simulator", func() { KVCacheTransferLatency: 2048, KVCacheTransferLatencyStdDev: 2048, } + + common.InitRandom(time.Now().UnixNano()) }) DescribeTable("should calculate inter token latency correctly",