Fix unit tests

satmihir · Sep 26, 2024 · 1f8055d · 1f8055d
1 parent 031febb
commit 1f8055d
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/pkg/config/tuning_test.go b/pkg/config/tuning_test.go
@@ -17,13 +17,13 @@ func TestGenerateTunedStructureConfig(t *testing.T) {
 	assert.Equal(t, int(conf.L), 3)
 	assert.Equal(t, int(conf.M), 1000)
 	assert.Equal(t, conf.Pi*25, float64(1))
-	assert.Equal(t, conf.Pd*25*10, float64(1))
+	assert.Equal(t, conf.Pd*25*1000, float64(1))
 }
 
 func TestDefaultStructureConfig(t *testing.T) {
 	conf := DefaultFairnessTrackerConfig()
 	assert.Equal(t, int(conf.L), 3)
 	assert.Equal(t, int(conf.M), 1000)
 	assert.Equal(t, conf.Pi*25, float64(1))
-	assert.Equal(t, conf.Pd*25*10, float64(1))
+	assert.Equal(t, conf.Pd*25*1000, float64(1))
 }
diff --git a/pkg/tracker/tracker_test.go b/pkg/tracker/tracker_test.go
@@ -40,7 +40,7 @@ func TestEndToEnd(t *testing.T) {
 	assert.True(t, resp.ShouldThrottle)
 
 	// It takes 10x more failures to get back to 0 probability
-	for i := 0; i < 300; i++ {
+	for i := 0; i < 30000; i++ {
 		_, err = trk.ReportOutcome(ctx, id, request.OutcomeSuccess)
 		assert.NoError(t, err)
 	}