Skip to content

Commit 3a6ef3e

Browse files
committed
fix
1 parent edaad5d commit 3a6ef3e

File tree

1 file changed

+17
-17
lines changed

1 file changed

+17
-17
lines changed

internal/orchestrator/bricksindex/bricks_index_test.go

Lines changed: 17 additions & 17 deletions
Original file line numberDiff line numberDiff line change
@@ -169,29 +169,29 @@ func TestBricksIndex(t *testing.T) {
169169
require.Len(t, index.Bricks, 12)
170170

171171
// Check if ports are correctly set
172-
b_ic, found := index.FindBrickByID("arduino:web_ui")
172+
bWebUI, found := index.FindBrickByID("arduino:web_ui")
173173
require.True(t, found)
174-
require.Equal(t, []string{"7000"}, b_ic.Ports)
174+
require.Equal(t, []string{"7000"}, bWebUI.Ports)
175175

176176
// Check if variables are correctly set
177-
b_ic, found = index.FindBrickByID("arduino:image_classification")
177+
bWebUI, found = index.FindBrickByID("arduino:image_classification")
178178
require.True(t, found)
179-
require.Equal(t, "Image Classification", b_ic.Name)
180-
require.Equal(t, "mobilenet-image-classification", b_ic.ModelName)
181-
require.True(t, b_ic.RequireModel)
182-
require.Len(t, b_ic.Variables, 2)
183-
require.Equal(t, "CUSTOM_MODEL_PATH", b_ic.Variables[0].Name)
184-
require.Equal(t, "/opt/models/ei/", b_ic.Variables[0].DefaultValue)
185-
require.Equal(t, "path to the custom model directory", b_ic.Variables[0].Description)
186-
require.Equal(t, "EI_CLASSIFICATION_MODEL", b_ic.Variables[1].Name)
187-
require.Equal(t, "/models/ootb/ei/mobilenet-v2-224px.eim", b_ic.Variables[1].DefaultValue)
188-
require.Equal(t, "path to the model file", b_ic.Variables[1].Description)
189-
require.False(t, b_ic.Variables[0].IsRequired())
190-
require.False(t, b_ic.Variables[1].IsRequired())
179+
require.Equal(t, "Image Classification", bWebUI.Name)
180+
require.Equal(t, "mobilenet-image-classification", bWebUI.ModelName)
181+
require.True(t, bWebUI.RequireModel)
182+
require.Len(t, bWebUI.Variables, 2)
183+
require.Equal(t, "CUSTOM_MODEL_PATH", bWebUI.Variables[0].Name)
184+
require.Equal(t, "/opt/models/ei/", bWebUI.Variables[0].DefaultValue)
185+
require.Equal(t, "path to the custom model directory", bWebUI.Variables[0].Description)
186+
require.Equal(t, "EI_CLASSIFICATION_MODEL", bWebUI.Variables[1].Name)
187+
require.Equal(t, "/models/ootb/ei/mobilenet-v2-224px.eim", bWebUI.Variables[1].DefaultValue)
188+
require.Equal(t, "path to the model file", bWebUI.Variables[1].Description)
189+
require.False(t, bWebUI.Variables[0].IsRequired())
190+
require.False(t, bWebUI.Variables[1].IsRequired())
191191

192-
b_db, found := index.FindBrickByID("arduino:dbstorage_tsstore")
192+
bDb, found := index.FindBrickByID("arduino:dbstorage_tsstore")
193193
require.True(t, found)
194-
require.False(t, b_db.RequireModel)
194+
require.False(t, bDb.RequireModel)
195195

196196
bNoRequireModel, found := index.FindBrickByID("arduino:missing-model-require")
197197
require.True(t, found)

0 commit comments

Comments
 (0)