Skip to content

Commit 11a6b2a

Browse files
committed
fix ut
1 parent a7c076e commit 11a6b2a

4 files changed

Lines changed: 729 additions & 19 deletions

File tree

backend/modules/evaluation/application/convertor/experiment/expt_test.go

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -2133,10 +2133,11 @@ func TestEvalConfConvert_ConvertToEntity_SetScoreWeight(t *testing.T) {
21332133

21342134
// 验证 ScoreWeight 被正确设置
21352135
for _, conf := range result.ConnectorConf.EvaluatorsConf.EvaluatorConf {
2136-
if conf.EvaluatorVersionID == 101 {
2136+
switch conf.EvaluatorVersionID {
2137+
case 101:
21372138
assert.NotNil(t, conf.ScoreWeight)
21382139
assert.Equal(t, scoreWeight1, *conf.ScoreWeight)
2139-
} else if conf.EvaluatorVersionID == 102 {
2140+
case 102:
21402141
assert.NotNil(t, conf.ScoreWeight)
21412142
assert.Equal(t, scoreWeight2, *conf.ScoreWeight)
21422143
}

backend/modules/evaluation/application/experiment_app_test.go

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -3379,9 +3379,9 @@ func TestExperimentApplication_ListExperimentTemplates_FilterOptionAndDefaultSor
33793379

33803380
t.Run("FilterOption为nil,不设置filters", func(t *testing.T) {
33813381
req := &exptpb.ListExperimentTemplatesRequest{
3382-
WorkspaceID: workspaceID,
3383-
PageNumber: gptr.Of(int32(1)),
3384-
PageSize: gptr.Of(int32(10)),
3382+
WorkspaceID: workspaceID,
3383+
PageNumber: gptr.Of(int32(1)),
3384+
PageSize: gptr.Of(int32(10)),
33853385
FilterOption: nil,
33863386
}
33873387

@@ -5010,7 +5010,7 @@ func TestExperimentApplication_UpdateAnnotateRecord(t *testing.T) {
50105010
DoAndReturn(func(_ context.Context, exptID int64, spaceID int64, session *entity.Session) (*entity.Experiment, error) {
50115011
return &entity.Experiment{
50125012
ID: exptID,
5013-
SpaceID: spaceID,
5013+
SpaceID: spaceID,
50145014
CreatedBy: "user1",
50155015
}, nil
50165016
})

backend/modules/evaluation/domain/service/expt_manage_impl_test.go

Lines changed: 15 additions & 13 deletions
Original file line numberDiff line numberDiff line change
@@ -1162,7 +1162,7 @@ func TestExptMangerImpl_fillExptTemplates(t *testing.T) {
11621162
mgrNoRepo.templateRepo = nil
11631163
expts := []*entity.Experiment{
11641164
{
1165-
ID: 1,
1165+
ID: 1,
11661166
ExptTemplateMeta: &entity.ExptTemplateMeta{ID: 100},
11671167
},
11681168
}
@@ -1184,11 +1184,11 @@ func TestExptMangerImpl_fillExptTemplates(t *testing.T) {
11841184
templateID2 := int64(200)
11851185
expts := []*entity.Experiment{
11861186
{
1187-
ID: 1,
1187+
ID: 1,
11881188
ExptTemplateMeta: &entity.ExptTemplateMeta{ID: templateID1},
11891189
},
11901190
{
1191-
ID: 2,
1191+
ID: 2,
11921192
ExptTemplateMeta: &entity.ExptTemplateMeta{ID: templateID2},
11931193
},
11941194
}
@@ -1222,9 +1222,10 @@ func TestExptMangerImpl_fillExptTemplates(t *testing.T) {
12221222
// 返回对应的模板
12231223
result := make([]*entity.ExptTemplate, 0, 2)
12241224
for _, id := range ids {
1225-
if id == templateID1 {
1225+
switch id {
1226+
case templateID1:
12261227
result = append(result, templates[0])
1227-
} else if id == templateID2 {
1228+
case templateID2:
12281229
result = append(result, templates[1])
12291230
}
12301231
}
@@ -1242,7 +1243,7 @@ func TestExptMangerImpl_fillExptTemplates(t *testing.T) {
12421243
templateID := int64(100)
12431244
expts := []*entity.Experiment{
12441245
{
1245-
ID: 1,
1246+
ID: 1,
12461247
ExptTemplateMeta: &entity.ExptTemplateMeta{ID: templateID},
12471248
},
12481249
}
@@ -1267,11 +1268,11 @@ func TestExptMangerImpl_fillExptTemplates(t *testing.T) {
12671268
templateID2 := int64(200)
12681269
expts := []*entity.Experiment{
12691270
{
1270-
ID: 1,
1271+
ID: 1,
12711272
ExptTemplateMeta: &entity.ExptTemplateMeta{ID: templateID1},
12721273
},
12731274
{
1274-
ID: 2,
1275+
ID: 2,
12751276
ExptTemplateMeta: &entity.ExptTemplateMeta{ID: templateID2},
12761277
},
12771278
}
@@ -1318,7 +1319,7 @@ func TestExptMangerImpl_fillExptTemplates(t *testing.T) {
13181319
templateID := int64(100)
13191320
expts := []*entity.Experiment{
13201321
{
1321-
ID: 1,
1322+
ID: 1,
13221323
ExptTemplateMeta: &entity.ExptTemplateMeta{ID: templateID},
13231324
},
13241325
}
@@ -1341,11 +1342,11 @@ func TestExptMangerImpl_fillExptTemplates(t *testing.T) {
13411342
templateID2 := int64(200)
13421343
expts := []*entity.Experiment{
13431344
{
1344-
ID: 1,
1345+
ID: 1,
13451346
ExptTemplateMeta: &entity.ExptTemplateMeta{ID: templateID1},
13461347
},
13471348
{
1348-
ID: 2,
1349+
ID: 2,
13491350
ExptTemplateMeta: &entity.ExptTemplateMeta{ID: templateID2},
13501351
},
13511352
}
@@ -1372,9 +1373,10 @@ func TestExptMangerImpl_fillExptTemplates(t *testing.T) {
13721373
// 返回对应的模板
13731374
result := make([]*entity.ExptTemplate, 0, 2)
13741375
for _, id := range ids {
1375-
if id == templateID1 {
1376+
switch id {
1377+
case templateID1:
13761378
result = append(result, templates[0])
1377-
} else if id == templateID2 {
1379+
case templateID2:
13781380
result = append(result, templates[1])
13791381
}
13801382
}

0 commit comments

Comments
 (0)